Function demoWorkflowAgent

examples/comprehensive/main.go:172–222

演示 3: WorkflowAgent 智能编排

(ctx context.Context)

// 演示 3: WorkflowAgent 智能编排
func demoWorkflowAgent(ctx context.Context) {
	// 创建一个简单的分析 workflow
	analysisWf := workflow.New("Analysis")

	analysisWf.AddStep(workflow.NewFunctionStep("analyze",
		func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
			query := input.Input.(string)
			fmt.Printf("  🔍 分析查询: %s\n", query)

			// 模拟分析
			result := map[string]any{
				"query":   query,
				"result":  "分析完成",
				"metrics": map[string]int{"items": 42, "quality": 95},
			}

			return &workflow.StepOutput{
				Content:  result,
				Metadata: make(map[string]any),
			}, nil
		},
	))

	// 创建 WorkflowAgent
	wfAgent := workflow.NewWorkflowAgent(
		"gpt-4",
		"你是一个数据分析助手。如果用户查询已在历史中，直接回答；否则运行 workflow。",
		true, // 启用历史
		5,    // 保留5次历史
	)

	wfAgent.AttachWorkflow(analysisWf)

	// 第一次查询 - 会运行 workflow
	fmt.Println("\n  第一次查询:")
	result1, err := wfAgent.Run(ctx, "分析销售数据")
	if err != nil {
		log.Printf("  ❌ 错误: %v", err)
	} else {
		fmt.Printf("  ✅ 结果: %s\n", result1)
	}

	// 查看历史
	history := wfAgent.GetWorkflowHistory()
	if len(history) > 0 {
		fmt.Printf("\n  📊 历史记录: %d 条\n", len(history))
		for i, item := range history {
			fmt.Printf("    %d. 输入: %v -> 状态: %s\n", i+1, item.Input, item.Status)
		}
	}
}