diff --git a/src/agents/productionAgent/index.ts b/src/agents/productionAgent/index.ts index bbbc725..08f6725 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 ecbfe71..dc02a15 100644 --- a/src/types/database.d.ts +++ b/src/types/database.d.ts @@ -1,43 +1,6 @@ -// @db-hash 147d0f569132c3ba4fedb17a1039d15f +// @db-hash 6aa15a584eba838157eddf2458c0e260 //该文件由脚本自动生成,请勿手动修改 -export interface _o_storyboard_old_20260402 { - 'createTime'?: number | null; - 'duration'?: string | null; - 'filePath'?: string | null; - 'flowId'?: number | null; - 'id'?: number; - 'index'?: number | null; - 'projectId'?: number | null; - 'prompt'?: string | null; - 'reason'?: string | null; - 'scriptId'?: number | null; - 'state'?: string | null; - 'trackId'?: number | null; -} -export interface _o_vendorConfig_old_20260401 { - 'author'?: string | null; - 'code'?: string | null; - 'createTime'?: number | null; - 'description'?: string | null; - 'enableEnglish'?: number | null; - 'icon'?: string | null; - 'id'?: string; - 'inputs'?: string | null; - 'inputValues'?: string | null; - 'models'?: string | null; - 'name'?: string | null; -} -export interface _o_videoTrack_old_20260402 { - 'id'?: number; - 'projectId'?: number | null; - 'prompt'?: string | null; - 'reason'?: string | null; - 'scriptId'?: number | null; - 'selectVideoId'?: number | null; - 'state'?: string | null; - 'videoId'?: number | null; -} export interface memories { 'content': string; 'createTime': number; @@ -210,11 +173,8 @@ export interface o_storyboard { 'prompt'?: string | null; 'reason'?: string | null; 'scriptId'?: number | null; - 'shouldGenerateImage'?: number | null; 'state'?: string | null; - 'track'?: string | null; 'trackId'?: number | null; - 'videoPrompt'?: string | null; } export interface o_tasks { 'describe'?: string | null; @@ -238,7 +198,6 @@ export interface o_vendorConfig { 'createTime'?: number | null; 'description'?: string | null; 'enable'?: number | null; - 'enableEnglish'?: number | null; 'icon'?: string | null; 'id'?: string; 'inputs'?: string | null; @@ -257,20 +216,17 @@ export interface o_video { 'videoTrackId'?: number | null; } export interface o_videoTrack { - 'duration'?: number | null; 'id'?: number; 'projectId'?: number | null; 'prompt'?: string | null; 'reason'?: string | null; 'scriptId'?: number | null; + 'selectVideoId'?: number | null; 'state'?: string | null; 'videoId'?: number | null; } export interface DB { - "_o_storyboard_old_20260402": _o_storyboard_old_20260402; - "_o_vendorConfig_old_20260401": _o_vendorConfig_old_20260401; - "_o_videoTrack_old_20260402": _o_videoTrack_old_20260402; "memories": memories; "o_agentDeploy": o_agentDeploy; "o_agentWorkData": o_agentWorkData;