diff --git a/fix_list_config.py b/fix_list_config.py new file mode 100644 index 000000000..6355326ac --- /dev/null +++ b/fix_list_config.py @@ -0,0 +1,19 @@ +path = r"D:\yingce\open-ai-canvas-main\web\src\pages\canvas\list-mode-generator.ts" +with open(path, "r", encoding="utf-8") as f: + content = f.read() + +# Fix import +content = content.replace( + 'import { resolveCanvasGenerationModel } from "@/lib/canvas/canvas-project-generation";', + 'import { buildGenerationConfig } from "@/lib/canvas/canvas-project-generation";' +) + +# Fix config building - use buildGenerationConfig instead of resolveCanvasGenerationModel +content = content.replace( + ' const model = resolveCanvasGenerationModel(config, undefined, "text");\n const listConfig = { ...config, model };', + ' const listConfig = buildGenerationConfig(config, sourceNode, "text");' +) + +with open(path, "w", encoding="utf-8") as f: + f.write(content) +print("done") diff --git a/web/src/pages/canvas/list-mode-generator.ts b/web/src/pages/canvas/list-mode-generator.ts index f279ed16d..15a21bc8e 100644 --- a/web/src/pages/canvas/list-mode-generator.ts +++ b/web/src/pages/canvas/list-mode-generator.ts @@ -2,7 +2,7 @@ import { nanoid } from "nanoid"; import { message } from "antd"; import { CanvasNodeType, type CanvasConnection, type CanvasNodeData, type CanvasBatchTableData, type CanvasBatchRow } from "@/types/canvas"; import { runBackendGenerationTask } from "@/services/api/generation-task"; -import { resolveCanvasGenerationModel } from "@/lib/canvas/canvas-project-generation"; +import { buildGenerationConfig } from "@/lib/canvas/canvas-project-generation"; import type { AiConfig } from "@/stores/use-config-store"; const LIST_MODE_SYSTEM_PROMPT = `你是一个电商内容分析助手。用户会给你多张产品图片和一个任务描述。 @@ -95,8 +95,7 @@ export async function handleListGenerate({ setRunningNodeId(sourceNodeId); try { - const model = resolveCanvasGenerationModel(config, undefined, "text"); - const listConfig = { ...config, model }; + const listConfig = buildGenerationConfig(config, sourceNode, "text"); const referenceImages = imageNodes.map((node) => ({ id: node.id,