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;