zyc/video-flow-toon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加章节ID明确标注

Browse Source
This commit is contained in:
zhishi 2026-03-27 02:00:09 +08:00
parent f06e7d7f9c
commit 1064bf77c8
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/agents/scriptAgent/index.ts
View File

@ -47,6 +47,7 @@ export async function decisionAI(ctx: AgentContext) {
const projectData = await u.db("o_project").where("id", resTool.data.projectId).first(); 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"); 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 = [ const projectInfo = [
"## 项目信息", "## 项目信息",
@ -57,7 +58,7 @@ export async function decisionAI(ctx: AgentContext) {
`目标改编视频画幅：${projectData?.videoRatio ?? "16:9"}`, `目标改编视频画幅：${projectData?.videoRatio ?? "16:9"}`,
].join("\n"); ].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({ const { textStream } = await u.Ai.Text("scriptAgent").stream({
system: prefixSystem + systemPrompt, system: prefixSystem + systemPrompt,
@ -152,7 +153,6 @@ function runSubAgent(parentCtx: AgentContext) {
let fullResponse = ""; let fullResponse = "";
for await (const chunk of subTextStream) { for await (const chunk of subTextStream) {
console.log("%c Line:155 🥛 chunk", "background:#fca650", chunk);
msg.send(chunk); msg.send(chunk);
fullResponse += chunk; fullResponse += chunk;
} }

2
src/agents/scriptAgent/tools.ts
View File

@ -35,7 +35,7 @@ export default (resTool: ResTool, toolsNames?: string[]) => {
get_novel_events: tool({ get_novel_events: tool({
description: "获取章节事件", description: "获取章节事件",
inputSchema: z.object({ inputSchema: z.object({
ids: z.array(z.number()).describe("章节id"), ids: z.array(z.number()).describe("章节id，注意区分"),
}), }),
execute: async ({ ids }) => { execute: async ({ ids }) => {
resTool.systemMessage(`正在阅读 章节事件 数据...`); resTool.systemMessage(`正在阅读 章节事件 数据...`);