@saber2pr/ai-agent 0.0.12 → 0.0.14

package/lib/core/agent-chain.d.ts

@@ -11,6 +11,7 @@ export default class McpChainAgent {
     private apiModel?;
     private targetDir;
     private verbose;
+    private runningTokenCounter;
     constructor(options?: AgentOptions);
     /**
      * 工具处理器包装逻辑：增加日志打印和 Token 监控

package/lib/core/agent-chain.js

@@ -52,6 +52,7 @@ class McpChainAgent {
         this.messages = [];
         this.encoder = (0, js_tiktoken_1.getEncoding)("cl100k_base");
         this.extraTools = [];
+        this.runningTokenCounter = 0; // 新增：用于追踪当前 Task 的动态消耗
         this.extraTools = (options === null || options === void 0 ? void 0 : options.tools) || [];
         this.maxTokens = (options === null || options === void 0 ? void 0 : options.maxTokens) || 100000;
         this.apiConfig = options === null || options === void 0 ? void 0 : options.apiConfig;
@@ -89,8 +90,15 @@ class McpChainAgent {
             const result = await handler(args);
             const content = typeof result === "string" ? result : JSON.stringify(result);
             // 3. 统计 Token 消耗
-            const tokens = this.encoder.encode(content).length;
-            console.log(`   [输出长度]: ${tokens} tokens`);
+            const outputTokens = this.encoder.encode(content).length;
+            console.log(`   [输出长度]: ${outputTokens} tokens`);
+            // 2. 更新动态计数器：
+            // 这里的逻辑是：当前基础 Context + 本次工具调用的输出
+            // 随着迭代增加，这个值会一直累加，直到任务结束存入 messages
+            const baseTokens = this.calculateTokens();
+            this.runningTokenCounter = baseTokens + outputTokens;
+            // 3. 打印正确且递增的状态
+            console.log(`   📊 状态: Context ${this.runningTokenCounter} / Limit ${this.maxTokens} tokens`);
             return content;
         };
     }

package/lib/core/agent.d.ts

@@ -16,10 +16,6 @@ export default class McpAgent {
      */
     private calculateTokens;
     private initTools;
-    /**
-     * 核心功能：内置代码分析工具（基于 engine/targetDir，可被外部通过 createDefaultBuiltinTools 替代）
-     */
-    private registerBuiltinTools;
     private ensureApiConfig;
     private loadMcpConfigs;
     init(): Promise<void>;

package/lib/core/agent.js

@@ -145,14 +145,6 @@ class McpAgent {
             this.allTools.push(...allTools);
         }
     }
-    /**
-     * 核心功能：内置代码分析工具（基于 engine/targetDir，可被外部通过 createDefaultBuiltinTools 替代）
-     */
-    registerBuiltinTools(options) {
-        this.allTools.push(...(0, builtin_1.createDefaultBuiltinTools)({
-            options,
-        }));
-    }
     // --- 初始化与环境准备 (API Config & MCP Servers) ---
     async ensureApiConfig() {
         if (this.apiConfig)

package/lib/utils/createTool.js

@@ -24,8 +24,8 @@ function createTool(options) {
             if (((_a = options.validateParams) === null || _a === void 0 ? void 0 : _a.length) > 0) {
                 for (const arg in args) {
                     if (options.validateParams.includes(arg)) {
-                        if (typeof args[arg] !== "string" || args[arg].trim() === "") {
-                            return `Error: 参数 '${arg}' 无效。收到的是: ${JSON.stringify(args[arg])}`;
+                        if (typeof args[arg] === 'undefined') {
+                            return `Error: 参数 '${arg}' 缺失`;
                         }
                     }
                 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saber2pr/ai-agent",
-  "version": "0.0.12",
+  "version": "0.0.14",
   "description": "AI Assistant CLI.",
   "author": "saber2pr",
   "license": "ISC",