子代理系统

// AgentDefinition: 代理定义接口
interface AgentDefinition {
  name: string;
  prompt: string;
  tools?: string[];        // 允许的工具列表
  maxTurns?: number;       // 最大交互轮次
  mcpServers?: McpConfig;  // MCP 配置
  model?: string;          // 使用的模型
}

// 内置代理定义示例
const exploreAgent: AgentDefinition = {
  name: 'exploreAgent',
  prompt: '你是一个代码库探索专家。使用只读工具快速理解'
    + '代码结构、找到关键文件和理解架构设计。'
    + '不要修改任何文件。',
  tools: ['Read', 'Glob', 'Grep', 'LSP', 'Bash'],
  maxTurns: 15
};

const planAgent: AgentDefinition = {
  name: 'planAgent',
  prompt: '你是一个实现规划专家。分析需求后制定详细的'
    + '实现方案，包括文件修改计划、测试策略和风险评估。',
  tools: ['Read', 'Glob', 'Grep', 'LSP'],
  maxTurns: 10
};

// AgentTool: 子代理生成工具
const AgentTool = {
  name: 'Agent',
  description: '创建子代理来执行复杂的多步骤任务',
  inputSchema: {
    type: 'object',
    properties: {
      agent: { type: 'string', description: '代理名称' },
      task: { type: 'string', description: '任务描述' }
    },
    required: ['agent', 'task']
  },
  execute: async (input: AgentToolInput) => {
    const definition = resolveAgent(input.agent);
    return runAgent(definition, input.task);
  }
};

// runAgent.ts: 子代理核心执行流程
async function* runAgent(
  definition: AgentDefinition,
  task: string
): AsyncGenerator<AgentProgress, AgentResult> {
  // 1. 创建子代理上下文
  const context = createSubagentContext({
    parentContext: getCurrentContext(),
    agentName: definition.name
  });

  // 2. 构建代理提示词
  const systemPrompt = buildAgentPrompt(
    definition.prompt,
    context
  );

  // 3. 过滤工具集
  const tools = filterTools(
    getAllTools(),
    definition.tools  // 只保留允许的工具
  );

  // 4. 初始化消息
  const messages: Message[] = [
    { role: 'user', content: task }
  ];

  // 5. 代理执行循环
  let turns = 0;
  while (turns < (definition.maxTurns || 20)) {
    // 递归调用 query()
    const response = await query({
      systemPrompt,
      messages,
      tools,
      model: definition.model
    });

    // 产出进度更新
    yield {
      type: 'progress',
      agent: definition.name,
      turn: turns,
      content: response.content
    };

    // 检查是否完成
    if (!response.toolCalls || response.toolCalls.length === 0) {
      break;
    }

    // 执行工具调用并追加消息
    for (const call of response.toolCalls) {
      const result = await executeTool(call, context);
      messages.push(
        { role: 'assistant', content: response.content },
        { role: 'tool', content: result }
      );
    }
    turns++;
  }

  // 6. 返回执行摘要
  return {
    agent: definition.name,
    summary: messages[messages.length - 1].content,
    turnsUsed: turns
  };
}

// forkSubagent.ts: 基于进程分叉的子代理
async function forkSubagent(
  definition: AgentDefinition,
  task: string
): Promise<AgentResult> {
  // 分叉当前进程，共享 KV 缓存
  const child = fork(process.argv[1], {
    env: {
      ...process.env,
      AGENT_MODE: 'subagent',
      AGENT_NAME: definition.name,
      AGENT_TASK: task,
      // 共享父进程的缓存路径
      CACHE_DIR: getCacheDir()
    }
  });

  return new Promise((resolve, reject) => {
    child.on('message', (msg: AgentMessage) => {
      if (msg.type === 'result') {
        resolve(msg.data);
      }
    });
    child.on('error', reject);
    child.on('exit', (code) => {
      if (code !== 0) {
        reject(new Error(
          'Subagent exited with code ' + code
        ));
      }
    });
  });
}

// loadAgentsDir.ts: 加载自定义代理
async function loadAgentsDir(
  dir: string
): Promise<AgentDefinition[]> {
  const agentsPath = join(dir, '.claude', 'agents');
  const files = await readdir(agentsPath).catch(() => []);
  const agents: AgentDefinition[] = [];

  for (const file of files) {
    if (!file.endsWith('.md')) continue;
    const content = await readFile(
      join(agentsPath, file), 'utf-8'
    );
    // 解析 frontmatter 获取代理配置
    const { frontmatter, body } = parseFrontmatter(content);
    agents.push({
      name: frontmatter.name || basename(file, '.md'),
      prompt: body,
      tools: frontmatter.tools,
      maxTurns: frontmatter.maxTurns,
      mcpServers: frontmatter.mcpServers
    });
  }
  return agents;
}

// agentColorManager.ts: 代理颜色分配
class AgentColorManager {
  private colors = [
    '#e94560', '#0f3460', '#533483',
    '#48c9b0', '#f39c12', '#3498db'
  ];
  private assignments = new Map<string, string>();

  getColor(agentName: string): string {
    if (!this.assignments.has(agentName)) {
      const idx = this.assignments.size % this.colors.length;
      this.assignments.set(agentName, this.colors[idx]);
    }
    return this.assignments.get(agentName)!;
  }
}