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;