zyc/video-flow-toon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch '108' of https://github.com/HBAI-Ltd/Toonflow-app into 108

Browse Source
This commit is contained in:
ACT丶流星雨 2026-03-24 14:59:11 +08:00
commit f1389a370d
18 changed files with 329 additions and 216 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/agents/scriptAgent/index.ts
View File

@ -40,7 +40,7 @@ export async function decisionAI(ctx: AgentContext) {
resTool.systemMessage("决策层AI 接管聊天"); resTool.systemMessage("决策层AI 接管聊天");
const memory = new Memory("scriptAgent", isolationKey); const memory = new Memory("scriptAgent", isolationKey);
console.log("%c Line:43 🥟 isolationKey", "background:#4fff4B", isolationKey);
await memory.add("user", text); await memory.add("user", text);
const [skill, mem] = await Promise.all([useSkill("script-agent", "decision"), memory.get(text)]); const [skill, mem] = await Promise.all([useSkill("script-agent", "decision"), memory.get(text)]);
@ -58,11 +58,9 @@ export async function decisionAI(ctx: AgentContext) {
`目标改编视频画幅：${projectData?.videoRatio ?? "16:9"}`, `目标改编视频画幅：${projectData?.videoRatio ?? "16:9"}`,
].join("\n"); ].join("\n");
// const prefixSystem = `${projectInfo}\n\n## 章节ID映射表\n${novelData.map((i: any) => `- ${i.id}: 第${i.index}章`).join("\n")}\n\n`; const prefixSystem = `${projectInfo}\n\n## 章节ID映射表\n${novelData.map((i: any) => `- ${i.id}: 第${i.index}章`).join("\n")}\n\n`;
const prefixSystem = `不管说什么直接调用 insert_script_to_sqlite 工具`
console.log("%c Line:57 🍧 prefixSystem", "background:#ea7e5c", prefixSystem); console.log("%c Line:57 🍧 prefixSystem", "background:#ea7e5c", prefixSystem);
const { textStream } = await u.Ai.Text("scriptAgent").stream({ const { textStream } = await u.Ai.Text("scriptAgent").stream({
system: prefixSystem + systemPrompt, system: prefixSystem + systemPrompt,
messages: [{ role: "user", content: text }], messages: [{ role: "user", content: text }],

5
src/agents/scriptAgent/tools.ts
View File

@ -117,14 +117,13 @@ export default (resTool: ResTool, toolsNames?: string[]) => {
type: i.type, type: i.type,
describe: i.desc, describe: i.desc,
projectId: resTool.data.projectId, projectId: resTool.data.projectId,
state: "未生成",
}); });
assetId.push(id); assetId.push(id);
} }
await u.db("o_script_assets").insert(assetId.map((i) => ({ scriptId, assetId: i }))); await u.db("o_scriptAssets").insert(assetId.map((i) => ({ scriptId, assetId: i })));
} }
socket.emit("setPlanData", { key: "script", value: scriptId });
return true; return true;
}, },
}), }),

4
src/lib/initDB.ts
View File

@ -44,7 +44,9 @@ export default async (knex: Knex, forceInit: boolean = false): Promise<void> =>
builder: (table) => { builder: (table) => {
table.integer("id").notNullable(); table.integer("id").notNullable();
table.string("name"); table.string("name");
table.text("styles"); table.text("fileUrl");
table.text("label");
table.text("prompt");
table.primary(["id"]); table.primary(["id"]);
table.unique(["id"]); table.unique(["id"]);
}, },

320
src/router.ts
View File

@ -1,166 +1,172 @@
// @routes-hash 060ac05250962088b128ef8200ab0471 // @routes-hash 055f8c83508ff9dfc8b98eb5108287f7
import { Express } from "express"; import { Express } from "express";
import route1 from "./routes/agents/clearMemory"; import route1 from "./routes/agents/clearMemory";
import route2 from "./routes/agents/getMemory"; import route2 from "./routes/agents/getMemory";
import route3 from "./routes/artStyle/getArtStyle"; import route3 from "./routes/artStyle/addArtStyle";
import route4 from "./routes/assets/addAssets"; import route4 from "./routes/artStyle/editArtStyle";
import route5 from "./routes/assets/batchDelete"; import route5 from "./routes/artStyle/extractStylePrompt";
import route6 from "./routes/assets/batchGenerationData"; import route6 from "./routes/artStyle/getArtStyle";
import route7 from "./routes/assets/delAssets"; import route7 from "./routes/assets/addAssets";
import route8 from "./routes/assets/getAssetsApi"; import route8 from "./routes/assets/batchDelete";
import route9 from "./routes/assets/getImage"; import route9 from "./routes/assets/batchGenerationData";
import route10 from "./routes/assets/saveAssets"; import route10 from "./routes/assets/delAssets";
import route11 from "./routes/assets/updateAssets"; import route11 from "./routes/assets/getAssetsApi";
import route12 from "./routes/assets/uploadClip"; import route12 from "./routes/assets/getImage";
import route13 from "./routes/assetsGenerate/generateAssets"; import route13 from "./routes/assets/saveAssets";
import route14 from "./routes/assetsGenerate/polishAssetsPrompt"; import route14 from "./routes/assets/updateAssets";
import route15 from "./routes/cornerScape/getAllAssets"; import route15 from "./routes/assets/uploadClip";
import route16 from "./routes/general/generalStatistics"; import route16 from "./routes/assetsGenerate/generateAssets";
import route17 from "./routes/general/getSingleProject"; import route17 from "./routes/assetsGenerate/polishAssetsPrompt";
import route18 from "./routes/general/updateProject"; import route18 from "./routes/cornerScape/getAllAssets";
import route19 from "./routes/login/login"; import route19 from "./routes/general/generalStatistics";
import route20 from "./routes/migrate/migrateData"; import route20 from "./routes/general/getSingleProject";
import route21 from "./routes/modelSelect/getModelDetail"; import route21 from "./routes/general/updateProject";
import route22 from "./routes/modelSelect/getModelList"; import route22 from "./routes/login/login";
import route23 from "./routes/novel/addNovel"; import route23 from "./routes/migrate/migrateData";
import route24 from "./routes/novel/batchDeleteNovel"; import route24 from "./routes/modelSelect/getModelDetail";
import route25 from "./routes/novel/delNovel"; import route25 from "./routes/modelSelect/getModelList";
import route26 from "./routes/novel/event/batchDeleteEvent"; import route26 from "./routes/novel/addNovel";
import route27 from "./routes/novel/event/deletEvent"; import route27 from "./routes/novel/batchDeleteNovel";
import route28 from "./routes/novel/event/generateEvents"; import route28 from "./routes/novel/delNovel";
import route29 from "./routes/novel/event/getEvent"; import route29 from "./routes/novel/event/batchDeleteEvent";
import route30 from "./routes/novel/getNovel"; import route30 from "./routes/novel/event/deletEvent";
import route31 from "./routes/novel/getNovelEventState"; import route31 from "./routes/novel/event/generateEvents";
import route32 from "./routes/novel/getNovelIndex"; import route32 from "./routes/novel/event/getEvent";
import route33 from "./routes/novel/updateNovel"; import route33 from "./routes/novel/getNovel";
import route34 from "./routes/other/deleteAllData"; import route34 from "./routes/novel/getNovelEventState";
import route35 from "./routes/other/getCaptcha"; import route35 from "./routes/novel/getNovelIndex";
import route36 from "./routes/production/assets/getAssetsData"; import route36 from "./routes/novel/updateNovel";
import route37 from "./routes/production/editStoryboard/generateStoryboardImage"; import route37 from "./routes/other/deleteAllData";
import route38 from "./routes/production/editStoryboard/getStoryboardFlow"; import route38 from "./routes/other/getCaptcha";
import route39 from "./routes/production/editStoryboard/saveStoryboardFlow"; import route39 from "./routes/production/assets/getAssetsData";
import route40 from "./routes/production/editStoryboard/updateStoryboardFlow"; import route40 from "./routes/production/editStoryboard/generateStoryboardImage";
import route41 from "./routes/production/exportImage"; import route41 from "./routes/production/editStoryboard/getStoryboardFlow";
import route42 from "./routes/production/getFlowData"; import route42 from "./routes/production/editStoryboard/saveStoryboardFlow";
import route43 from "./routes/production/getProductionData"; import route43 from "./routes/production/editStoryboard/updateStoryboardFlow";
import route44 from "./routes/production/getStoryboardData"; import route44 from "./routes/production/exportImage";
import route45 from "./routes/production/saveFlowData"; import route45 from "./routes/production/getFlowData";
import route46 from "./routes/production/workbench/confirmSelection"; import route46 from "./routes/production/getProductionData";
import route47 from "./routes/production/workbench/delVideo"; import route47 from "./routes/production/getStoryboardData";
import route48 from "./routes/production/workbench/generateVideo"; import route48 from "./routes/production/saveFlowData";
import route49 from "./routes/production/workbench/getChatLines"; import route49 from "./routes/production/workbench/confirmSelection";
import route50 from "./routes/production/workbench/getVideoModelDetail"; import route50 from "./routes/production/workbench/delVideo";
import route51 from "./routes/production/workbench/videoPolling"; import route51 from "./routes/production/workbench/generateVideo";
import route52 from "./routes/project/addProject"; import route52 from "./routes/production/workbench/getChatLines";
import route53 from "./routes/project/delProject"; import route53 from "./routes/production/workbench/getVideoModelDetail";
import route54 from "./routes/project/editProject"; import route54 from "./routes/production/workbench/videoPolling";
import route55 from "./routes/project/getProject"; import route55 from "./routes/project/addProject";
import route56 from "./routes/script/addScript"; import route56 from "./routes/project/delProject";
import route57 from "./routes/script/delScript"; import route57 from "./routes/project/editProject";
import route58 from "./routes/script/exportScript"; import route58 from "./routes/project/getProject";
import route59 from "./routes/script/getScrptApi"; import route59 from "./routes/script/addScript";
import route60 from "./routes/script/updateScript"; import route60 from "./routes/script/delScript";
import route61 from "./routes/scriptAgent/getPlanData"; import route61 from "./routes/script/exportScript";
import route62 from "./routes/scriptAgent/setPlanData"; import route62 from "./routes/script/getScrptApi";
import route63 from "./routes/setting/agentDeploy/deployAgentModel"; import route63 from "./routes/script/updateScript";
import route64 from "./routes/setting/agentDeploy/getAgentDeploy"; import route64 from "./routes/scriptAgent/getPlanData";
import route65 from "./routes/setting/agentDeploy/oneClickToFillIn"; import route65 from "./routes/scriptAgent/setPlanData";
import route66 from "./routes/setting/dbConfig/clearData"; import route66 from "./routes/setting/agentDeploy/agentSetKey";
import route67 from "./routes/setting/getTextModel"; import route67 from "./routes/setting/agentDeploy/deployAgentModel";
import route68 from "./routes/setting/loginConfig/getUser"; import route68 from "./routes/setting/agentDeploy/getAgentDeploy";
import route69 from "./routes/setting/loginConfig/updateUserPwd"; import route69 from "./routes/setting/dbConfig/clearData";
import route70 from "./routes/setting/memoryConfig/getMemory"; import route70 from "./routes/setting/getTextModel";
import route71 from "./routes/setting/memoryConfig/sureMemory"; import route71 from "./routes/setting/loginConfig/getUser";
import route72 from "./routes/setting/vendorConfig/addVendor"; import route72 from "./routes/setting/loginConfig/updateUserPwd";
import route73 from "./routes/setting/vendorConfig/deleteVendor"; import route73 from "./routes/setting/memoryConfig/getMemory";
import route74 from "./routes/setting/vendorConfig/getVendorList"; import route74 from "./routes/setting/memoryConfig/sureMemory";
import route75 from "./routes/setting/vendorConfig/modelTest"; import route75 from "./routes/setting/vendorConfig/addVendor";
import route76 from "./routes/setting/vendorConfig/updateVendor"; import route76 from "./routes/setting/vendorConfig/deleteVendor";
import route77 from "./routes/task/getTaskApi"; import route77 from "./routes/setting/vendorConfig/getVendorList";
import route78 from "./routes/task/getTaskCategories"; import route78 from "./routes/setting/vendorConfig/modelTest";
import route79 from "./routes/task/taskDetails"; import route79 from "./routes/setting/vendorConfig/updateVendor";
import route80 from "./routes/test/test"; import route80 from "./routes/task/getTaskApi";
import route81 from "./routes/task/getTaskCategories";
import route82 from "./routes/task/taskDetails";
import route83 from "./routes/test/test";
export default async (app: Express) => { export default async (app: Express) => {
app.use("/api/agents/clearMemory", route1); app.use("/api/agents/clearMemory", route1);
app.use("/api/agents/getMemory", route2); app.use("/api/agents/getMemory", route2);
app.use("/api/artStyle/getArtStyle", route3); app.use("/api/artStyle/addArtStyle", route3);
app.use("/api/assets/addAssets", route4); app.use("/api/artStyle/editArtStyle", route4);
app.use("/api/assets/batchDelete", route5); app.use("/api/artStyle/extractStylePrompt", route5);
app.use("/api/assets/batchGenerationData", route6); app.use("/api/artStyle/getArtStyle", route6);
app.use("/api/assets/delAssets", route7); app.use("/api/assets/addAssets", route7);
app.use("/api/assets/getAssetsApi", route8); app.use("/api/assets/batchDelete", route8);
app.use("/api/assets/getImage", route9); app.use("/api/assets/batchGenerationData", route9);
app.use("/api/assets/saveAssets", route10); app.use("/api/assets/delAssets", route10);
app.use("/api/assets/updateAssets", route11); app.use("/api/assets/getAssetsApi", route11);
app.use("/api/assets/uploadClip", route12); app.use("/api/assets/getImage", route12);
app.use("/api/assetsGenerate/generateAssets", route13); app.use("/api/assets/saveAssets", route13);
app.use("/api/assetsGenerate/polishAssetsPrompt", route14); app.use("/api/assets/updateAssets", route14);
app.use("/api/cornerScape/getAllAssets", route15); app.use("/api/assets/uploadClip", route15);
app.use("/api/general/generalStatistics", route16); app.use("/api/assetsGenerate/generateAssets", route16);
app.use("/api/general/getSingleProject", route17); app.use("/api/assetsGenerate/polishAssetsPrompt", route17);
app.use("/api/general/updateProject", route18); app.use("/api/cornerScape/getAllAssets", route18);
app.use("/api/login/login", route19); app.use("/api/general/generalStatistics", route19);
app.use("/api/migrate/migrateData", route20); app.use("/api/general/getSingleProject", route20);
app.use("/api/modelSelect/getModelDetail", route21); app.use("/api/general/updateProject", route21);
app.use("/api/modelSelect/getModelList", route22); app.use("/api/login/login", route22);
app.use("/api/novel/addNovel", route23); app.use("/api/migrate/migrateData", route23);
app.use("/api/novel/batchDeleteNovel", route24); app.use("/api/modelSelect/getModelDetail", route24);
app.use("/api/novel/delNovel", route25); app.use("/api/modelSelect/getModelList", route25);
app.use("/api/novel/event/batchDeleteEvent", route26); app.use("/api/novel/addNovel", route26);
app.use("/api/novel/event/deletEvent", route27); app.use("/api/novel/batchDeleteNovel", route27);
app.use("/api/novel/event/generateEvents", route28); app.use("/api/novel/delNovel", route28);
app.use("/api/novel/event/getEvent", route29); app.use("/api/novel/event/batchDeleteEvent", route29);
app.use("/api/novel/getNovel", route30); app.use("/api/novel/event/deletEvent", route30);
app.use("/api/novel/getNovelEventState", route31); app.use("/api/novel/event/generateEvents", route31);
app.use("/api/novel/getNovelIndex", route32); app.use("/api/novel/event/getEvent", route32);
app.use("/api/novel/updateNovel", route33); app.use("/api/novel/getNovel", route33);
app.use("/api/other/deleteAllData", route34); app.use("/api/novel/getNovelEventState", route34);
app.use("/api/other/getCaptcha", route35); app.use("/api/novel/getNovelIndex", route35);
app.use("/api/production/assets/getAssetsData", route36); app.use("/api/novel/updateNovel", route36);
app.use("/api/production/editStoryboard/generateStoryboardImage", route37); app.use("/api/other/deleteAllData", route37);
app.use("/api/production/editStoryboard/getStoryboardFlow", route38); app.use("/api/other/getCaptcha", route38);
app.use("/api/production/editStoryboard/saveStoryboardFlow", route39); app.use("/api/production/assets/getAssetsData", route39);
app.use("/api/production/editStoryboard/updateStoryboardFlow", route40); app.use("/api/production/editStoryboard/generateStoryboardImage", route40);
app.use("/api/production/exportImage", route41); app.use("/api/production/editStoryboard/getStoryboardFlow", route41);
app.use("/api/production/getFlowData", route42); app.use("/api/production/editStoryboard/saveStoryboardFlow", route42);
app.use("/api/production/getProductionData", route43); app.use("/api/production/editStoryboard/updateStoryboardFlow", route43);
app.use("/api/production/getStoryboardData", route44); app.use("/api/production/exportImage", route44);
app.use("/api/production/saveFlowData", route45); app.use("/api/production/getFlowData", route45);
app.use("/api/production/workbench/confirmSelection", route46); app.use("/api/production/getProductionData", route46);
app.use("/api/production/workbench/delVideo", route47); app.use("/api/production/getStoryboardData", route47);
app.use("/api/production/workbench/generateVideo", route48); app.use("/api/production/saveFlowData", route48);
app.use("/api/production/workbench/getChatLines", route49); app.use("/api/production/workbench/confirmSelection", route49);
app.use("/api/production/workbench/getVideoModelDetail", route50); app.use("/api/production/workbench/delVideo", route50);
app.use("/api/production/workbench/videoPolling", route51); app.use("/api/production/workbench/generateVideo", route51);
app.use("/api/project/addProject", route52); app.use("/api/production/workbench/getChatLines", route52);
app.use("/api/project/delProject", route53); app.use("/api/production/workbench/getVideoModelDetail", route53);
app.use("/api/project/editProject", route54); app.use("/api/production/workbench/videoPolling", route54);
app.use("/api/project/getProject", route55); app.use("/api/project/addProject", route55);
app.use("/api/script/addScript", route56); app.use("/api/project/delProject", route56);
app.use("/api/script/delScript", route57); app.use("/api/project/editProject", route57);
app.use("/api/script/exportScript", route58); app.use("/api/project/getProject", route58);
app.use("/api/script/getScrptApi", route59); app.use("/api/script/addScript", route59);
app.use("/api/script/updateScript", route60); app.use("/api/script/delScript", route60);
app.use("/api/scriptAgent/getPlanData", route61); app.use("/api/script/exportScript", route61);
app.use("/api/scriptAgent/setPlanData", route62); app.use("/api/script/getScrptApi", route62);
app.use("/api/setting/agentDeploy/deployAgentModel", route63); app.use("/api/script/updateScript", route63);
app.use("/api/setting/agentDeploy/getAgentDeploy", route64); app.use("/api/scriptAgent/getPlanData", route64);
app.use("/api/setting/agentDeploy/oneClickToFillIn", route65); app.use("/api/scriptAgent/setPlanData", route65);
app.use("/api/setting/dbConfig/clearData", route66); app.use("/api/setting/agentDeploy/agentSetKey", route66);
app.use("/api/setting/getTextModel", route67); app.use("/api/setting/agentDeploy/deployAgentModel", route67);
app.use("/api/setting/loginConfig/getUser", route68); app.use("/api/setting/agentDeploy/getAgentDeploy", route68);
app.use("/api/setting/loginConfig/updateUserPwd", route69); app.use("/api/setting/dbConfig/clearData", route69);
app.use("/api/setting/memoryConfig/getMemory", route70); app.use("/api/setting/getTextModel", route70);
app.use("/api/setting/memoryConfig/sureMemory", route71); app.use("/api/setting/loginConfig/getUser", route71);
app.use("/api/setting/vendorConfig/addVendor", route72); app.use("/api/setting/loginConfig/updateUserPwd", route72);
app.use("/api/setting/vendorConfig/deleteVendor", route73); app.use("/api/setting/memoryConfig/getMemory", route73);
app.use("/api/setting/vendorConfig/getVendorList", route74); app.use("/api/setting/memoryConfig/sureMemory", route74);
app.use("/api/setting/vendorConfig/modelTest", route75); app.use("/api/setting/vendorConfig/addVendor", route75);
app.use("/api/setting/vendorConfig/updateVendor", route76); app.use("/api/setting/vendorConfig/deleteVendor", route76);
app.use("/api/task/getTaskApi", route77); app.use("/api/setting/vendorConfig/getVendorList", route77);
app.use("/api/task/getTaskCategories", route78); app.use("/api/setting/vendorConfig/modelTest", route78);
app.use("/api/task/taskDetails", route79); app.use("/api/setting/vendorConfig/updateVendor", route79);
app.use("/api/test/test", route80); app.use("/api/task/getTaskApi", route80);
app.use("/api/task/getTaskCategories", route81);
app.use("/api/task/taskDetails", route82);
app.use("/api/test/test", route83);
} }

30
src/routes/artStyle/addArtStyle.ts Normal file
View File

@ -0,0 +1,30 @@
import express from "express";
import u from "@/utils";
import { z } from "zod";
import { v4 as uuidv4 } from "uuid";
import { success } from "@/lib/responseFormat";
import { validateFields } from "@/middleware/middleware";
const router = express.Router();
export default router.post(
"/",
validateFields({
name: z.string(),
fileUrl: z.string(),
prompt: z.string(),
}),
async (req, res) => {
const { name, fileUrl, prompt } = req.body;
const imagePath = `/artStyle/${uuidv4()}.jpg`;
const matches = fileUrl.match(/^data:image\/\w+;base64,(.+)$/);
const realBase64 = matches ? matches[1] : fileUrl;
await u.oss.writeFile(imagePath, Buffer.from(realBase64, "base64"));
await u.db("o_artStyle").insert({
name,
fileUrl: imagePath,
label: name,
prompt,
});
res.status(200).send(success("艺术风格添加成功"));
},
);

34
src/routes/artStyle/editArtStyle.ts Normal file
View File

@ -0,0 +1,34 @@
import express from "express";
import u from "@/utils";
import { z } from "zod";
import { v4 as uuidv4 } from "uuid";
import { success } from "@/lib/responseFormat";
import { validateFields } from "@/middleware/middleware";
const router = express.Router();
export default router.post(
"/",
validateFields({
id: z.number(),
name: z.string(),
fileUrl: z.string(),
prompt: z.string(),
}),
async (req, res) => {
const { id, name, fileUrl, prompt } = req.body;
const imagePath = `/artStyle/${uuidv4()}.jpg`;
const matches = fileUrl.match(/^data:image\/\w+;base64,(.+)$/);
const realBase64 = matches ? matches[1] : fileUrl;
await u.oss.writeFile(imagePath, Buffer.from(realBase64, "base64"));
await u
.db("o_artStyle")
.update({
name,
fileUrl: imagePath,
label: name,
prompt,
})
.where("id", id);
res.status(200).send(success("艺术风格编辑成功"));
},
);

37
src/routes/artStyle/extractStylePrompt.ts Normal file
View File

@ -0,0 +1,37 @@
import express from "express";
import u from "@/utils";
import { z } from "zod";
import { success } from "@/lib/responseFormat";
import { validateFields } from "@/middleware/middleware";
const router = express.Router();
export default router.post(
"/",
validateFields({
images: z.array(z.string()),
}),
async (req, res) => {
const { images } = req.body;
try {
const resText = await u.Ai.Text("universalAgent").invoke({
system:
'请根据以下图片数据，提取出图片的画风提示词，用于生成图片时指定风格，要求简洁且具有艺术性,只需要画风提示词，不需要其他内容："比如：`(画风：2D动漫风格,2d animation style)`,`(画风：照片级真人超写实,photorealistic, lifelike, ultra detailed)`，`(画风：3D国创,Chinese 3D animation style)`等,如果图片风格无法描述，可以返回`无法描述`,多张图片时，只输出一个综合的画风提示词，要求包含所有图片的共同风格特征，输出格式必须严格按照示例中的格式，必须包含`画风`二字，且必须使用括号括起来，括号内必须包含中文和英文的画风描述，并用逗号分隔，英文部分需要翻译成地道的英文提示词',
messages: [
{
role: "user",
content: [
...images.map((image: string) => ({
type: "image" as const,
image,
})),
],
},
],
});
res.status(200).send(success(resText.text));
} catch (e) {
const err = u.error(e);
res.status(500).send({ message: err.message });
}
},
);

20
src/routes/artStyle/getArtStyle.ts
View File

@ -1,19 +1,15 @@
import express from "express"; import express from "express";
import u from "@/utils"; import u from "@/utils";
import { z } from "zod";
import { success } from "@/lib/responseFormat"; import { success } from "@/lib/responseFormat";
import { validateFields } from "@/middleware/middleware";
const router = express.Router(); const router = express.Router();
export default router.post( export default router.post("/", async (req, res) => {
"/", const list = await u.db("o_artStyle").select("*");
validateFields({ const data = await Promise.all(
name: z.string(), list.map(async (item: any) => {
const fileUrl = await u.oss.getFileUrl(item.fileUrl);
return { ...item, fileUrl };
}), }),
async (req, res) => {
const { name } = req.body;
const data = await u.db("o_artStyle").where("name", name).select("styles").first();
const styles = data?.styles ? JSON.parse(data.styles) : [];
res.status(200).send(success(styles));
},
); );
res.status(200).send(success(data));
});

8
src/routes/assets/getAssetsApi.ts
View File

@ -28,7 +28,7 @@ export default router.post(
query = query.andWhere("name", "like", `%${name}%`); query = query.andWhere("name", "like", `%${name}%`);
} }
// 分页查询 // 分页查询
const parentAssets = await query.where("o_assets.assetId", null).offset(offset).limit(limit); const parentAssets = await query.where("o_assets.assetsId", null).offset(offset).limit(limit);
// 获取所有子资产供关联使用 // 获取所有子资产供关联使用
let childQuery = u let childQuery = u
@ -37,7 +37,7 @@ export default router.post(
.select("o_assets.*", "o_image.filePath") .select("o_assets.*", "o_image.filePath")
.where("o_assets.projectId", projectId) .where("o_assets.projectId", projectId)
.andWhere("o_assets.type", type) .andWhere("o_assets.type", type)
.whereNotNull("o_assets.assetId"); .whereNotNull("o_assets.assetsId");
if (name) { if (name) {
childQuery = childQuery.andWhere("o_assets.name", "like", `%${name}%`); childQuery = childQuery.andWhere("o_assets.name", "like", `%${name}%`);
} }
@ -47,7 +47,7 @@ export default router.post(
const result = await Promise.all( const result = await Promise.all(
parentAssets.map(async (parent) => ({ parentAssets.map(async (parent) => ({
...parent, ...parent,
sonAssets: childAssets.filter((child) => child.assetId === parent.id), sonAssets: childAssets.filter((child) => child.assetsId === parent.id),
src: parent.filePath && (await u.oss.getFileUrl(parent.filePath!)), src: parent.filePath && (await u.oss.getFileUrl(parent.filePath!)),
})), })),
); );
@ -57,7 +57,7 @@ export default router.post(
.db("o_assets") .db("o_assets")
.where("projectId", projectId) .where("projectId", projectId)
.andWhere("type", type) .andWhere("type", type)
.andWhere("assetId", null) .andWhere("assetsId", null)
.andWhere((qb) => { .andWhere((qb) => {
if (name) { if (name) {
qb.andWhere("name", "like", `%${name}%`); qb.andWhere("name", "like", `%${name}%`);

2
src/routes/production/assets/getAssetsData.ts
View File

@ -13,7 +13,7 @@ export default router.post(
}), }),
async (req, res) => { async (req, res) => {
const { projectId } = req.body; const { projectId } = req.body;
const parentAssetsData = await u.db("o_assets").where("projectId", projectId).whereNotNull("assetId"); const parentAssetsData = await u.db("o_assets").where("projectId", projectId).whereNotNull("assetsId");
const parentIds = parentAssetsData.map((i) => i.id); const parentIds = parentAssetsData.map((i) => i.id);
const sonAssetsData = await u.db("o_assets").whereIn("assetsId", parentIds); const sonAssetsData = await u.db("o_assets").whereIn("assetsId", parentIds);
const sonAssetsMap: Record<number, o_assets[]> = {}; const sonAssetsMap: Record<number, o_assets[]> = {};

4
src/routes/production/getFlowData.ts
View File

@ -33,7 +33,7 @@ export default router.post(
.leftJoin("o_image", "o_assets.imageId", "o_image.id") .leftJoin("o_image", "o_assets.imageId", "o_image.id")
.select("o_assets.*", "o_image.filePath") .select("o_assets.*", "o_image.filePath")
.where("o_assets.projectId", projectId) .where("o_assets.projectId", projectId)
.whereNotNull("o_assets.assetId"); .whereNotNull("o_assets.assetsId");
if (!sqlData) { if (!sqlData) {
const flowData: FlowData = { const flowData: FlowData = {
@ -47,7 +47,7 @@ export default router.post(
src: item.filePath && (await u.oss.getFileUrl(item.filePath!)), src: item.filePath && (await u.oss.getFileUrl(item.filePath!)),
derive: await Promise.all( derive: await Promise.all(
childAssetsData childAssetsData
.filter((child) => child.assetId === item.id) .filter((child) => child.assetsId === item.id)
.map(async (child) => ({ .map(async (child) => ({
id: child.id, id: child.id,
assetsId: item.id, assetsId: item.id,

1
src/routes/production/workbench/generateVideo.ts
View File

@ -29,7 +29,6 @@ export default router.post(
}), }),
async (req, res) => { async (req, res) => {
const { scriptId, projectId, storyboardId, prompt, data, model, duration, resolution, audio, mode } = req.body; const { scriptId, projectId, storyboardId, prompt, data, model, duration, resolution, audio, mode } = req.body;
console.log("%c Line:32 🥤 req.body", "background:#465975", req.body);
const videoPath = `/${projectId}/video/${uuidv4()}.mp4`; //视频保存路径 const videoPath = `/${projectId}/video/${uuidv4()}.mp4`; //视频保存路径
//新增 //新增

3
src/routes/script/getScrptApi.ts
View File

@ -28,7 +28,6 @@ export default router.post(
data.map((i) => i.id), data.map((i) => i.id),
) )
.select("o_assets.id", "o_assets.name", "o_scriptAssets.scriptId"); .select("o_assets.id", "o_assets.name", "o_scriptAssets.scriptId");
console.log("%c Line:23 🍷 assetsData", "background:#ffdd4d", assetsData);
const scriptAssetsMap: Record<number, { id: number; name: string }[]> = {}; const scriptAssetsMap: Record<number, { id: number; name: string }[]> = {};
assetsData.forEach((i) => { assetsData.forEach((i) => {
if (scriptAssetsMap[i.scriptId]) { if (scriptAssetsMap[i.scriptId]) {
@ -42,7 +41,7 @@ export default router.post(
name: i.name, name: i.name,
content: i.content, content: i.content,
createTime: i.createTime, createTime: i.createTime,
assets: scriptAssetsMap[i.id!] || [], relatedAssets: scriptAssetsMap[i.id!] || [],
})); }));
res.status(200).send(success(returnData)); res.status(200).send(success(returnData));
}, },

22
src/routes/setting/agentDeploy/agentSetKey.ts Normal file
View File

@ -0,0 +1,22 @@
import express from "express";
import { success } from "@/lib/responseFormat";
import u from "@/utils";
import { z } from "zod";
import { validateFields } from "@/middleware/middleware";
const router = express.Router();
export default router.post(
"/",
validateFields({
id: z.array(z.number()),
}),
async (req, res) => {
const { id } = req.body;
await u.db("o_agentDeploy").whereIn("id", id).where("disabled", "<>", 1).update({
model: "gpt-4.1",
modelName: "1:gpt-4.1",
vendorId: 1,
});
res.status(200).send(success("配置成功"));
},
);

25
src/routes/setting/agentDeploy/deployAgentModel.ts
View File

@ -5,18 +5,15 @@ import { z } from "zod";
import { validateFields } from "@/middleware/middleware"; import { validateFields } from "@/middleware/middleware";
const router = express.Router(); const router = express.Router();
export default router.post( export default router.post("/", validateFields({
"/", id: z.number(),
validateFields({ name: z.string(),
id: z.array(z.number()), model: z.string(),
}), modelName: z.string(),
async (req, res) => { vendorId: z.number().nullable(),
const { id } = req.body; desc: z.string(),
await u.db("o_agentDeploy").whereIn("id", id).where("disabled", "<>", 1).update({ }), async (req, res) => {
model: "gpt-4.1", const { id, name, model, modelName, vendorId, desc } = req.body;
modelName: "1:gpt-4.1", await u.db("o_agentDeploy").where({ id }).update({ id, name, model, modelName, vendorId, desc });
vendorId: 1,
});
res.status(200).send(success("配置成功")); res.status(200).send(success("配置成功"));
}, });
);

9
src/routes/setting/agentDeploy/oneClickToFillIn.ts
View File

@ -1,9 +0,0 @@
import express from "express";
import { success } from "@/lib/responseFormat";
import u from "@/utils";
const router = express.Router();
export default router.post("/", async (req, res) => {
const data = await u.db("o_agentDeploy").leftJoin("o_vendorConfig", "o_vendorConfig.id", "o_agentDeploy.vendorId").select("o_agentDeploy.*", "o_vendorConfig.icon");
res.status(200).send(success(data));
});

9
src/types/database.d.ts vendored
View File

@ -1,4 +1,4 @@
// @db-hash f6a9a8164252ce954394431079615459 // @db-hash 47c0e014bdbd44b60c4ebc95f4d99e0e
//该文件由脚本自动生成，请勿手动修改 //该文件由脚本自动生成，请勿手动修改
export interface memories { export interface memories {
@ -32,11 +32,14 @@ export interface o_agentWorkData {
'updateTime'?: number | null; 'updateTime'?: number | null;
} }
export interface o_artStyle { export interface o_artStyle {
'fileUrl'?: string | null;
'id'?: number; 'id'?: number;
'label'?: string | null;
'name'?: string | null; 'name'?: string | null;
'styles'?: string | null; 'prompt'?: string | null;
} }
export interface o_assets { export interface o_assets {
'assetsId'?: number | null;
'describe'?: string | null; 'describe'?: string | null;
'id'?: number; 'id'?: number;
'imageId'?: number | null; 'imageId'?: number | null;
@ -45,9 +48,7 @@ export interface o_assets {
'prompt'?: string | null; 'prompt'?: string | null;
'remark'?: string | null; 'remark'?: string | null;
'scriptId'?: number | null; 'scriptId'?: number | null;
'sonId'?: number | null;
'startTime'?: number | null; 'startTime'?: number | null;
'state'?: string | null;
'type'?: string | null; 'type'?: string | null;
} }
export interface o_assets2Storyboard { export interface o_assets2Storyboard {

2
src/utils/cleanNovel.ts
View File

@ -41,12 +41,14 @@ class CleanNovel {
], ],
tools: skill.tools, tools: skill.tools,
}); });
console.log("%c Line:35 🍆 resData", "background:#fca650", resData);
const preData = resData.text; const preData = resData.text;
this.emitter.emit("item", { id: novel.id, event: preData }); this.emitter.emit("item", { id: novel.id, event: preData });
totalEvent.push({ id: novel.id!, event: preData }); totalEvent.push({ id: novel.id!, event: preData });
} catch (e) { } catch (e) {
console.log("%c Line:51 🍩 e", "background:#93c0a4", e);
this.emitter.emit("item", { id: novel.id, event: null, errorReason: u.error(e).message }); this.emitter.emit("item", { id: novel.id, event: null, errorReason: u.error(e).message });
} }
} }