@saber2pr/ai-agent 0.0.32 → 0.0.34

package/lib/core/agent-graph.js

@@ -255,16 +255,24 @@ class McpGraphAgent {
         return config;
     }
     async chat(query = "开始代码审计") {
-        await this.ensureInitialized();
-        await this.getModel();
-        const app = await this.createGraph();
-        const stream = await app.stream({
-            messages: [new messages_1.HumanMessage(query)],
-            mode: "auto",
-            targetCount: this.maxTargetCount,
-        }, { configurable: { thread_id: "auto_worker" }, recursionLimit: this.recursionLimit, debug: this.verbose, });
-        for await (const output of stream)
-            this.renderOutput(output);
+        try {
+            await this.ensureInitialized();
+            await this.getModel();
+            const app = await this.createGraph();
+            const stream = await app.stream({
+                messages: [new messages_1.HumanMessage(query)],
+                mode: "auto",
+                targetCount: this.maxTargetCount,
+            }, { configurable: { thread_id: "auto_worker" }, recursionLimit: this.recursionLimit, debug: this.verbose, });
+            for await (const output of stream)
+                this.renderOutput(output);
+        }
+        catch (error) {
+            console.error("\n❌ Chat 过程中发生错误:", error);
+        }
+        finally {
+            await this.closeMcpClients();
+        }
     }
     async start() {
         await this.ensureInitialized();
@@ -373,7 +381,6 @@ class McpGraphAgent {
 # 🎯 核心指令
 1. **任务终结判定**：当你已经读取了用户要求的文件、回答了问题或完成了审计目标时，必须立即提供最终回复。
 2. **回复格式**：任务完成时，请以 "Final Answer:" 开头进行总结，此时不再调用任何工具。
-3. **避免重复**：不要重复调用相同的工具和参数。如果工具返回错误，请分析原因（如参数 Schema 错误）而不是盲目重试。{recentToolCalls}
 
 # 📝 审计任务专项
 1. 避免在同一个文件上陷入无限循环尝试。

package/lib/core/agent.d.ts

@@ -9,7 +9,9 @@ export default class McpAgent {
     private maxTokens;
     private apiConfig;
     private targetDir;
+    private mcpClients;
     constructor(options?: AgentOptions);
+    private closeMcpClients;
     /**
      * 计算当前消息列表的总 Token 消耗
      * 兼容多模态内容 (Content Parts) 和 工具调用 (Tool Calls)

package/lib/core/agent.js

@@ -55,6 +55,7 @@ class McpAgent {
         this.messages = [];
         this.encoder = (0, js_tiktoken_1.getEncoding)("cl100k_base");
         this.extraTools = [];
+        this.mcpClients = [];
         this.targetDir = (options === null || options === void 0 ? void 0 : options.targetDir) || process.cwd();
         this.extraTools = (options === null || options === void 0 ? void 0 : options.tools) || []; // 接收外部传入的工具
         this.maxTokens = (options === null || options === void 0 ? void 0 : options.maxTokens) || 100000; // 默认 100k
@@ -63,12 +64,11 @@ class McpAgent {
 你的目标是理解并分析用户项目，请务必遵循以下工作流：
 
 ### 第一阶段：全景感知 (The "Where" Phase)
-1. **必须首先调用 'get_directory_tree'**：获取项目完整文件列表，包括样式文件 (.less, .css) 和资源文件。
-2. 结合目录结构，观察项目架构（如 Monorepo 结构或 src 布局）。
+1. **必须首先调用 'list_directory'** 了解全局，再深入具体文件。
 
 ### 第二阶段：逻辑映射 (The "What" Phase)
 1. **调用 'get_repo_map'**：针对代码文件提取导出定义，理解模块间的调用关系。
-2. 如果需要查看具体的样式定义，直接使用 'read_text_file' 读取 .less 或 .css 文件。
+2. 如果需要查看具体的样式定义，直接使用 'read_text_file' 读取文件内容。
 
 ### 核心原则：
 - 不要猜测文件是否存在，先看目录树。
@@ -87,6 +87,16 @@ class McpAgent {
         });
         this.initTools(options); // 注入外部工具
     }
+    // ✅ 新增：关闭连接
+    async closeMcpClients() {
+        for (const client of this.mcpClients) {
+            try {
+                await client.close();
+            }
+            catch (e) { }
+        }
+        this.mcpClients = [];
+    }
     /**
      * 计算当前消息列表的总 Token 消耗
      * 兼容多模态内容 (Content Parts) 和 工具调用 (Tool Calls)
@@ -198,6 +208,7 @@ class McpAgent {
                 });
                 const client = new index_js_1.Client({ name, version: "1.0.0" }, { capabilities: {} });
                 await client.connect(transport);
+                this.mcpClients.push(client);
                 const { tools } = await client.listTools();
                 this.allTools.push(...tools.map((t) => ({
                     type: "function",
@@ -348,15 +359,24 @@ class McpAgent {
      * @returns AI 的最终答复内容
      */
     async chat(input) {
-        if (!this.openai) {
-            await this.init();
+        try {
+            if (!this.openai) {
+                await this.init();
+            }
+            // 调用现有的处理逻辑
+            // 假设你的 processChat 已经处理了所有的 tool_calls 循环
+            await this.processChat(input);
+            // 返回消息列表中的最后一条 AI 回复
+            const lastMsg = this.messages[this.messages.length - 1];
+            return lastMsg.role === 'assistant' ? lastMsg.content : '';
+        }
+        catch (error) {
+            console.error("\n❌ 系统错误:", error.message);
+            return '';
+        }
+        finally {
+            await this.closeMcpClients();
         }
-        // 调用现有的处理逻辑
-        // 假设你的 processChat 已经处理了所有的 tool_calls 循环
-        await this.processChat(input);
-        // 返回消息列表中的最后一条 AI 回复
-        const lastMsg = this.messages[this.messages.length - 1];
-        return lastMsg.role === 'assistant' ? lastMsg.content : '';
     }
     // 修改原来的 start 方法，使其内部也调用 chat
     async start() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saber2pr/ai-agent",
-  "version": "0.0.32",
+  "version": "0.0.34",
   "description": "AI Assistant CLI.",
   "author": "saber2pr",
   "license": "ISC",
@@ -23,15 +23,14 @@
     "prepublishOnly": "tsc"
   },
   "dependencies": {
-    "@langchain/core": "0.3.39",
-    "@langchain/langgraph": "^0.2.39",
-    "@langchain/openai": "0.4.0",
+    "@langchain/core": "0.3.40",
+    "@langchain/langgraph": "^0.2.74",
+    "@langchain/openai": "0.4.9",
     "@modelcontextprotocol/sdk": "^1.25.3",
     "@saber2pr/ts-context-mcp": "^0.0.8",
     "diff": "^8.0.3",
     "glob": "^10.5.0",
     "js-tiktoken": "^1.0.21",
-    "langchain": "0.3.15",
     "minimatch": "^10.0.1",
     "openai": "^6.16.0",
     "zod": "3.25",
@@ -40,7 +39,7 @@
   },
   "resolutions": {
     "zod": "3.25",
-    "@langchain/core": "0.3.39"
+    "@langchain/core": "0.3.40"
   },
   "devDependencies": {
     "@types/node": "^16.3.3",