优化工具

src/index.ts

@@ -4,13 +4,7 @@ import { Command } from 'commander';
 import { Agent } from './core/agent.js';
 import { TerminalUI } from './ui/terminal.js';
 import { loadConfig, initConfig } from './utils/config.js';
-import {
-  bashTool,
-  readFileTool,
-  writeFileTool,
-  listDirTool,
-  searchFilesTool,
-} from './tools/index.js';
+import { allTools } from './tools/index.js';
 
 const program = new Command();
 
@@ -35,15 +29,11 @@ program
     const config = loadConfig();
     const agent = new Agent(config);
 
-    // 注册工具
-    agent.registerTool(bashTool);
-    agent.registerTool(readFileTool);
-    agent.registerTool(writeFileTool);
-    agent.registerTool(listDirTool);
-    agent.registerTool(searchFilesTool);
+    // 注册所有工具
+    agent.registerTools(allTools);
 
     try {
-      const response = await agent.chat(question, (text) => {
+      await agent.chat(question, (text) => {
         process.stdout.write(text);
       });
       console.log('');
@@ -62,11 +52,7 @@ program.action(async () => {
   const agent = new Agent(config);
 
   // 注册所有工具
-  agent.registerTool(bashTool);
-  agent.registerTool(readFileTool);
-  agent.registerTool(writeFileTool);
-  agent.registerTool(listDirTool);
-  agent.registerTool(searchFilesTool);
+  agent.registerTools(allTools);
 
   // 启动终端 UI
   const ui = new TerminalUI(agent);