76 lines
2.4 KiB
TypeScript
76 lines
2.4 KiB
TypeScript
import { tool } from "ai";
|
||
import { z } from "zod";
|
||
import u from "@/utils";
|
||
import { useSkill } from "@/utils/agent/skillsTools";
|
||
import { createAGUIStream } from "@/utils/agent/aguiTools";
|
||
|
||
interface FlowData {
|
||
script: {
|
||
blocks: string[];
|
||
};
|
||
}
|
||
|
||
export default (isolationKey: string, agui: ReturnType<typeof createAGUIStream>) => {
|
||
const flowData: FlowData = {
|
||
script: {
|
||
blocks: [],
|
||
},
|
||
};
|
||
return {
|
||
get_project_info: tool({
|
||
description: "获取项目信息",
|
||
inputSchema: z.object({}),
|
||
execute: async () => {
|
||
return `
|
||
项目名称：仙逆
|
||
视频风格：玄幻3D动漫
|
||
视频类型：短剧
|
||
项目描述：讲述了乡村平凡少年王林以心中之感动，逆仙而修，求的不仅是长生，更多的是摆脱那背后的蝼蚁之身。他坚信道在人为，以平庸的资质踏入修真仙途，历经坎坷风雨，凭着其聪睿的心智，一步一步走向巅峰，凭一己之力，扬名修真界。
|
||
总集数：24集每集2分钟
|
||
当前集数：3集
|
||
`;
|
||
},
|
||
}),
|
||
get_state: tool({
|
||
description: "获取工作流指定板块数据",
|
||
inputSchema: z.object({
|
||
block: z.enum(["script"]).describe("板块名称，如 script"),
|
||
}),
|
||
execute: async ({ block }) => {
|
||
return flowData[block];
|
||
},
|
||
}),
|
||
execution: tool({
|
||
description: "执行层，负责具体执行具体的任务",
|
||
inputSchema: z.object({
|
||
taskDescription: z.string().describe("具体的任务描述详细信息"),
|
||
}),
|
||
execute: async ({ taskDescription }) => {
|
||
agui.custom("systemMessage", "已由 执行层AI 接管对话");
|
||
|
||
const skill = await useSkill("production-agent", "execution");
|
||
|
||
const { textStream } = await u.Ai.Text("productionAgent").stream({
|
||
system: skill.prompt,
|
||
messages: [{ role: "user", content: `请完成任务：${taskDescription}` }],
|
||
tools: {
|
||
...skill.tools,
|
||
},
|
||
});
|
||
|
||
let msg: ReturnType<typeof agui.textMessage> | null = null;
|
||
let fullResponse = "";
|
||
|
||
for await (const chunk of textStream) {
|
||
if (!msg) msg = agui.textMessage();
|
||
msg.send(chunk);
|
||
fullResponse += chunk;
|
||
}
|
||
msg?.end();
|
||
|
||
return { found: true, memories: ["第一条记忆内容", "第二条记忆内容"] };
|
||
},
|
||
}),
|
||
};
|
||
};
|