Function testWorkflowAgent

examples/workflow_complete/main.go:248–292

测试 4: WorkflowAgent

(ctx context.Context)

// 测试 4: WorkflowAgent
func testWorkflowAgent(ctx context.Context) {
	// 创建一个简单的 workflow
	wf := workflow.New("AgentWorkflow")

	wf.AddStep(workflow.NewFunctionStep("process", func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
		return &workflow.StepOutput{
			Content:  fmt.Sprintf("处理完成: %v", input.Input),
			Metadata: make(map[string]any),
		}, nil
	}))

	// 创建 WorkflowAgent
	agent := workflow.NewWorkflowAgent("gpt-4", "", true, 5)
	agent.AttachWorkflow(wf)

	fmt.Println("  测试同步执行:")
	result, err := agent.Run(ctx, "测试查询")
	if err != nil {
		fmt.Printf("    ❌ 错误: %v\n", err)
	} else {
		fmt.Printf("    ✅ 结果: %v\n", result)
	}

	// 测试流式执行
	fmt.Println("  测试流式执行:")
	eventCount := 0
	for event := range agent.RunStream(ctx, "流式测试") {
		eventCount++
		if event.Type == workflow.AgentEventComplete {
			fmt.Printf("    ✅ 流式完成, 收到 %d 个事件\n", eventCount)
		}
		if event.Error != nil {
			fmt.Printf("    ❌ 错误: %v\n", event.Error)
		}
	}

	// 测试 AgenticExecute
	fmt.Println("  测试 AgenticExecute:")
	result2, err := wf.AgenticExecute(ctx, agent, "Agentic 查询")
	if err != nil {
		fmt.Printf("    ❌ 错误: %v\n", err)
	} else {
		fmt.Printf("    ✅ 结果: %v\n", result2)
	}
}