diff --git a/src/agents/productionAgent/index.ts b/src/agents/productionAgent/index.ts index 07ea389..14d1158 100644 --- a/src/agents/productionAgent/index.ts +++ b/src/agents/productionAgent/index.ts @@ -40,9 +40,6 @@ export async function decisionAI(ctx: AgentContext) { const memory = new Memory("productionAgent", isolationKey); await memory.add("user", text); - // const { skillPaths } = await useSkill({ mainSkill: "production_agent_decision" }); - // const prompt = await fs.promises.readFile(skillPaths.mainSkill, "utf-8"); - const skill = path.join(u.getPath("skills"), "production_agent_decision.md"); const prompt = await fs.promises.readFile(skill, "utf-8"); @@ -141,7 +138,8 @@ function createSubAgent(parentCtx: AgentContext) { "分镜面板:", "```", ].join("\n"); - // "剧本:", + const projectData = await u.db("o_project").where("id", resTool.data.projectId).first(); + const modelInfo = `项目使用的模型如下:\n图像模型:${projectData?.imageModel}\n视频模型:${projectData?.videoModel}`; const projectInfo = await u.db("o_project").where("id", resTool.data.projectId).first(); if (!projectInfo) throw new Error(`项目不存在,ID: ${resTool.data.projectId}`); @@ -153,7 +151,7 @@ function createSubAgent(parentCtx: AgentContext) { name: "执行导演", memoryKey: "assistant:execution", messages: [ - { role: "assistant", content: artSkills.prompt }, + { role: "assistant", content: artSkills.prompt + `\n${modelInfo}` }, { role: "user", content: prompt }, ], tools: { ...artSkills.tools }, diff --git a/src/types/database.d.ts b/src/types/database.d.ts index 04139ce..dc02a15 100644 --- a/src/types/database.d.ts +++ b/src/types/database.d.ts @@ -1,4 +1,4 @@ -// @db-hash c0d74bd27b3a41b397705c93d1737a3b +// @db-hash 6aa15a584eba838157eddf2458c0e260 //该文件由脚本自动生成,请勿手动修改 export interface memories { @@ -197,7 +197,7 @@ export interface o_vendorConfig { 'code'?: string | null; 'createTime'?: number | null; 'description'?: string | null; - 'enableEnglish'?: number | null; + 'enable'?: number | null; 'icon'?: string | null; 'id'?: string; 'inputs'?: string | null;