diff --git a/src/agents/scriptAgent/index.ts b/src/agents/scriptAgent/index.ts index e507e7f..0f727bd 100644 --- a/src/agents/scriptAgent/index.ts +++ b/src/agents/scriptAgent/index.ts @@ -47,6 +47,7 @@ export async function decisionAI(ctx: AgentContext) { const projectData = await u.db("o_project").where("id", resTool.data.projectId).first(); const novelData = await u.db("o_novel").where("projectId", resTool.data.projectId).select("id", "chapterIndex as index"); + console.log("%c Line:50 🥒 novelData", "background:#2eafb0", novelData); const projectInfo = [ "## 项目信息", @@ -57,7 +58,7 @@ export async function decisionAI(ctx: AgentContext) { `目标改编视频画幅:${projectData?.videoRatio ?? "16:9"}`, ].join("\n"); - const prefixSystem = `${projectInfo}\n\n## 章节ID映射表\n${novelData.map((i: any) => `- ${i.id}: 第${i.index}章`).join("\n")}\n\n`; + const prefixSystem = `${projectInfo}\n\n## 章节ID映射表\n${novelData.map((i: any) => `- 章节ID:${i.id}: 第${i.index}章`).join("\n")}\n\n`; const { textStream } = await u.Ai.Text("scriptAgent").stream({ system: prefixSystem + systemPrompt, @@ -152,7 +153,6 @@ function runSubAgent(parentCtx: AgentContext) { let fullResponse = ""; for await (const chunk of subTextStream) { - console.log("%c Line:155 🥛 chunk", "background:#fca650", chunk); msg.send(chunk); fullResponse += chunk; } diff --git a/src/agents/scriptAgent/tools.ts b/src/agents/scriptAgent/tools.ts index 420416e..f05d736 100644 --- a/src/agents/scriptAgent/tools.ts +++ b/src/agents/scriptAgent/tools.ts @@ -35,7 +35,7 @@ export default (resTool: ResTool, toolsNames?: string[]) => { get_novel_events: tool({ description: "获取章节事件", inputSchema: z.object({ - ids: z.array(z.number()).describe("章节id"), + ids: z.array(z.number()).describe("章节id,注意区分"), }), execute: async ({ ids }) => { resTool.systemMessage(`正在阅读 章节事件 数据...`);