59
59
<WorkflowDesign
60
60
v-if =" currentStep === 1"
61
61
v-model =" formData"
62
- :provider =" provider "
62
+ :provider =" llmProvider "
63
63
ref =" workflowDesignRef"
64
64
/>
65
65
</div >
@@ -74,6 +74,7 @@ import * as WorkflowApi from '@/api/ai/workflow'
74
74
import BasicInfo from ' ./BasicInfo.vue'
75
75
import WorkflowDesign from ' ./WorkflowDesign.vue'
76
76
import { ModelApi } from ' @/api/ai/model/model'
77
+ import { AiModelTypeEnum } from ' @/views/ai/utils/constants'
77
78
78
79
const router = useRouter ()
79
80
const { delView } = useTagsViewStore ()
@@ -104,31 +105,35 @@ const formData: any = ref({
104
105
graph: ' ' ,
105
106
status: CommonStatusEnum .ENABLE
106
107
})
107
- // TODO @lesan:待接入
108
- const provider = ref <any >()
108
+ const llmProvider = ref <any >([])
109
109
const workflowData = ref <any >({})
110
110
provide (' workflowData' , workflowData )
111
111
112
112
/** 初始化数据 */
113
113
const actionType = route .params .type as string
114
114
const initData = async () => {
115
+ // 编辑情况下,需要加载工作流配置
115
116
if (actionType === ' update' ) {
116
117
const workflowId = route .params .id as string
117
118
formData .value = await WorkflowApi .getWorkflow (workflowId )
118
119
workflowData .value = JSON .parse (formData .value .graph )
119
120
}
120
121
121
- const apiKeys = await ModelApi .getModelSimpleList (1 )
122
- provider .value = {
122
+ // 加载模型列表
123
+ const models = await ModelApi .getModelSimpleList (AiModelTypeEnum .CHAT )
124
+ llmProvider .value = {
123
125
llm : () =>
124
- apiKeys .map (({ id , name }) => ({
126
+ models .map (({ id , name }) => ({
125
127
value: id ,
126
128
label: name
127
129
})),
128
130
knowledge : () => [],
129
131
internal : () => []
130
132
}
133
+ // TODO @lesan:知识库(可以看下 knowledge)
134
+ // TODO @lesan:搜索引擎(这个之前有个 pr 搞了,,,可能来接下)
131
135
136
+ // 设置当前步骤
132
137
currentStep .value = 0
133
138
}
134
139
@@ -164,17 +169,17 @@ const handleSave = async () => {
164
169
165
170
// 更新表单数据
166
171
const data = {
167
- ... formData .value
172
+ ... formData .value ,
173
+ graph: JSON .stringify (workflowData .value )
168
174
}
169
-
170
- data .graph = JSON .stringify (workflowData .value )
171
-
172
175
if (actionType === ' update' ) {
173
176
await WorkflowApi .updateWorkflow (data )
174
177
} else {
175
178
await WorkflowApi .createWorkflow (data )
176
179
}
177
180
181
+ // 保存成功,提示并跳转到列表页
182
+ message .success (' 保存成功' )
178
183
delView (unref (router .currentRoute ))
179
184
await router .push ({ name: ' AiWorkflow' })
180
185
} catch (error : any ) {
0 commit comments