@saber2pr/ai-agent 0.0.10 → 0.0.11

package/lib/tools/ts-lsp/index.js

@@ -0,0 +1,56 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getTsLspTools = void 0;
+const ts_context_mcp_1 = require("@saber2pr/ts-context-mcp");
+const createTool_1 = require("../../utils/createTool");
+const getTsLspTools = (targetDir) => {
+    const engine = new ts_context_mcp_1.PromptEngine(targetDir);
+    return [
+        (0, createTool_1.createTool)({
+            name: "get_repo_map",
+            description: "获取项目全局文件结构及导出清单，用于快速定位代码",
+            parameters: { type: "object", properties: {} },
+            handler: async () => {
+                engine.refresh();
+                return engine.getRepoMap();
+            },
+        }),
+        (0, createTool_1.createTool)({
+            name: "analyze_deps",
+            description: "分析指定文件的依赖关系，支持 tsconfig 路径别名解析",
+            parameters: { type: "object", properties: { filePath: { type: "string", description: "文件相对路径" } } },
+            validateParams: ["filePath"],
+            handler: async ({ filePath }) => engine.getDeps(filePath),
+        }),
+        (0, createTool_1.createTool)({
+            name: "read_skeleton",
+            description: "提取文件的结构定义（接口、类、方法签名），忽略具体实现以节省 Token",
+            parameters: { type: "object", properties: { filePath: { type: "string", description: "文件相对路径" } } },
+            validateParams: ["filePath"],
+            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)({
+            name: "get_method_body",
+            description: "获取指定文件内某个方法或函数的完整实现代码",
+            parameters: { type: "object", properties: { filePath: { type: "string", description: "文件相对路径" }, methodName: { type: "string", description: "方法名或函数名" } } },
+            validateParams: ["filePath", 'methodName'],
+            handler: async ({ filePath, methodName }) => {
+                return engine.getMethodImplementation(filePath, methodName);
+            },
+        }),
+    ];
+};
+exports.getTsLspTools = getTsLspTools;

package/lib/types/type.d.ts

@@ -16,12 +16,6 @@ export interface ToolInfo {
     _client?: Client;
     _originalName?: string;
 }
-export interface CustomTool {
-    name: string;
-    description: string;
-    parameters: any;
-    handler: (args: any) => Promise<any>;
-}
 export interface McpConfig {
     mcpServers: {
         [key: string]: {
@@ -34,8 +28,7 @@ export interface McpConfig {
 export interface AgentOptions {
     targetDir?: string;
     /** 外部传入的内置工具列表，不传则使用默认的 registerBuiltinTools */
-    builtinTools?: ToolInfo[];
-    tools?: CustomTool[];
+    tools?: ToolInfo[];
     extraSystemPrompt?: any;
     maxTokens?: number;
     apiConfig?: ApiConfig;

package/lib/utils/createTool.d.ts

@@ -3,9 +3,7 @@ export interface CreateToolOptions {
     name: string;
     description: string;
     parameters: any;
-    handler: (args: any) => Promise<any>;
-    maxTokens: number;
-    getCurrentTokens: () => number;
+    handler: (args: any) => Promise<string>;
     validateParams?: string[];
 }
 export declare function createTool(options: CreateToolOptions): ToolInfo;

package/lib/utils/createTool.js

@@ -11,9 +11,6 @@ function createTool(options) {
         },
         _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') {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saber2pr/ai-agent",
-  "version": "0.0.10",
+  "version": "0.0.11",
   "description": "AI Assistant CLI.",
   "author": "saber2pr",
   "license": "ISC",
@@ -28,7 +28,10 @@
     "@saber2pr/ts-context-mcp": "^0.0.6",
     "js-tiktoken": "^1.0.21",
     "langchain": "~0.3",
-    "openai": "^6.16.0"
+    "openai": "^6.16.0",
+    "diff": "^8.0.3",
+    "glob": "^10.5.0",
+    "minimatch": "^10.0.1"
   },
   "devDependencies": {
     "@types/node": "^16.3.3",