wonder/gen2d
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(pipeline): PromptOptimizer 集成到生成管线，新增优化 API

Browse Source 
- PipelineInput 新增 Tags/UserNote 字段
- 管线重构为: START → PromptOptimizer → AssetGenerator → QualitySupervisor → FormatAdapter
- promptOptimizerNode: 有标签调用 PromptAgent，无标签补技术参数，合并风格与重试信息
- PromptBuilder 移除，提示词构建逻辑并入 PromptOptimizer
- 新增 POST /api/v1/prompt/optimize 接口
- main.go 注入 LLM/ImageGen 配置，注册 prompt 路由
- inference.go 新增 InitImageGenConfig 注入
This commit is contained in:
Gmaker689
2026-05-24 20:45:18 +08:00
parent 788cd021ef
commit 9ca1e65221
7 changed files with 281 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/internal/service/pipeline.go
+17 -18
View File
@@ -8,18 +8,18 @@ import (
)
const (
nodePromptBuilder = "prompt_builder"
nodePromptOptimizer = "prompt_optimizer"
nodeAssetGenerator = "asset_generator"
nodeQualitySupervisor = "quality_supervisor"
nodeFormatAdapter = "format_adapter"
)
// NewGenerateGraph 创建四阶段生成管线 Graph。
// NewGenerateGraph 创建生成管线 Graph（PromptOptimizer → AssetGenerator → QualitySupervisor → FormatAdapter）。
//
// START → PromptBuilder → AssetGenerator → QualitySupervisor
// ├── pass → FormatAdapter → END
// └── fail, retry<3 → PromptBuilder
// └── fail, retry>=3 → FormatAdapter (降级)
// START → PromptOptimizer → AssetGenerator → QualitySupervisor
// ├── pass → FormatAdapter → END
// └── fail, retry<3 → PromptOptimizer
// └── fail, retry>=3 → FormatAdapter (降级)
func NewGenerateGraph() (*compose.Graph[PipelineInput, PipelineOutput], error) {
g := compose.NewGraph[PipelineInput, PipelineOutput](
compose.WithGenLocalState(func(ctx context.Context) *PipelineState {
@@ -27,12 +27,11 @@ func NewGenerateGraph() (*compose.Graph[PipelineInput, PipelineOutput], error) {
}),
)
// 添加节点
if err := g.AddLambdaNode(nodePromptBuilder, promptBuilderNode,
compose.WithStatePreHandler(promptBuilderPreHandler),
compose.WithStatePostHandler(promptBuilderPostHandler),
if err := g.AddLambdaNode(nodePromptOptimizer, promptOptimizerNode,
compose.WithStatePreHandler(promptOptimizerPreHandler),
compose.WithStatePostHandler(promptOptimizerPostHandler),
); err != nil {
return nil, fmt.Errorf("add %s node: %w", nodePromptBuilder, err)
return nil, fmt.Errorf("add %s node: %w", nodePromptOptimizer, err)
}
if err := g.AddLambdaNode(nodeAssetGenerator, assetGeneratorNode,
@@ -49,12 +48,12 @@ func NewGenerateGraph() (*compose.Graph[PipelineInput, PipelineOutput], error) {
return nil, fmt.Errorf("add %s node: %w", nodeFormatAdapter, err)
}
// 连线：正常路径
if err := g.AddEdge(compose.START, nodePromptBuilder); err != nil {
return nil, fmt.Errorf("add edge START->%s: %w", nodePromptBuilder, err)
// 连线：START → PromptOptimizer → AssetGenerator → Supervisor
if err := g.AddEdge(compose.START, nodePromptOptimizer); err != nil {
return nil, fmt.Errorf("add edge START->%s: %w", nodePromptOptimizer, err)
}
if err := g.AddEdge(nodePromptBuilder, nodeAssetGenerator); err != nil {
return nil, fmt.Errorf("add edge %s->%s: %w", nodePromptBuilder, nodeAssetGenerator, err)
if err := g.AddEdge(nodePromptOptimizer, nodeAssetGenerator); err != nil {
return nil, fmt.Errorf("add edge %s->%s: %w", nodePromptOptimizer, nodeAssetGenerator, err)
}
if err := g.AddEdge(nodeAssetGenerator, nodeQualitySupervisor); err != nil {
return nil, fmt.Errorf("add edge %s->%s: %w", nodeAssetGenerator, nodeQualitySupervisor, err)
@@ -73,7 +72,7 @@ func NewGenerateGraph() (*compose.Graph[PipelineInput, PipelineOutput], error) {
})
return next, nil
},
map[string]bool{nodePromptBuilder: true, nodeFormatAdapter: true},
map[string]bool{nodePromptOptimizer: true, nodeFormatAdapter: true},
)); err != nil {
return nil, fmt.Errorf("add branch at %s: %w", nodeQualitySupervisor, err)
}
@@ -81,7 +80,7 @@ func NewGenerateGraph() (*compose.Graph[PipelineInput, PipelineOutput], error) {
return g, nil
}
// RunPipeline 编译并执行生成管线
// RunPipeline 编译并执行生成管线。
func RunPipeline(ctx context.Context, in PipelineInput) (*PipelineOutput, error) {
g, err := NewGenerateGraph()
if err != nil {