Function NewSubAgentMiddleware

pkg/middleware/subagent.go:84–143

NewSubAgentMiddleware 创建子代理中间件

(config *SubAgentMiddlewareConfig)

// NewSubAgentMiddleware 创建子代理中间件
func NewSubAgentMiddleware(config *SubAgentMiddlewareConfig) (*SubAgentMiddleware, error) {
	// 设置默认超时
	defaultTimeout := config.DefaultTimeout
	if defaultTimeout == 0 {
		defaultTimeout = 1 * time.Hour
	}

	m := &SubAgentMiddleware{
		BaseMiddleware:         NewBaseMiddleware("subagent", 200),
		agents:                 make(map[string]SubAgent),
		factory:                config.Factory,
		enableParallel:         config.EnableParallel,
		enableAsync:            config.EnableAsync,
		enableProcessIsolation: config.EnableProcessIsolation,
		defaultTimeout:         defaultTimeout,
	}

	// 创建或使用提供的管理器
	switch {
	case config.Manager != nil:
		m.manager = config.Manager
	case config.EnableProcessIsolation:
		m.manager = builtin.NewFileSubagentManager()
		saLog.Info(context.Background(), "using FileSubagentManager", nil)
	case config.EnableAsync:
		m.manager = newGoroutineSubagentManager(m)
		saLog.Info(context.Background(), "using in-memory async manager", nil)
	}

	// 默认启用 general-purpose 子代理
	specs := config.Specs
	if config.EnableGeneralPurpose || (len(specs) == 0 && !config.EnableParallel) {
		// 添加通用子代理规格
		generalPurposeSpec := SubAgentSpec{
			Name:        "general-purpose",
			Description: "通用子代理,用于执行复杂、多步骤的隔离任务",
			Prompt: `你是一个通用的 AI 助手,专注于执行复杂的、多步骤的任务。
你有完整的工具集,可以独立完成被委托的任务。
请仔细分析任务需求,制定计划并逐步执行。`,
			InheritMiddlewares: true, // 继承父代理的中间件
		}
		specs = append([]SubAgentSpec{generalPurposeSpec}, specs...)
	}

	// 初始化子代理
	if config.Factory != nil {
		for _, spec := range specs {
			agent, err := config.Factory(context.Background(), spec)
			if err != nil {
				saLog.Error(context.Background(), "failed to create subagent", map[string]any{"name": spec.Name, "error": err.Error()})
				continue
			}
			m.agents[spec.Name] = agent
			saLog.Info(context.Background(), "created subagent", map[string]any{"name": spec.Name})
		}
	}

	saLog.Info(context.Background(), "initialized", map[string]any{"subagents": len(m.agents)})