Function main

examples/workflow_router/main.go:14–192

()

)

func main() {
	fmt.Println("=== Aster Router 流式执行示例 ===")

	ctx := context.Background()

	// 创建一些步骤
	step1 := workflow.NewFunctionStep("analyze", func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
		fmt.Println("  🔍 Analyzing input...")
		return &workflow.StepOutput{
			Content:  map[string]any{"analysis": "complex", "priority": "high"},
			Metadata: make(map[string]any),
		}, nil
	})

	step2 := workflow.NewFunctionStep("process_complex", func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
		fmt.Println("  ⚙️  Processing complex case...")
		return &workflow.StepOutput{
			Content:  "Processed with advanced algorithm",
			Metadata: make(map[string]any),
		}, nil
	})

	step3 := workflow.NewFunctionStep("process_simple", func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
		fmt.Println("  ⚡ Processing simple case...")
		return &workflow.StepOutput{
			Content:  "Processed with basic algorithm",
			Metadata: make(map[string]any),
		}, nil
	})

	step4 := workflow.NewFunctionStep("finalize", func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
		fmt.Println("  ✅ Finalizing...")
		return &workflow.StepOutput{
			Content:  fmt.Sprintf("Final result: %v", input.PreviousStepContent),
			Metadata: make(map[string]any),
		}, nil
	})

	// 创建链式路由器 - 根据分析结果选择不同的处理链
	router := workflow.ChainRouter("smart_processor",
		func(input *workflow.StepInput) []string {
			// 根据前一步的分析结果决定执行路径
			if input.PreviousStepContent != nil {
				if analysis, ok := input.PreviousStepContent.(map[string]any); ok {
					if analysis["analysis"] == "complex" {
						fmt.Println("\n📍 Router 选择: complex 路径 (2步)")
						return []string{"process_complex", "finalize"}
					}
				}
			}
			fmt.Println("\n📍 Router 选择: simple 路径 (2步)")
			return []string{"process_simple", "finalize"}
		},
		map[string]workflow.Step{
			"process_complex": step2,
			"process_simple":  step3,
			"finalize":        step4,
		},