Function main

examples/subagent/main.go:14–169

()

)

func main() {
	fmt.Println("=== SubAgent Middleware 示例 ===")

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

	// 2. 定义子代理规格
	specs := []middleware.SubAgentSpec{
		{
			Name:        "researcher",
			Description: "深度研究和分析专家",
			Prompt:      "你是一个专注于深度研究的 AI。仔细分析问题,提供详细的研究报告。",
		},
		{
			Name:        "coder",
			Description: "代码编写和重构专家",
			Prompt:      "你是一个专业的程序员。编写高质量、可维护的代码。",
		},
		{
			Name:        "reviewer",
			Description: "代码审查和质量检查专家",
			Prompt:      "你是一个严格的代码审查员。识别潜在问题,提出改进建议。",
		},
	}
	fmt.Printf("✓ 定义了 %d 个子代理规格\n", len(specs))

	// 3. 创建子代理工厂
	factory := func(ctx context.Context, spec middleware.SubAgentSpec) (middleware.SubAgent, error) {
		// 这里使用 SimpleSubAgent 作为演示
		// 实际应用中应该创建真正的 Agent 实例
		execFn := func(ctx context.Context, description string, parentContext map[string]any) (string, error) {
			return fmt.Sprintf("[%s] 已执行任务: %s\n系统提示: %s\n上下文: %v",
				spec.Name, description, spec.Prompt, parentContext), 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,
	})
	if err != nil {
		log.Fatalf("创建 SubAgentMiddleware 失败: %v", err)
	}
	fmt.Println("✓ 创建 SubAgentMiddleware")

	// 5. 创建 FilesystemMiddleware
	fsMiddleware := middleware.NewFilesystemMiddleware(&middleware.FilesystemMiddlewareConfig{
		Backend:        backend,
		EnableEviction: true,
	})
	fmt.Println("✓ 创建 FilesystemMiddleware")

	// 6. 创建 Middleware Stack