fix: 修复管线context取消、添加FIFO任务队列、移除重复提示词优化
三个修复: 1. runPipelineBg 改用 context.Background(),避免 HTTP 响应返回后 Gin 取消 request context 导致后台管线静默失败 2. 新增 TaskQueue FIFO 串行队列,任务提交后进入 pending 状态排队, 按提交顺序逐个执行,前端轮询显示排队中 3. promptOptimizerNode 移除 RunPromptAgent 调用,提示词优化仅由 前端在提交前通过 /api/v1/prompt/optimize 执行一次,管线内只做 风格合并和技术参数追加
This commit is contained in:
@@ -14,10 +14,10 @@ import (
|
||||
"github.com/cloudwego/eino/compose"
|
||||
)
|
||||
|
||||
// promptOptimizerNode 节点:调用 PromptAgent 生成规范提示词,合并风格与重试信息。
|
||||
// 输入 PipelineInput,输出最终提示词字符串(直接供 AssetGenerator 消费)。
|
||||
// promptOptimizerNode 节点:合并风格描述、重试信息与技术参数,输出最终提示词。
|
||||
// 提示词优化已由前端在提交前完成,管线内不再重复调用 PromptAgent。
|
||||
var promptOptimizerNode = compose.InvokableLambda(func(ctx context.Context, in PipelineInput) (string, error) {
|
||||
// 合并风格描述,注入原始 Prompt 中
|
||||
// 合并风格描述
|
||||
styleDesc := buildStyleDescription(in.ProjectStyle, in.TaskStyle)
|
||||
if styleDesc != "" {
|
||||
if in.Prompt != "" {
|
||||
@@ -36,26 +36,11 @@ var promptOptimizerNode = compose.InvokableLambda(func(ctx context.Context, in P
|
||||
}
|
||||
}
|
||||
|
||||
if len(in.Tags) == 0 && in.Prompt == "" {
|
||||
return "", fmt.Errorf("pipeline: Prompt and Tags are both empty")
|
||||
if in.Prompt == "" {
|
||||
return "", fmt.Errorf("pipeline: prompt is empty")
|
||||
}
|
||||
|
||||
// 有标签时调用 PromptAgent 优化提示词
|
||||
if len(in.Tags) > 0 {
|
||||
agentIn := PromptAgentInput{
|
||||
Tags: in.Tags,
|
||||
AssetType: in.AssetType,
|
||||
Prompt: in.Prompt,
|
||||
UserNote: in.UserNote,
|
||||
}
|
||||
output, err := RunPromptAgent(ctx, agentIn)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("prompt agent: %w", err)
|
||||
}
|
||||
return output.Prompt, nil
|
||||
}
|
||||
|
||||
// 无标签时直接使用原始 Prompt,补上技术参数段
|
||||
// 追加技术参数段,不再调用 PromptAgent
|
||||
return appendTechNotes(in.Prompt, in.AssetType, in.Params), nil
|
||||
})
|
||||
|
||||
|
||||
Reference in New Issue
Block a user