Function main

examples/subagent-async/main.go:16–279

()

)

func main() {
	fmt.Println("=== SubAgent 异步执行示例 ===")
	fmt.Println()

	// 1. 创建 Backend
	backend := backends.NewStateBackend()
	fmt.Println("✓ 创建 StateBackend")

	// 2. 定义子代理规格
	specs := []middleware.SubAgentSpec{
		{
			Name:        "researcher",
			Description: "深度研究和分析专家",
			Prompt:      "你是一个专注于深度研究的 AI。仔细分析问题,提供详细的研究报告。",
		},
		{
			Name:        "data-analyst",
			Description: "数据分析专家",
			Prompt:      "你是一个数据分析专家。处理和分析大量数据,生成洞察报告。",
		},
	}
	fmt.Printf("✓ 定义了 %d 个子代理规格\n", len(specs))

	// 3. 创建子代理工厂（模拟长时间运行的任务）
	factory := func(ctx context.Context, spec middleware.SubAgentSpec) (middleware.SubAgent, error) {
		execFn := func(ctx context.Context, description string, parentContext map[string]any) (string, error) {
			// 模拟长时间运行的任务
			fmt.Printf("[%s] 开始执行任务: %s\n", spec.Name, description)
			time.Sleep(5 * time.Second) // 模拟5秒的处理时间

			result := fmt.Sprintf(`[%s] 任务完成报告
任务描述: %s
执行时间: 5秒
系统提示: %s
上下文: %v

分析结果:
1. 任务已成功完成
2. 处理了大量数据
3. 生成了详细报告`, spec.Name, description, spec.Prompt, parentContext)

			fmt.Printf("[%s] 任务完成\n", spec.Name)
			return result, nil
		}
		return middleware.NewSimpleSubAgent(spec.Name, spec.Prompt, execFn), nil
	}
	fmt.Println("✓ 创建子代理工厂")

	// 4. 创建 SubAgentMiddleware（启用异步执行）
	subagentMiddleware, err := middleware.NewSubAgentMiddleware(&middleware.SubAgentMiddlewareConfig{
		Specs:                  specs,
		Factory:                factory,
		EnableParallel:         true,
		EnableAsync:            true,  // 启用异步执行
		EnableProcessIsolation: false, // 使用 goroutine 模式（演示用）
		DefaultTimeout:         30 * time.Second,
	})
	if err != nil {