diff --git a/web/app/components/workflow/hooks/use-workflow.ts b/web/app/components/workflow/hooks/use-workflow.ts index 3c27b5c91b..03ad421572 100644 --- a/web/app/components/workflow/hooks/use-workflow.ts +++ b/web/app/components/workflow/hooks/use-workflow.ts @@ -481,7 +481,6 @@ export const useWorkflowInit = () => { return acc }, {} as Record), environmentVariables: res.environment_variables?.map(env => env.value_type === 'secret' ? { ...env, value: '[__HIDDEN__]' } : env) || [], - // #TODO chatVar sync# conversationVariables: res.conversation_variables || [], }) setSyncWorkflowDraftHash(res.hash) diff --git a/web/app/components/workflow/utils.ts b/web/app/components/workflow/utils.ts index aaf333f4d7..0264b00f3d 100644 --- a/web/app/components/workflow/utils.ts +++ b/web/app/components/workflow/utils.ts @@ -212,6 +212,13 @@ export const preprocessNodesAndEdges = (nodes: Node[], edges: Edge[]) => { } } +export const correctProvider = (provider: string) => { + if (provider.includes('/')) + return provider + + return `langgenius/${provider}/${provider}` +} + export const initialNodes = (originNodes: Node[], originEdges: Edge[]) => { const { nodes, edges } = preprocessNodesAndEdges(cloneDeep(originNodes), cloneDeep(originEdges)) const firstNode = nodes[0] @@ -275,6 +282,19 @@ export const initialNodes = (originNodes: Node[], originEdges: Edge[]) => { iterationNodeData.error_handle_mode = iterationNodeData.error_handle_mode || ErrorHandleMode.Terminated } + // legacy provider handle + if (node.data.type === BlockEnum.LLM) + (node as any).data.model.provider = correctProvider((node as any).data.model.provider) + + if (node.data.type === BlockEnum.KnowledgeRetrieval && (node as any).data.multiple_retrieval_config.reranking_model) + (node as any).data.multiple_retrieval_config.reranking_model.provider = correctProvider((node as any).data.multiple_retrieval_config.reranking_model.provider) + + if (node.data.type === BlockEnum.QuestionClassifier) + (node as any).data.model.provider = correctProvider((node as any).data.model.provider) + + if (node.data.type === BlockEnum.ParameterExtractor) + (node as any).data.model.provider = correctProvider((node as any).data.model.provider) + return node }) } @@ -428,7 +448,7 @@ export const genNewNodeTitleFromOld = (oldTitle: string) => { if (match) { const title = match[1] - const num = parseInt(match[2], 10) + const num = Number.parseInt(match[2], 10) return `${title} (${num + 1})` } else {