Function loopExample

examples/workflow-agents/main.go:115–163

loopExample 循环优化示例

(ctx context.Context)

// loopExample 循环优化示例
func loopExample(ctx context.Context) {
	// 创建优化流程的子 Agent
	agents := []workflow.Agent{
		NewMockAgent("Critic", "评估当前方案"),
		NewMockAgent("Improver", "提出改进建议"),
	}

	// 创建 LoopAgent（最多 3 次迭代）
	loop, err := workflow.NewLoopAgent(workflow.LoopConfig{
		Name:          "OptimizationLoop",
		SubAgents:     agents,
		MaxIterations: 3,
		StopCondition: func(event *session.Event) bool {
			// 如果评分达到 90 分，提前停止
			if score, ok := event.Metadata["quality_score"].(int); ok {
				return score >= 90
			}
			return false
		},
	})
	if err != nil {
		log.Fatal(err)
	}

	// 执行
	fmt.Println("开始循环优化:")
	iteration := 0
	reader := loop.Execute(ctx, "优化代码质量")
	for {
		event, err := reader.Recv()
		if err != nil {
			if errors.Is(err, io.EOF) {
				break
			}
			log.Printf("错误: %v", err)
			break
		}

		// 追踪迭代次数
		if iterNum, ok := event.Metadata["loop_iteration"].(uint); ok {
			if uint(iteration) != iterNum {
				iteration = int(iterNum)
				fmt.Printf("\n--- 迭代 %d ---\n", iteration)
			}
		}

		printEvent(event)
	}
}

// nestedExample 嵌套工作流示例
func nestedExample(ctx context.Context) {