@saber2pr/ai-agent 0.0.6 → 0.0.7

package/lib/agent.d.ts

@@ -10,6 +10,7 @@ export interface AgentOptions {
     tools?: any[];
     /** 注入到 System Prompt 中的额外指令/规则/上下文 */
     extraSystemPrompt?: any;
+    maxTokens?: number;
 }
 export default class McpAgent {
     private openai;
@@ -20,6 +21,7 @@ export default class McpAgent {
     private engine;
     private encoder;
     private extraTools;
+    private maxTokens;
     constructor(options?: AgentOptions);
     /**
      * 计算当前消息列表的总 Token 消耗
@@ -36,6 +38,11 @@ export default class McpAgent {
     private loadMcpConfigs;
     init(): Promise<void>;
     private processChat;
+    /**
+   * 裁剪上下文消息列表
+   * 保留第一条 System 消息，并移除中间的旧消息直到低于阈值
+   */
+    private pruneMessages;
     /**
      * 简易 Loading 动画辅助函数
      */

package/lib/agent.js

@@ -56,11 +56,25 @@ class McpAgent {
         this.extraTools = [];
         this.engine = new ts_context_mcp_1.PromptEngine((options === null || options === void 0 ? void 0 : options.targetDir) || process.cwd());
         this.extraTools = (options === null || options === void 0 ? void 0 : options.tools) || []; // 接收外部传入的工具
-        let baseSystemPrompt = `你是一个专业的 AI 代码架构师。
-你可以访问本地文件系统并利用 AST (抽象语法树) 技术分析代码。
-你的核心目标是提供准确的代码结构、依赖关系和逻辑分析。
-请先使用 get_repo_map 查看项目整体代码结构。
-请优先使用 read_skeleton 查看结构，只有在必要时才使用 read_full_code 或 get_method_body。`;
+        this.maxTokens = (options === null || options === void 0 ? void 0 : options.maxTokens) || 100000; // 默认 100k
+        let baseSystemPrompt = `你是一个专业的 AI 代码架构师，具备深度的源码分析与工程化处理能力。
+
+    ### 核心操作规范：
+    1. **全局扫描（强制首选）**：在开始任何分析任务前，你【必须】首先调用 'get_repo_map'。这是你理解项目目录结构、技术栈及模块关系的唯一权威来源。
+    2. **循序渐进的分析路径**：
+       - 优先使用 'read_skeleton' 提取接口、类和函数签名，以最低的 Token 成本建立代码逻辑视图。
+       - 仅在需要深入分析具体业务逻辑、提取精准代码块或进行代码修改建议时，才使用 'read_full_code' 或 'get_method_body'。
+    3. **真实性原则**：
+       - 所有的代码分析、行号定位和逻辑推断必须基于工具返回的真实内容，严禁基于文件名进行虚假猜测。
+       - 如果工具返回结果为空或报错，应尝试调整路径或更换工具。
+    
+    ### 技术能力：
+    - 精通 TypeScript/JavaScript 及其 AST 结构，能准确识别各种复杂的声明与调用关系。
+    - 能够理解代码间的依赖链路，并结合项目上下文给出合理的架构建议。
+    
+    ### 执行准则：
+    - **任务导向**：直接通过工具链解决问题，减少不必要的中间对话。
+    - **自主决策**：根据任务需求自主选择最合适的工具组合，无需每一步都向用户请示。`;
         // 2. 拼接额外指令
         if (options === null || options === void 0 ? void 0 : options.extraSystemPrompt) {
             const extra = typeof options.extraSystemPrompt === 'string'
@@ -211,7 +225,13 @@ class McpAgent {
                         required: ["filePath", "methodName"],
                     },
                 },
-                _handler: async ({ filePath, methodName }) => this.engine.getMethodImplementation(filePath, methodName),
+                _handler: async ({ filePath, methodName }) => {
+                    // --- 新增：同样的 Token 守卫 ---
+                    if (this.calculateTokens() > this.maxTokens) {
+                        return `[SYSTEM WARNING]: Token 消耗已达上限，禁止获取详细方法体。请利用已获取的 Skeleton 信息进行分析。`;
+                    }
+                    return this.engine.getMethodImplementation(filePath, methodName);
+                },
             },
             {
                 type: "function",
@@ -228,6 +248,11 @@ class McpAgent {
                 },
                 // 核心实现：直接利用 fs 读取
                 _handler: async ({ filePath }) => {
+                    // --- 新增：Token 守卫 ---
+                    const currentTokens = this.calculateTokens();
+                    if (currentTokens > this.maxTokens) {
+                        return `[SYSTEM WARNING]: 当前上下文已达到 ${currentTokens} tokens (上限 ${this.maxTokens})。为了保证系统稳定，已拦截 read_full_code。请立即根据已知信息进行总结或停止阅读更多代码。`;
+                    }
                     try {
                         if (typeof filePath !== 'string' || !filePath) {
                             return "Error: filePath 不能为空";
@@ -329,9 +354,18 @@ class McpAgent {
         var _a;
         this.messages.push({ role: 'user', content: userInput });
         while (true) {
+            // --- 新增：发送请求前先检查并裁剪 ---
+            this.pruneMessages();
             // 打印当前上下文的累计 Token
             const currentInputTokens = this.calculateTokens();
             console.log(`\n📊 当前上下文累计: ${currentInputTokens} tokens`);
+            // 如果接近上限（如 80%），在消息队列中插入一条隐含的系统指令
+            if (currentInputTokens > this.maxTokens * 0.8 && currentInputTokens <= this.maxTokens) {
+                this.messages.push({
+                    role: "system",
+                    content: "注意：上下文即将耗尽。请停止读取新文件，优先处理现有信息并尽快输出结果。"
+                });
+            }
             const stopLoading = this.showLoading("🤖 Agent 正在思考...");
             let response;
             try {
@@ -388,6 +422,24 @@ class McpAgent {
             }
         }
     }
+    /**
+   * 裁剪上下文消息列表
+   * 保留第一条 System 消息，并移除中间的旧消息直到低于阈值
+   */
+    pruneMessages() {
+        const currentTokens = this.calculateTokens();
+        if (currentTokens <= this.maxTokens)
+            return;
+        console.log(`\n⚠️ 上下文达到限制 (${currentTokens} tokens)，正在自动裁剪...`);
+        // 策略：保留索引 0 (System)，从索引 1 开始删除
+        // 每次删除一对 (通常是助理请求 + 工具回复，或者用户提问 + 助理回答)
+        while (this.calculateTokens() > this.maxTokens && this.messages.length > 2) {
+            // 始终保留系统提示词 (index 0) 和最后一条消息 (保持对话连贯)
+            // 删除索引为 1 的消息
+            this.messages.splice(1, 1);
+        }
+        console.log(`✅ 裁剪完成，当前上下文: ${this.calculateTokens()} tokens`);
+    }
     /**
      * 简易 Loading 动画辅助函数
      */

package/package.json

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