feat: 添加 /agent 命令支持交互式切换 Agent 模式
- 在 AgentRegistry 添加 listPrimaryAgents() 方法 - 在 Agent 类添加 setAgentMode/getAgentMode 方法 - 在 TerminalUI 实现 /agent 命令: - /agent 显示当前模式和可用列表 - /agent <name> 切换到指定 Agent - /agent default 切换回默认模式 - 提示符显示当前 Agent 模式(如 @plan You >) - 为 build Agent 添加 prompt 配置
This commit is contained in:
+76
-11
@@ -1,6 +1,7 @@
|
||||
import * as readline from 'readline';
|
||||
import chalk from 'chalk';
|
||||
import type { Agent } from '../core/agent.js';
|
||||
import { agentRegistry } from '../agent/index.js';
|
||||
|
||||
export class TerminalUI {
|
||||
private agent: Agent;
|
||||
@@ -28,11 +29,12 @@ export class TerminalUI {
|
||||
console.log(chalk.cyan('║') + chalk.gray(' Powered by DeepSeek / Claude ') + chalk.cyan('║'));
|
||||
console.log(chalk.cyan('╚════════════════════════════════════════╝\n'));
|
||||
console.log(chalk.gray('输入你的问题,或使用以下命令:'));
|
||||
console.log(chalk.yellow(' /help') + chalk.gray(' - 显示帮助'));
|
||||
console.log(chalk.yellow(' /clear') + chalk.gray(' - 清空对话历史'));
|
||||
console.log(chalk.yellow(' /compact') + chalk.gray(' - 压缩对话历史'));
|
||||
console.log(chalk.yellow(' /context') + chalk.gray(' - 查看上下文使用情况'));
|
||||
console.log(chalk.yellow(' /exit') + chalk.gray(' - 退出程序'));
|
||||
console.log(chalk.yellow(' /help') + chalk.gray(' - 显示帮助'));
|
||||
console.log(chalk.yellow(' /agent') + chalk.gray(' - 切换 Agent 模式'));
|
||||
console.log(chalk.yellow(' /clear') + chalk.gray(' - 清空对话历史'));
|
||||
console.log(chalk.yellow(' /compact') + chalk.gray(' - 压缩对话历史'));
|
||||
console.log(chalk.yellow(' /context') + chalk.gray(' - 查看上下文使用情况'));
|
||||
console.log(chalk.yellow(' /exit') + chalk.gray(' - 退出程序'));
|
||||
console.log('');
|
||||
}
|
||||
|
||||
@@ -56,10 +58,70 @@ export class TerminalUI {
|
||||
return colorFn(`[${used}/${limit}]`);
|
||||
}
|
||||
|
||||
// 处理 /agent 命令
|
||||
private handleAgentCommand(args: string): boolean {
|
||||
const agentName = args.trim();
|
||||
|
||||
// 无参数:显示当前模式和可用 Agent
|
||||
if (!agentName) {
|
||||
const currentMode = this.agent.getAgentModeName();
|
||||
const availableAgents = agentRegistry.listPrimaryAgents();
|
||||
|
||||
console.log(chalk.cyan('\n🤖 Agent 模式:'));
|
||||
console.log(chalk.white(` 当前: ${currentMode === 'default' ? chalk.green('default (通用助手)') : chalk.yellow(currentMode)}`));
|
||||
console.log('');
|
||||
console.log(chalk.white(' 可用 Agent:'));
|
||||
console.log(chalk.gray(' • default - 通用编程助手'));
|
||||
for (const agent of availableAgents) {
|
||||
const marker = agent.name === currentMode ? chalk.green(' ✓') : '';
|
||||
console.log(chalk.gray(` • ${agent.name} - ${agent.description}${marker}`));
|
||||
}
|
||||
console.log('');
|
||||
console.log(chalk.gray(' 用法: /agent <name> 切换到指定 Agent'));
|
||||
console.log(chalk.gray(' /agent default 切换回默认模式'));
|
||||
console.log('');
|
||||
return true;
|
||||
}
|
||||
|
||||
// 切换到默认模式
|
||||
if (agentName === 'default') {
|
||||
this.agent.setAgentMode(null);
|
||||
console.log(chalk.green('\n✓ 已切换到 default (通用助手) 模式\n'));
|
||||
return true;
|
||||
}
|
||||
|
||||
// 切换到指定 Agent
|
||||
const agent = agentRegistry.get(agentName);
|
||||
if (!agent) {
|
||||
const availableNames = ['default', ...agentRegistry.listPrimaryAgents().map(a => a.name)];
|
||||
console.log(chalk.red(`\n✗ 未找到 Agent: ${agentName}`));
|
||||
console.log(chalk.gray(` 可用: ${availableNames.join(', ')}\n`));
|
||||
return true;
|
||||
}
|
||||
|
||||
// 检查是否为 subagent 模式(不能作为主交互 Agent)
|
||||
if (agent.mode === 'subagent') {
|
||||
console.log(chalk.yellow(`\n⚠ Agent "${agentName}" 是 subagent 模式,只能通过 task 工具调用`));
|
||||
console.log(chalk.gray(' 提示: 使用 /agent 查看可交互的 Agent 列表\n'));
|
||||
return true;
|
||||
}
|
||||
|
||||
this.agent.setAgentMode(agent);
|
||||
console.log(chalk.green(`\n✓ 已切换到 ${agent.name} 模式`));
|
||||
console.log(chalk.gray(` ${agent.description}\n`));
|
||||
return true;
|
||||
}
|
||||
|
||||
// 处理特殊命令
|
||||
private async handleCommand(input: string): Promise<boolean> {
|
||||
const command = input.toLowerCase().trim();
|
||||
|
||||
// 检查 /agent 命令(带参数)
|
||||
if (command.startsWith('/agent')) {
|
||||
const args = input.substring(6).trim();
|
||||
return this.handleAgentCommand(args);
|
||||
}
|
||||
|
||||
switch (command) {
|
||||
case '/help':
|
||||
console.log(chalk.cyan('\n📖 帮助信息:'));
|
||||
@@ -70,11 +132,12 @@ export class TerminalUI {
|
||||
console.log(chalk.gray(' • 回答编程问题'));
|
||||
console.log('');
|
||||
console.log(chalk.white(' 命令:'));
|
||||
console.log(chalk.gray(' • /help - 显示此帮助'));
|
||||
console.log(chalk.gray(' • /clear - 清空对话历史'));
|
||||
console.log(chalk.gray(' • /compact - 压缩对话历史,释放上下文空间'));
|
||||
console.log(chalk.gray(' • /context - 显示当前上下文使用情况'));
|
||||
console.log(chalk.gray(' • /exit - 退出程序'));
|
||||
console.log(chalk.gray(' • /help - 显示此帮助'));
|
||||
console.log(chalk.gray(' • /agent - 切换 Agent 模式'));
|
||||
console.log(chalk.gray(' • /clear - 清空对话历史'));
|
||||
console.log(chalk.gray(' • /compact - 压缩对话历史,释放上下文空间'));
|
||||
console.log(chalk.gray(' • /context - 显示当前上下文使用情况'));
|
||||
console.log(chalk.gray(' • /exit - 退出程序'));
|
||||
console.log('');
|
||||
return true;
|
||||
|
||||
@@ -136,7 +199,9 @@ export class TerminalUI {
|
||||
|
||||
// 显示带上下文使用情况的提示符
|
||||
const contextInfo = this.formatContextUsage();
|
||||
this.rl.question(`${contextInfo} ${chalk.green('You >')} `, (answer) => {
|
||||
const agentMode = this.agent.getAgentModeName();
|
||||
const agentIndicator = agentMode === 'default' ? '' : chalk.magenta(`@${agentMode} `);
|
||||
this.rl.question(`${contextInfo} ${agentIndicator}${chalk.green('You >')} `, (answer) => {
|
||||
resolve(answer ?? '');
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user