diff --git a/src/lib/initDB.ts b/src/lib/initDB.ts index 235a9b1..8e28da3 100644 --- a/src/lib/initDB.ts +++ b/src/lib/initDB.ts @@ -355,6 +355,11 @@ description: 专注于从剧本内容中提取所使用的资产(角色、场 - 场景中的固定陈设不需要单独提取为道具,除非该物件有独立剧情作用 `, }, + { + name: "视频提示词生成", + type: "videoPromptGeneration", + data: "根据以下提示词生成一个视频提示词", + }, ]); }, }, diff --git a/src/routes/production/workbench/generateVideoPrompt.ts b/src/routes/production/workbench/generateVideoPrompt.ts index 2c49ebb..f1990f6 100644 --- a/src/routes/production/workbench/generateVideoPrompt.ts +++ b/src/routes/production/workbench/generateVideoPrompt.ts @@ -17,10 +17,11 @@ export default router.post( const { trackId, projectId, prompt, model } = req.body; const [id, modelData] = model.split(":"); const projectData = await u.db("o_project").select("*").where({ id: projectId }).first(); + const videoPrompt = await u.db("o_prompt").where("type", "videoPromptGeneration").first(); const artStyle = projectData?.artStyle || "无"; const visualManual = u.getArtPrompt(artStyle, "art_storyboard_video"); const { text } = await u.Ai.Text("universalAi").invoke({ - system: visualManual, + system: `${videoPrompt?.data},${visualManual}`, messages: [ { role: "user",