@saber2pr/ai-agent 0.0.9 → 0.0.10

package/README.md

@@ -19,6 +19,14 @@ A high-performance AI Agent toolkit designed for automated code auditing, reposi
 ## 🛠️ Installation
 
 ```bash
+sudo npm i -g @saber2pr/ai-agent
+
+# call openapi
+sagent
+
+# call third api
+sagent-chain
+
 # Clone the repository
 git clone https://github.com/saber2pr/ai-agent.git
 cd ai-agent

package/lib/cli-chain.js

@@ -4,6 +4,6 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-const agent_chain_1 = __importDefault(require("./agent-chain"));
-const manager = new agent_chain_1.default();
-manager.start();
+const agent_chain_1 = __importDefault(require("./core/agent-chain"));
+const agent = new agent_chain_1.default();
+agent.start();

package/lib/cli.js

@@ -4,6 +4,6 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-const agent_1 = __importDefault(require("./agent"));
-const manager = new agent_1.default();
-manager.start();
+const agent_1 = __importDefault(require("./core/agent"));
+const agent = new agent_1.default();
+agent.start();

package/lib/config/config.d.ts

@@ -0,0 +1 @@
+export declare const CONFIG_FILE: string;

package/lib/config/config.js

@@ -0,0 +1,9 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CONFIG_FILE = void 0;
+const path_1 = __importDefault(require("path"));
+const os_1 = __importDefault(require("os"));
+exports.CONFIG_FILE = path_1.default.join(os_1.default.homedir(), ".saber2pr-agent.json");

package/lib/{agent-chain.d.ts → core/agent-chain.d.ts}

@@ -1,28 +1,7 @@
-import { BaseChatModel } from "@langchain/core/language_models/chat_models";
-interface ApiConfig {
-    baseURL: string;
-    apiKey: string;
-    model: string;
-}
-export interface CustomTool {
-    name: string;
-    description: string;
-    parameters: any;
-    handler: (args: any) => Promise<any>;
-}
-export interface AgentOptions {
-    targetDir?: string;
-    tools?: CustomTool[];
-    extraSystemPrompt?: any;
-    maxTokens?: number;
-    apiConfig?: ApiConfig;
-    apiModel?: BaseChatModel;
-    maxIterations?: number;
-}
+import { AgentOptions } from '../types/type';
 export default class McpChainAgent {
     private allTools;
     private messages;
-    private engine;
     private encoder;
     private extraTools;
     private maxTokens;
@@ -30,6 +9,8 @@ export default class McpChainAgent {
     private apiConfig;
     private maxIterations;
     private apiModel?;
+    private targetDir;
+    private verbose;
     constructor(options?: AgentOptions);
     /**
      * 工具处理器包装逻辑：增加日志打印和 Token 监控
@@ -45,4 +26,3 @@ export default class McpChainAgent {
     start(): Promise<void>;
     private ensureApiConfig;
 }
-export {};

package/lib/{agent-chain.js → core/agent-chain.js}

@@ -37,28 +37,29 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const fs_1 = __importDefault(require("fs"));
-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 openai_1 = require("@langchain/openai");
-const tools_1 = require("@langchain/core/tools");
 const agents_1 = require("langchain/agents");
+const readline = __importStar(require("readline"));
 const prompts_1 = require("@langchain/core/prompts");
-const CONFIG_FILE = path_1.default.join(os_1.default.homedir(), ".saber2pr-agent.json");
+const tools_1 = require("@langchain/core/tools");
+const openai_1 = require("@langchain/openai");
+const builtin_1 = require("../tools/builtin");
+const config_1 = require("../config/config");
+const jsonSchemaToZod_1 = require("../utils/jsonSchemaToZod");
 class McpChainAgent {
     constructor(options) {
+        var _a;
         this.allTools = [];
         this.messages = [];
         this.encoder = (0, js_tiktoken_1.getEncoding)("cl100k_base");
         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) || [];
         this.maxTokens = (options === null || options === void 0 ? void 0 : options.maxTokens) || 100000;
         this.apiConfig = options === null || options === void 0 ? void 0 : options.apiConfig;
         this.maxIterations = (options === null || options === void 0 ? void 0 : options.maxIterations) || 20;
         this.apiModel = options === null || options === void 0 ? void 0 : options.apiModel;
+        this.targetDir = (options === null || options === void 0 ? void 0 : options.targetDir) || process.cwd();
+        this.verbose = (options === null || options === void 0 ? void 0 : options.verbose) || false;
         const baseSystemPrompt = `你是一个专业的 AI 代码架构师，具备深度的源码分析与工程化处理能力。
     
 ### 核心操作规范：
@@ -73,7 +74,19 @@ class McpChainAgent {
                 ? `${baseSystemPrompt}\n\n[额外指令]:\n${JSON.stringify(options.extraSystemPrompt)}`
                 : baseSystemPrompt,
         });
-        this.registerBuiltinTools();
+        if ((_a = options === null || options === void 0 ? void 0 : options.builtinTools) === null || _a === void 0 ? void 0 : _a.length) {
+            this.allTools.push(...options.builtinTools.map((t) => ({
+                type: t.type,
+                function: t.function,
+                _handler: this.wrapHandler(t.function.name, t._handler),
+            })));
+        }
+        else {
+            this.registerBuiltinTools({
+                ...options,
+                ...this,
+            });
+        }
         this.injectCustomTools();
     }
     /**
@@ -95,64 +108,16 @@ class McpChainAgent {
             return content;
         };
     }
-    registerBuiltinTools() {
-        const builtinTools = [
-            {
-                type: "function",
-                function: { name: "get_repo_map", description: "获取项目全局结构图和导出清单", parameters: { type: "object" } },
-                _handler: this.wrapHandler("get_repo_map", async () => {
-                    this.engine.refresh();
-                    return this.engine.getRepoMap();
-                }),
-            },
-            {
-                type: "function",
-                function: {
-                    name: "read_skeleton",
-                    description: "读取代码骨架（接口、类定义等），非常节省 Token",
-                    parameters: { type: "object", properties: { filePath: { type: "string" } } },
-                },
-                _handler: this.wrapHandler("read_skeleton", async ({ filePath }) => this.engine.getSkeleton(filePath)),
-            },
-            {
-                type: "function",
-                function: {
-                    name: "read_full_code",
-                    description: "读取完整源码。注意：仅在需要具体行号或精细逻辑时使用",
-                    parameters: { type: "object", properties: { filePath: { type: "string" } } },
-                },
-                _handler: this.wrapHandler("read_full_code", 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 不能为空";
-                        }
-                        // 拼合绝对路径
-                        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}`;
-                        }
-                        const content = fs_1.default.readFileSync(fullPath, "utf-8");
-                        // 加上行号，AI 就能在 generate_review 里给出准确的 line 参数
-                        return content.split('\n')
-                            .map((line, i) => `${i + 1} | ${line}`)
-                            .join('\n');
-                    }
-                    catch (err) {
-                        return `Error: 读取文件失败: ${err.message}`;
-                    }
-                }),
-            }
-        ];
-        this.allTools.push(...builtinTools);
+    registerBuiltinTools(options) {
+        const defaults = (0, builtin_1.createDefaultBuiltinTools)({
+            options,
+            getCurrentTokens: () => this.calculateTokens(),
+        });
+        this.allTools.push(...defaults.map((t) => ({
+            type: t.type,
+            function: t.function,
+            _handler: this.wrapHandler(t.function.name, t._handler),
+        })));
     }
     injectCustomTools() {
         for (const tool of this.extraTools) {
@@ -196,11 +161,19 @@ class McpChainAgent {
                 streaming: false
             });
         }
-        const langchainTools = this.allTools.map(t => new tools_1.DynamicTool({
-            name: t.function.name,
-            description: t.function.description || "",
-            func: t._handler
-        }));
+        const langchainTools = this.allTools.map(t => {
+            return new tools_1.DynamicStructuredTool({
+                name: t.function.name,
+                description: t.function.description || "",
+                // 定义 schema 告诉 LangChain 这是一个对象输入
+                // passthrough() 允许接收未在 schema 中显式定义的其他字段
+                schema: (0, jsonSchemaToZod_1.jsonSchemaToZod)(t.function.parameters),
+                func: async (args) => {
+                    // 这里的 args 已经被 LangChain 自动解析为对象
+                    return await t._handler(args);
+                },
+            });
+        });
         const prompt = prompts_1.PromptTemplate.fromTemplate(`
 {system_prompt}
 
@@ -208,39 +181,32 @@ class McpChainAgent {
 --------------------
 {tools}
 
+工具名称列表: [{tool_names}]
+
 ### 📝 交互协议格式 (PROTOCOL)
 --------------------
-为了确保任务成功，你必须【严格】遵守以下 ReAct 交互格式：
-
-1. **思考与行动阶段 (Thought & Action)**:
-    Thought: 我需要执行什么操作？
-    Action: 工具名称 (必须是 [{tool_names}] 之一)
-    Action Input: 工具的 JSON 参数 (例如: {{"filePath": "src/index.ts"}})
-    
-    【⚠️ 极其重要】：当你输出 "Action Input" 后，必须【立即停止】输出，静静等待 Observation（工具返回结果）。严禁在此阶段输出 "Final Answer"。
+你必须严格遵守以下回复格式：
 
-2. **反馈阶段 (Observation)**:
-    Observation: 工具返回的真实数据。
+Thought: [此处写下你的思考过程]
+\`\`\`json
+{{
+  "action": "工具名称",
+  "action_input": {{ "参数名": "参数值" }}
+}}
+\`\`\`
 
-3. **最终结论阶段 (Final Answer)**:
-    当且仅当你已经从工具中获得了足够信息并完成所有审计任务时：
-    Thought: 我已经完成了所有分析，可以生成最终报告。
-    Final Answer: 任务总结陈述。
-
-### 🚫 强制禁止行为 (STRICT PROHIBITIONS)
---------------------
-- **严禁虚构**：禁止使用 'path/to/file' 等占位符，必须使用 'get_repo_map' 返回的真实路径。
-- **严禁冲突**：严禁在同一次回复中同时出现 "Action" 和 "Final Answer"。
-- **严禁对话**：不要向用户提问或进行闲聊，你的唯一目标是完成审计并调用 'generate_review'。
+注意：
+- 每次调用工具前必须先写 Thought。
+- 最终结论请使用 "action": "Final Answer"。
 
 Begin!
 Question: {input}
 Thought: {agent_scratchpad}`);
-        const agent = await (0, agents_1.createReactAgent)({ llm: model, tools: langchainTools, prompt });
+        const agent = await (0, agents_1.createStructuredChatAgent)({ llm: model, tools: langchainTools, prompt });
         this.executor = new agents_1.AgentExecutor({
             agent,
             tools: langchainTools,
-            verbose: false, // 我们已经有了 wrapHandler 日志，关闭原生 verbose 以保持整洁
+            verbose: this.verbose, // 我们已经有了 wrapHandler 日志，关闭原生 verbose 以保持整洁
             handleParsingErrors: true,
             maxIterations: this.maxIterations
         });
@@ -261,11 +227,23 @@ Thought: {agent_scratchpad}`);
                 // --- 新增：使用回调函数捕获 Thought ---
                 callbacks: [{
                         handleAgentAction: (action, runId) => {
-                            // 在 ReAct Agent 中，thought 通常包含在 log 字段中，且在 Action 之前
+                            // action.log 包含了 LLM 输出的所有文本（包含 Thought 和 JSON）
                             if (action.log) {
-                                const thought = action.log.split('Action:')[0].trim();
-                                if (thought) {
-                                    console.log(`\n💭 [思考]: ${thought.replace('Thought:', '').trim()}`);
+                                const log = action.log.trim();
+                                // 正则说明：匹配 Thought: 到 ```json 或 { 之间的内容
+                                const thoughtMatch = log.match(/Thought:\s*([\s\S]*?)(?=(?:```json|\{|$))/i);
+                                if (thoughtMatch && thoughtMatch[1]) {
+                                    const thought = thoughtMatch[1].trim();
+                                    if (thought) {
+                                        console.log(`\n💭 [思考]: ${thought}`);
+                                    }
+                                }
+                                else if (!log.startsWith('{') && !log.startsWith('```')) {
+                                    // 备选方案：如果没匹配到 Thought 标签，但有非 JSON 开头的文字，也打印出来
+                                    const rawThought = log.split(/```json|\{/)[0].trim();
+                                    if (rawThought) {
+                                        console.log(`\n💭 [思考]: ${rawThought}`);
+                                    }
                                 }
                             }
                         }
@@ -299,7 +277,7 @@ Thought: {agent_scratchpad}`);
         await this.init();
         const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
         console.log(`\n🚀 AI 助手启动 (LangChain 核心)`);
-        console.log(`📂 目标目录: ${this.engine.getRootDir()}`);
+        console.log(`📂 目标目录: ${this.targetDir}`);
         const chatLoop = () => {
             rl.question("\n👤 你: ", async (input) => {
                 if (!input.trim())
@@ -321,8 +299,8 @@ Thought: {agent_scratchpad}`);
     async ensureApiConfig() {
         if (this.apiConfig)
             return this.apiConfig;
-        if (fs_1.default.existsSync(CONFIG_FILE)) {
-            return JSON.parse(fs_1.default.readFileSync(CONFIG_FILE, "utf-8"));
+        if (fs_1.default.existsSync(config_1.CONFIG_FILE)) {
+            return JSON.parse(fs_1.default.readFileSync(config_1.CONFIG_FILE, "utf-8"));
         }
         const rl = readline.createInterface({
             input: process.stdin,
@@ -335,7 +313,7 @@ Thought: {agent_scratchpad}`);
             apiKey: await question("? API Key: "),
             model: await question("? Model Name (如 gpt-4o): "),
         };
-        fs_1.default.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
+        fs_1.default.writeFileSync(config_1.CONFIG_FILE, JSON.stringify(config, null, 2));
         rl.close();
         return config;
     }

package/lib/{agent.d.ts → core/agent.d.ts}

@@ -1,34 +1,14 @@
-interface ApiConfig {
-    baseURL: string;
-    apiKey: string;
-    model: string;
-}
-export interface CustomTool {
-    name: string;
-    description: string;
-    parameters: any;
-    handler: (args: any) => Promise<any>;
-}
-export interface AgentOptions {
-    targetDir?: string;
-    /** 自定义工具扩展 */
-    tools?: any[];
-    /** 注入到 System Prompt 中的额外指令/规则/上下文 */
-    extraSystemPrompt?: any;
-    maxTokens?: number;
-    apiConfig?: ApiConfig;
-}
+import { AgentOptions } from '../types/type';
 export default class McpAgent {
     private openai;
     private modelName;
-    private clients;
     private allTools;
     private messages;
-    private engine;
     private encoder;
     private extraTools;
     private maxTokens;
     private apiConfig;
+    private targetDir;
     constructor(options?: AgentOptions);
     /**
      * 计算当前消息列表的总 Token 消耗
@@ -37,8 +17,7 @@ export default class McpAgent {
     private calculateTokens;
     private injectCustomTools;
     /**
-     * 核心功能：内置代码分析工具
-     * 这里的逻辑直接调用 PromptEngine，不走网络请求，效率极高
+     * 核心功能：内置代码分析工具（基于 engine/targetDir，可被外部通过 createDefaultBuiltinTools 替代）
      */
     private registerBuiltinTools;
     private ensureApiConfig;
@@ -62,4 +41,3 @@ export default class McpAgent {
     chat(input: string): Promise<string>;
     start(): Promise<void>;
 }
-export {};

package/lib/{agent.js → core/agent.js}

@@ -36,25 +36,25 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-const openai_1 = __importDefault(require("openai"));
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
 const fs_1 = __importDefault(require("fs"));
-const path_1 = __importDefault(require("path"));
+const js_tiktoken_1 = require("js-tiktoken");
+const openai_1 = __importDefault(require("openai"));
 const os_1 = __importDefault(require("os"));
+const path_1 = __importDefault(require("path"));
 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");
+const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
+const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
+const builtin_1 = require("../tools/builtin");
+const config_1 = require("../config/config");
 class McpAgent {
     constructor(options) {
+        var _a;
         this.modelName = "";
-        this.clients = [];
         this.allTools = [];
         this.messages = [];
         this.encoder = (0, js_tiktoken_1.getEncoding)("cl100k_base");
         this.extraTools = [];
-        this.engine = new ts_context_mcp_1.PromptEngine((options === null || options === void 0 ? void 0 : options.targetDir) || process.cwd());
+        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
         this.apiConfig = options === null || options === void 0 ? void 0 : options.apiConfig;
@@ -87,8 +87,16 @@ class McpAgent {
             role: "system",
             content: baseSystemPrompt,
         });
-        // 初始化内置工具
-        this.registerBuiltinTools();
+        // 初始化内置工具：外部传入则使用，否则走默认 registerBuiltinTools
+        if ((_a = options === null || options === void 0 ? void 0 : options.builtinTools) === null || _a === void 0 ? void 0 : _a.length) {
+            this.allTools.push(...options.builtinTools);
+        }
+        else {
+            this.registerBuiltinTools({
+                ...options,
+                ...this,
+            });
+        }
         this.injectCustomTools(); // 注入外部工具
     }
     /**
@@ -147,146 +155,20 @@ class McpAgent {
         }
     }
     /**
-     * 核心功能：内置代码分析工具
-     * 这里的逻辑直接调用 PromptEngine，不走网络请求，效率极高
+     * 核心功能：内置代码分析工具（基于 engine/targetDir，可被外部通过 createDefaultBuiltinTools 替代）
      */
-    registerBuiltinTools() {
-        const builtinTools = [
-            {
-                type: "function",
-                function: {
-                    name: "get_repo_map",
-                    description: "获取项目全局文件结构及导出清单，用于快速定位代码",
-                    parameters: { type: "object", properties: {} },
-                },
-                _handler: async () => {
-                    this.engine.refresh();
-                    return this.engine.getRepoMap();
-                },
-            },
-            {
-                type: "function",
-                function: {
-                    name: "analyze_deps",
-                    description: "分析指定文件的依赖关系，支持 tsconfig 路径别名解析",
-                    parameters: {
-                        type: "object",
-                        properties: {
-                            filePath: { type: "string", description: "文件相对路径" },
-                        },
-                        required: ["filePath"],
-                    },
-                },
-                _handler: async ({ filePath }) => this.engine.getDeps(filePath),
-            },
-            {
-                type: "function",
-                function: {
-                    name: "read_skeleton",
-                    description: "提取文件的结构定义（接口、类、方法签名），忽略具体实现以节省 Token",
-                    parameters: {
-                        type: "object",
-                        properties: {
-                            filePath: { type: "string", description: "文件相对路径" },
-                        },
-                        required: ["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",
-                function: {
-                    name: "get_method_body",
-                    description: "获取指定文件内某个方法或函数的完整实现代码",
-                    parameters: {
-                        type: "object",
-                        properties: {
-                            filePath: { type: "string", description: "文件路径" },
-                            methodName: { type: "string", description: "方法名或函数名" },
-                        },
-                        required: ["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",
-                function: {
-                    name: "read_full_code",
-                    description: "读取指定文件的完整源代码内容。当需要分析具体实现逻辑或查找硬编码字符串时使用。",
-                    parameters: {
-                        type: "object",
-                        properties: {
-                            filePath: { type: "string", description: "文件相对路径" },
-                        },
-                        required: ["filePath"],
-                    },
-                },
-                // 核心实现：直接利用 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 不能为空";
-                        }
-                        // 拼合绝对路径
-                        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}`;
-                        }
-                        const content = fs_1.default.readFileSync(fullPath, "utf-8");
-                        // 加上行号，AI 就能在 generate_review 里给出准确的 line 参数
-                        return content.split('\n')
-                            .map((line, i) => `${i + 1} | ${line}`)
-                            .join('\n');
-                    }
-                    catch (err) {
-                        return `Error: 读取文件失败: ${err.message}`;
-                    }
-                },
-            },
-        ];
-        this.allTools.push(...builtinTools);
+    registerBuiltinTools(options) {
+        this.allTools.push(...(0, builtin_1.createDefaultBuiltinTools)({
+            options,
+            getCurrentTokens: () => this.calculateTokens(),
+        }));
     }
     // --- 初始化与环境准备 (API Config & MCP Servers) ---
     async ensureApiConfig() {
         if (this.apiConfig)
             return this.apiConfig;
-        if (fs_1.default.existsSync(CONFIG_FILE)) {
-            return JSON.parse(fs_1.default.readFileSync(CONFIG_FILE, "utf-8"));
+        if (fs_1.default.existsSync(config_1.CONFIG_FILE)) {
+            return JSON.parse(fs_1.default.readFileSync(config_1.CONFIG_FILE, "utf-8"));
         }
         const rl = readline.createInterface({
             input: process.stdin,
@@ -299,7 +181,7 @@ class McpAgent {
             apiKey: await question("? API Key: "),
             model: await question("? Model Name (如 gpt-4o): "),
         };
-        fs_1.default.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
+        fs_1.default.writeFileSync(config_1.CONFIG_FILE, JSON.stringify(config, null, 2));
         rl.close();
         return config;
     }
@@ -483,7 +365,7 @@ class McpAgent {
             input: process.stdin,
             output: process.stdout,
         });
-        console.log(`\n🚀 代码助手已启动 (目标目录: ${this.engine.getRootDir()})`);
+        console.log(`\n🚀 代码助手已启动 (目标目录: ${this.targetDir})`);
         const chatLoop = () => {
             rl.question("\n👤 你: ", async (input) => {
                 if (input.toLowerCase() === "exit")

package/lib/index.d.ts

@@ -1,3 +1,3 @@
-export * from './agent';
-export { default as McpChainAgent } from './agent-chain';
-export { default } from './agent';
+export * from './core/agent';
+export { default as McpChainAgent } from './core/agent-chain';
+export { default } from './core/agent';

package/lib/index.js

@@ -18,8 +18,8 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = exports.McpChainAgent = void 0;
-__exportStar(require("./agent"), exports);
-var agent_chain_1 = require("./agent-chain");
+__exportStar(require("./core/agent"), exports);
+var agent_chain_1 = require("./core/agent-chain");
 Object.defineProperty(exports, "McpChainAgent", { enumerable: true, get: function () { return __importDefault(agent_chain_1).default; } });
-var agent_1 = require("./agent");
+var agent_1 = require("./core/agent");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return __importDefault(agent_1).default; } });

package/lib/tools/builtin.d.ts

@@ -0,0 +1,7 @@
+import { AgentOptions, ToolInfo } from '../types/type';
+export interface BuiltinToolsContext {
+    options?: AgentOptions;
+    /** 可选：获取当前已用 token 数 */
+    getCurrentTokens?: () => number;
+}
+export declare function createDefaultBuiltinTools(context: BuiltinToolsContext): ToolInfo[];

package/lib/tools/builtin.js

@@ -0,0 +1,98 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createDefaultBuiltinTools = createDefaultBuiltinTools;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const ts_context_mcp_1 = require("@saber2pr/ts-context-mcp");
+const createTool_1 = require("../utils/createTool");
+function createDefaultBuiltinTools(context) {
+    const { options, getCurrentTokens } = context;
+    const engine = new ts_context_mcp_1.PromptEngine((options === null || options === void 0 ? void 0 : options.targetDir) || process.cwd());
+    const maxTokens = (options === null || options === void 0 ? void 0 : options.maxTokens) || 100000;
+    const rootDir = () => engine.getRootDir();
+    return [
+        (0, createTool_1.createTool)({
+            maxTokens,
+            getCurrentTokens,
+            name: "get_repo_map",
+            description: "获取项目全局文件结构及导出清单，用于快速定位代码",
+            parameters: { type: "object", properties: {} },
+            handler: async () => {
+                engine.refresh();
+                return engine.getRepoMap();
+            },
+        }),
+        (0, createTool_1.createTool)({
+            maxTokens,
+            getCurrentTokens,
+            name: "analyze_deps",
+            description: "分析指定文件的依赖关系，支持 tsconfig 路径别名解析",
+            parameters: { type: "object", properties: { filePath: { type: "string", description: "文件相对路径", required: true } } },
+            validateParams: ["filePath"],
+            handler: async ({ filePath }) => engine.getDeps(filePath),
+        }),
+        (0, createTool_1.createTool)({
+            maxTokens,
+            getCurrentTokens,
+            name: "read_skeleton",
+            description: "提取文件的结构定义（接口、类、方法签名），忽略具体实现以节省 Token",
+            parameters: { type: "object", properties: { filePath: { type: "string", description: "文件相对路径" } } },
+            handler: async (args) => {
+                const pathArg = args === null || args === void 0 ? void 0 : args.filePath;
+                if (typeof pathArg !== "string" || pathArg.trim() === "") {
+                    return `Error: 参数 'filePath' 无效。收到的是: ${JSON.stringify(pathArg)}`;
+                }
+                try {
+                    engine.refresh();
+                    const result = engine.getSkeleton(pathArg);
+                    return result || `// Warning: 文件 ${pathArg} 存在但未找到任何可提取的结构。`;
+                }
+                catch (error) {
+                    return `Error: 解析文件 ${pathArg} 时发生内部错误: ${error.message}`;
+                }
+            },
+        }),
+        (0, createTool_1.createTool)({
+            maxTokens,
+            getCurrentTokens,
+            name: "get_method_body",
+            description: "获取指定文件内某个方法或函数的完整实现代码",
+            parameters: { type: "object", properties: { filePath: { type: "string", description: "文件相对路径" }, methodName: { type: "string", description: "方法名或函数名" } } },
+            handler: async ({ filePath, methodName }) => {
+                return engine.getMethodImplementation(filePath, methodName);
+            },
+        }),
+        (0, createTool_1.createTool)({
+            maxTokens,
+            getCurrentTokens,
+            name: "read_full_code",
+            description: "读取指定文件的完整源代码内容。当需要分析具体实现逻辑或查找硬编码字符串时使用。",
+            parameters: { type: "object", properties: { filePath: { type: "string", description: "文件相对路径" } } },
+            handler: async ({ filePath }) => {
+                try {
+                    if (typeof filePath !== "string" || !filePath) {
+                        return "Error: filePath 不能为空";
+                    }
+                    const fullPath = path_1.default.resolve(rootDir(), filePath);
+                    if (!fullPath.startsWith(rootDir())) {
+                        return "Error: 权限拒绝，禁止访问项目目录外的文件。";
+                    }
+                    if (!fs_1.default.existsSync(fullPath)) {
+                        return `Error: 文件不存在: ${filePath}`;
+                    }
+                    const content = fs_1.default.readFileSync(fullPath, "utf-8");
+                    return content
+                        .split("\n")
+                        .map((line, i) => `${i + 1} | ${line}`)
+                        .join("\n");
+                }
+                catch (err) {
+                    return `Error: 读取文件失败: ${err.message}`;
+                }
+            },
+        }),
+    ];
+}

package/lib/types/type.d.ts

@@ -0,0 +1,45 @@
+import { BaseChatModel } from "@langchain/core/language_models/chat_models";
+import { Client } from "@modelcontextprotocol/sdk/client/index";
+export interface ApiConfig {
+    baseURL: string;
+    apiKey: string;
+    model: string;
+}
+export interface ToolInfo {
+    type: "function";
+    function: {
+        name: string;
+        description?: string;
+        parameters: any;
+    };
+    _handler?: (args: any) => Promise<any>;
+    _client?: Client;
+    _originalName?: string;
+}
+export interface CustomTool {
+    name: string;
+    description: string;
+    parameters: any;
+    handler: (args: any) => Promise<any>;
+}
+export interface McpConfig {
+    mcpServers: {
+        [key: string]: {
+            command: string;
+            args?: string[];
+            env?: Record<string, string>;
+        };
+    };
+}
+export interface AgentOptions {
+    targetDir?: string;
+    /** 外部传入的内置工具列表，不传则使用默认的 registerBuiltinTools */
+    builtinTools?: ToolInfo[];
+    tools?: CustomTool[];
+    extraSystemPrompt?: any;
+    maxTokens?: number;
+    apiConfig?: ApiConfig;
+    apiModel?: BaseChatModel;
+    maxIterations?: number;
+    verbose?: boolean;
+}

package/lib/types/type.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/lib/utils/createTool.d.ts

@@ -0,0 +1,11 @@
+import { ToolInfo } from "../types/type";
+export interface CreateToolOptions {
+    name: string;
+    description: string;
+    parameters: any;
+    handler: (args: any) => Promise<any>;
+    maxTokens: number;
+    getCurrentTokens: () => number;
+    validateParams?: string[];
+}
+export declare function createTool(options: CreateToolOptions): ToolInfo;

package/lib/utils/createTool.js

@@ -0,0 +1,39 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createTool = createTool;
+function createTool(options) {
+    return {
+        type: "function",
+        function: {
+            name: options.name,
+            description: options.description,
+            parameters: options.parameters,
+        },
+        _handler: async (input) => {
+            var _a;
+            if (options.getCurrentTokens && options.maxTokens != null && options.getCurrentTokens() > options.maxTokens) {
+                return `[SYSTEM WARNING]: Token 消耗已达上限，禁止获取详细方法体。请利用已获取的 Skeleton 信息进行分析。`;
+            }
+            // 兼容处理：如果 input 是字符串，尝试解析为 JSON 对象
+            let args = input;
+            if (typeof input === 'string') {
+                try {
+                    args = JSON.parse(input);
+                }
+                catch {
+                    args = input;
+                }
+            }
+            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])}`;
+                        }
+                    }
+                }
+            }
+            return options.handler(args);
+        },
+    };
+}

package/lib/utils/jsonSchemaToZod.d.ts

@@ -0,0 +1,2 @@
+import { z } from "zod";
+export declare function jsonSchemaToZod(parameters: any): z.ZodObject<{}, z.core.$loose>;

package/lib/utils/jsonSchemaToZod.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.jsonSchemaToZod = jsonSchemaToZod;
+const zod_1 = require("zod");
+// 将 JSON Schema 转换为 Zod Schema 的简易转换器
+function jsonSchemaToZod(parameters) {
+    var _a;
+    if (!parameters || !parameters.properties) {
+        return zod_1.z.object({}).loose();
+    }
+    const obj = {};
+    const properties = parameters.properties;
+    for (const key in properties) {
+        const prop = properties[key];
+        let schema = zod_1.z.any();
+        if (prop.type === "string") {
+            schema = zod_1.z.string();
+        }
+        else if (prop.type === "number") {
+            schema = zod_1.z.number();
+        }
+        else if (prop.type === "boolean") {
+            schema = zod_1.z.boolean();
+        }
+        if (prop.description) {
+            schema = schema.describe(prop.description);
+        }
+        // 默认都设为 optional 以防 Agent 报错，除非在 JSON Schema 的 required 数组中
+        const isRequired = (_a = parameters.required) === null || _a === void 0 ? void 0 : _a.includes(key);
+        obj[key] = isRequired ? schema : schema.optional();
+    }
+    return zod_1.z.object(obj).loose();
+}

package/package.json

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