@saber2pr/ai-agent 0.0.4 → 0.0.5

package/lib/agent.d.ts

@@ -5,7 +5,13 @@ export default class McpAgent {
     private allTools;
     private messages;
     private engine;
+    private encoder;
     constructor();
+    /**
+     * 计算当前消息列表的总 Token 消耗
+     * 兼容多模态内容 (Content Parts) 和 工具调用 (Tool Calls)
+     */
+    private calculateTokens;
     /**
      * 核心功能：内置代码分析工具
      * 这里的逻辑直接调用 PromptEngine，不走网络请求，效率极高
@@ -14,10 +20,10 @@ export default class McpAgent {
     private ensureApiConfig;
     private loadMcpConfigs;
     init(): Promise<void>;
+    private processChat;
     /**
-     * 核心交互循环 (Reasoning Loop)
-     * 允许 AI 连续调用工具来解决复杂代码问题
+     * 简易 Loading 动画辅助函数
      */
-    private processChat;
+    private showLoading;
     start(): Promise<void>;
 }

package/lib/agent.js

@@ -44,6 +44,7 @@ const path_1 = __importDefault(require("path"));
 const os_1 = __importDefault(require("os"));
 const readline = __importStar(require("readline"));
 const ts_context_mcp_1 = require("@saber2pr/ts-context-mcp"); // 引入我们的核心引擎
+const js_tiktoken_1 = require("js-tiktoken");
 const CONFIG_FILE = path_1.default.join(os_1.default.homedir(), ".saber2pr-agent.json");
 class McpAgent {
     constructor() {
@@ -51,6 +52,7 @@ class McpAgent {
         this.clients = [];
         this.allTools = [];
         this.messages = [];
+        this.encoder = (0, js_tiktoken_1.getEncoding)("cl100k_base");
         // 默认以当前工作目录为分析目标
         this.engine = new ts_context_mcp_1.PromptEngine(process.cwd());
         this.messages.push({
@@ -58,11 +60,54 @@ class McpAgent {
             content: `你是一个专业的 AI 代码架构师。
 你可以访问本地文件系统并利用 AST (抽象语法树) 技术分析代码。
 你的核心目标是提供准确的代码结构、依赖关系和逻辑分析。
+请先使用 get_repo_map 查看项目整体代码结构。
 请优先使用 read_skeleton 查看结构，只有在必要时才使用 read_full_code 或 get_method_body。`,
         });
         // 初始化内置工具
         this.registerBuiltinTools();
     }
+    /**
+     * 计算当前消息列表的总 Token 消耗
+     * 兼容多模态内容 (Content Parts) 和 工具调用 (Tool Calls)
+     */
+    calculateTokens() {
+        let total = 0;
+        for (const msg of this.messages) {
+            // 1. 处理消息内容 (Content)
+            if (msg.content) {
+                if (typeof msg.content === "string") {
+                    // 普通文本消息
+                    total += this.encoder.encode(msg.content).length;
+                }
+                else if (Array.isArray(msg.content)) {
+                    // 多模态/复合内容消息 (ChatCompletionContentPart[])
+                    for (const part of msg.content) {
+                        if (part.type === "text" && "text" in part) {
+                            total += this.encoder.encode(part.text || "").length;
+                        }
+                        // 注意：图片 (image_url) 的 Token 计算通常基于分辨率，tiktoken 无法计算
+                    }
+                }
+            }
+            // 2. 处理助手角色发出的工具调用请求 (Assistant Tool Calls)
+            // 这是为了统计 AI 发出的指令所占用的 Token
+            if (msg.role === "assistant" && msg.tool_calls) {
+                for (const call of msg.tool_calls) {
+                    if (call.type === "function") {
+                        // 统计函数名和参数字符串
+                        total += this.encoder.encode(call.function.name).length;
+                        total += this.encoder.encode(call.function.arguments).length;
+                    }
+                }
+            }
+            // 3. 处理工具返回的结果 (Tool Role)
+            // 在 processChat 中，我们确保了工具返回的 result 最终被转为了 string
+            if (msg.role === "tool" && typeof msg.content === "string") {
+                total += this.encoder.encode(msg.content).length;
+            }
+        }
+        return total;
+    }
     /**
      * 核心功能：内置代码分析工具
      * 这里的逻辑直接调用 PromptEngine，不走网络请求，效率极高
@@ -109,7 +154,25 @@ class McpAgent {
                         required: ["filePath"],
                     },
                 },
-                _handler: async ({ filePath }) => this.engine.getSkeleton(filePath),
+                _handler: async (args) => {
+                    // 1. 严格路径守卫：防止 undefined 或空字符串进入 path 模块
+                    const pathArg = args === null || args === void 0 ? void 0 : args.filePath;
+                    if (typeof pathArg !== 'string' || pathArg.trim() === '') {
+                        return `Error: 参数 'filePath' 无效。收到的是: ${JSON.stringify(pathArg)}`;
+                    }
+                    try {
+                        // 2. 刷新引擎状态，确保分析的是最新的文件内容
+                        this.engine.refresh();
+                        // 3. 执行获取
+                        const result = this.engine.getSkeleton(pathArg);
+                        // 4. 空值回退：防止 getSkeleton 返回 null 导致后续统计 Token 时崩溃
+                        return result || `// Warning: 文件 ${pathArg} 存在但未找到任何可提取的结构。`;
+                    }
+                    catch (error) {
+                        // 5. 捕获 AST 级别的 match 错误
+                        return `Error: 解析文件 ${pathArg} 时发生内部错误: ${error.message}`;
+                    }
+                },
             },
             {
                 type: "function",
@@ -127,6 +190,41 @@ class McpAgent {
                 },
                 _handler: async ({ filePath, methodName }) => this.engine.getMethodImplementation(filePath, methodName),
             },
+            {
+                type: "function",
+                function: {
+                    name: "read_full_code",
+                    description: "读取指定文件的完整源代码内容。当需要分析具体实现逻辑或查找硬编码字符串时使用。",
+                    parameters: {
+                        type: "object",
+                        properties: {
+                            filePath: { type: "string", description: "文件相对路径" },
+                        },
+                        required: ["filePath"],
+                    },
+                },
+                // 核心实现：直接利用 fs 读取
+                _handler: async ({ filePath }) => {
+                    try {
+                        if (typeof filePath !== 'string' || !filePath) {
+                            return "Error: filePath 不能为空";
+                        }
+                        // 拼合绝对路径
+                        const fullPath = path_1.default.resolve(this.engine.getRootDir(), filePath);
+                        // 安全检查：防止 AI 尝试读取项目外的敏感文件
+                        if (!fullPath.startsWith(this.engine.getRootDir())) {
+                            return "Error: 权限拒绝，禁止访问项目目录外的文件。";
+                        }
+                        if (!fs_1.default.existsSync(fullPath)) {
+                            return `Error: 文件不存在: ${filePath}`;
+                        }
+                        return fs_1.default.readFileSync(fullPath, "utf-8");
+                    }
+                    catch (err) {
+                        return `Error: 读取文件失败: ${err.message}`;
+                    }
+                },
+            },
         ];
         this.allTools.push(...builtinTools);
     }
@@ -200,50 +298,84 @@ class McpAgent {
             }
         }
     }
-    /**
-     * 核心交互循环 (Reasoning Loop)
-     * 允许 AI 连续调用工具来解决复杂代码问题
-     */
     async processChat(userInput) {
-        this.messages.push({ role: "user", content: userInput });
+        var _a;
+        this.messages.push({ role: 'user', content: userInput });
         while (true) {
-            const response = await this.openai.chat.completions.create({
-                model: this.modelName,
-                messages: this.messages,
-                tools: this.allTools.map(({ _handler, _client, _originalName, ...rest }) => rest),
-                tool_choice: "auto",
-            });
+            // 打印当前上下文的累计 Token
+            const currentInputTokens = this.calculateTokens();
+            console.log(`\n📊 当前上下文累计: ${currentInputTokens} tokens`);
+            const stopLoading = this.showLoading("🤖 Agent 正在思考...");
+            let response;
+            try {
+                response = await this.openai.chat.completions.create({
+                    model: this.modelName,
+                    messages: this.messages,
+                    tools: this.allTools.map(({ _handler, _client, _originalName, ...rest }) => rest),
+                    tool_choice: 'auto'
+                });
+            }
+            finally {
+                stopLoading();
+            }
             const message = response.choices[0].message;
             this.messages.push(message);
+            // 计算本次 AI 回复生成的 Token
+            const completionTokens = ((_a = response.usage) === null || _a === void 0 ? void 0 : _a.completion_tokens) ||
+                (message.content ? this.encoder.encode(message.content).length : 0);
+            console.log(`✨ AI 回复消耗: ${completionTokens} tokens`);
             if (!message.tool_calls) {
                 console.log(`\n🤖 Agent: ${message.content}`);
                 break;
             }
-            console.log(`\n⚙️ 正在思考并执行 ${message.tool_calls.length} 个操作...`);
+            console.log(`\n⚙️ 正在执行 ${message.tool_calls.length} 个操作...`);
             for (const call of message.tool_calls) {
-                const tool = this.allTools.find((t) => t.function.name === call.function.name);
+                const tool = this.allTools.find(t => t.function.name === call.function.name);
+                const args = JSON.parse(call.function.arguments);
+                // 打印文件路径提示
+                if (args.filePath) {
+                    console.log(`   📂 正在查看文件: ${args.filePath}`);
+                }
+                console.log(`   🛠️  执行: ${call.function.name}`);
                 let result;
                 if (tool === null || tool === void 0 ? void 0 : tool._handler) {
-                    // 执行内置 PromptEngine 工具
-                    result = await tool._handler(JSON.parse(call.function.arguments));
+                    result = await tool._handler(args);
                 }
                 else if ((tool === null || tool === void 0 ? void 0 : tool._client) && tool._originalName) {
-                    // 执行外部 MCP 工具
                     const mcpRes = await tool._client.callTool({
                         name: tool._originalName,
-                        arguments: JSON.parse(call.function.arguments),
+                        arguments: args
                     });
                     result = mcpRes.content;
                 }
+                const resultContent = typeof result === "string" ? result : JSON.stringify(result);
+                // 打印工具返回结果的 Token 消耗
+                const toolResultTokens = this.encoder.encode(resultContent).length;
+                console.log(`   📝 工具输出: ${toolResultTokens} tokens`);
                 this.messages.push({
-                    role: "tool",
+                    role: 'tool',
                     tool_call_id: call.id,
-                    content: typeof result === "string" ? result : JSON.stringify(result),
+                    content: resultContent
                 });
-                console.log(`   - 完成: ${call.function.name}`);
+                console.log(`   ✅ 完成: ${call.function.name}`);
             }
         }
     }
+    /**
+     * 简易 Loading 动画辅助函数
+     */
+    showLoading(text) {
+        const chars = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
+        let i = 0;
+        const timer = setInterval(() => {
+            process.stdout.write(`\r${chars[i]} ${text}`);
+            i = (i + 1) % chars.length;
+        }, 80);
+        return () => {
+            clearInterval(timer);
+            process.stdout.write('\r\x1b[K'); // 清除当前行
+        };
+    }
     async start() {
         await this.init();
         const rl = readline.createInterface({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saber2pr/ai-agent",
-  "version": "0.0.4",
+  "version": "0.0.5",
   "description": "AI Assistant CLI.",
   "author": "saber2pr",
   "license": "ISC",
@@ -22,7 +22,8 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.25.3",
-    "@saber2pr/ts-context-mcp": "^0.0.5",
+    "@saber2pr/ts-context-mcp": "^0.0.6",
+    "js-tiktoken": "^1.0.21",
     "openai": "^6.16.0"
   },
   "devDependencies": {