super/yaoyaoai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

11

Browse Source
This commit is contained in:
fengchuanhn@gmail.com
2026-05-18 08:52:36 +08:00
parent caaa898d90
commit c559cb1581
5 changed files with 124 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
scripts/nodejs/douyin_pipeline.js
View File

@@ -49,22 +49,26 @@ async function rewriteWithAI(content, modelConfig, rewriteConfig, customPrompt)
rewriteConfig || {},
customPrompt
);
log("info", "pipeline.rewritePrompt", { prompt: prompt });
log("info", "pipeline.apiUrl", { apiUrl: modelConfig.apiUrl });
log("info", "pipeline.apiKey", { apiKey: modelConfig.apiKey });
log("info", "pipeline.apiModelId", { apiModelId: modelConfig.apiModelId });
const result = await pipelineNative.openaiChat({
apiUrl: modelConfig.apiUrl,
apiKey: modelConfig.apiKey,
model: modelConfig.modelId,
model: modelConfig.apiModelId,
messages: [
{ role: "system", content: "你是专业的短视频文案改写助手。" },
{ role: "system", content: "你是专业的短视频文案改写助手。请把用户提供的文案改写,要求保持原文的风格和主题，但使用不同的表达方式。" },
{ role: "user", content: prompt },
],
temperature:
typeof modelConfig.temperature === "number" ? modelConfig.temperature : 0.8,
max_tokens: modelConfig.maxTokens || 2000,
temperature: 0.8,
max_tokens: 2e3,
});
if (!result?.success) {
log("error", "rewrite.failed", { error: result?.error });
return null;
}
log("info", "pipeline.rewriteOk", { result: result.content });
return result.content;
}