@saber2pr/ai-agent 0.0.57 → 0.0.59

package/lib/adapters/llm.d.ts

@@ -1,11 +1,9 @@
-import { AgentGraphLLMResponse, AgentGraphModel, StreamChunkCallback } from '../model/AgentGraphModel';
 import { BaseMessage } from '@langchain/core/messages';
+import { AgentGraphLLMResponse, AgentGraphModel, StreamChunkCallback } from '../model/AgentGraphModel';
 import { CreateAgentOptions } from '../types/type';
 export declare class LLMModel extends AgentGraphModel {
-    private chatId;
     private options;
     constructor(options: CreateAgentOptions);
-    resetChat(): void;
     callApi(prompt: string): Promise<AgentGraphLLMResponse>;
     /**
      * 流式调用 API：发送 stream: true，自动适配多种响应格式（SSE / NDJSON / 普通 JSON）。

package/lib/adapters/llm.js

@@ -3,16 +3,13 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.LLMModel = void 0;
 const AgentGraphModel_1 = require("../model/AgentGraphModel");
 class LLMModel extends AgentGraphModel_1.AgentGraphModel {
-    chatId;
     options;
     constructor(options) {
-        super();
-        this.chatId = '';
+        super({
+            targetDir: options.targetDir,
+        });
         this.options = options;
     }
-    resetChat() {
-        this.chatId = '';
-    }
     async callApi(prompt) {
         const response = await fetch(this.options.apiUrl, {
             method: 'POST',
@@ -172,7 +169,7 @@ class LLMModel extends AgentGraphModel_1.AgentGraphModel {
             text: fullText,
             reasoning,
             token,
-            duration
+            duration,
         };
     }
 }

package/lib/core/agent-graph.js

@@ -175,7 +175,7 @@ class McpGraphAgent {
         const mcpToolInfos = await this.initMcpTools();
         // 合并内置、手动传入和 MCP 工具
         const allToolInfos = [...builtinToolInfos, ...(this.options.tools || []), ...mcpToolInfos];
-        this.langchainTools = allToolInfos.map(t => (0, convertToLangChainTool_1.convertToLangChainTool)(t));
+        this.langchainTools = allToolInfos.map(t => (0, convertToLangChainTool_1.convertToLangChainTool)(t, { allTools: allToolInfos }));
         this.toolNode = new prebuilt_1.ToolNode(this.langchainTools);
         return {
             builtinToolInfos,

package/lib/core/agent.js

@@ -306,7 +306,7 @@ class McpAgent {
                 console.log(`   📦 传入参数: \x1b[2m${JSON.stringify(args)}\x1b[0m`);
                 let result;
                 if (tool?._handler) {
-                    result = await tool._handler(args);
+                    result = await tool._handler(args, { allTools: this.allTools });
                 }
                 else if (tool?._client && tool._originalName) {
                     const mcpRes = await tool._client.callTool({

package/lib/model/AgentGraphModel.d.ts

@@ -13,12 +13,17 @@ export declare abstract class AgentGraphModel extends BaseChatModel {
     protected boundTools?: any[];
     private mcpEnabled?;
     private mcpTools?;
+    protected chatId: string;
+    protected targetDir?: string;
+    resetChat(): void;
     setMcpTools(tools: any[]): void;
     setMcpEnabled(enabled?: boolean): void;
     getMcpEnabled(): boolean;
     getMcpTools(): any[];
     private isMcpTool;
-    constructor(fields?: BaseChatModelParams);
+    constructor(fields?: BaseChatModelParams & {
+        targetDir?: string;
+    });
     bindTools(tools: any[]): any;
     abstract callApi(prompt: string, lastMsg: BaseMessage): Promise<AgentGraphLLMResponse>;
     /**

package/lib/model/AgentGraphModel.js

@@ -6,10 +6,16 @@ const messages_1 = require("@langchain/core/messages");
 const function_calling_1 = require("@langchain/core/utils/function_calling");
 const generateToolMarkdown_1 = require("../utils/generateToolMarkdown");
 const kit_1 = require("../utils/kit");
+const getRepoMapPrompt_1 = require("../utils/getRepoMapPrompt");
 class AgentGraphModel extends chat_models_1.BaseChatModel {
     boundTools;
     mcpEnabled = true;
     mcpTools = [];
+    chatId = '';
+    targetDir;
+    resetChat() {
+        this.chatId = '';
+    }
     setMcpTools(tools) {
         this.mcpTools = tools;
     }
@@ -28,6 +34,10 @@ class AgentGraphModel extends chat_models_1.BaseChatModel {
     }
     constructor(fields) {
         super(fields || {});
+        this.chatId = '';
+        this.mcpTools = [];
+        this.mcpEnabled = true;
+        this.targetDir = fields?.targetDir;
     }
     bindTools(tools) {
         this.boundTools = tools.map(t => (0, function_calling_1.convertToOpenAITool)(t));
@@ -170,6 +180,7 @@ class AgentGraphModel extends chat_models_1.BaseChatModel {
     serializeMessages(messages) {
         const systemMsg = messages.find(m => m._getType() === 'system');
         const lastMsg = messages[messages.length - 1];
+        const isFirstMessage = this.chatId === '';
         const format = (m) => {
             const content = typeof m.content === 'string' ? m.content : JSON.stringify(m.content, null, 2);
             return `${m._getType().toUpperCase()}: ${content}`;
@@ -178,9 +189,21 @@ class AgentGraphModel extends chat_models_1.BaseChatModel {
         const toolsContext = tools.length
             ? `${(0, generateToolMarkdown_1.generateToolMarkdown)(tools)}`
             : '';
-        return `
-${format(systemMsg)}
+        const systemContext = isFirstMessage ? `
+${isFirstMessage ? format(systemMsg) : ''}
 ${toolsContext}
+${(0, getRepoMapPrompt_1.getIncrementalRepoMapPrompt)(this.targetDir, isFirstMessage)}
+`.trim() : `
+## Active Tools (Summary)
+Available: ${(0, kit_1.getArray)(tools).map(t => t.function.name).join(', ')}
+
+**Self-Correction**: If you encounter "tool not found" or need parameter details, you MUST call:
+- \`get_all_tools_schema\`: Retrieve full tool definitions and schemas.
+
+${(0, getRepoMapPrompt_1.getIncrementalRepoMapPrompt)(this.targetDir, isFirstMessage)}
+`.trim();
+        return `
+${systemContext}
 # Current Progress
 ${format(lastMsg)}
 # Output Requirement

package/lib/tools/builtin.js

@@ -2,11 +2,13 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createDefaultBuiltinTools = createDefaultBuiltinTools;
 const filesystem_1 = require("./filesystem");
+const get_all_tools_schema_1 = require("./loader/get_all_tools_schema");
 const ts_lsp_1 = require("./ts-lsp");
 function createDefaultBuiltinTools(context) {
     const { options } = context;
     return [
         ...(0, ts_lsp_1.getTsLspTools)(options?.targetDir || process.cwd()),
         ...(0, filesystem_1.getFilesystemTools)(options?.targetDir || process.cwd()),
+        get_all_tools_schema_1.getAllToolsSchema,
     ];
 }

package/lib/tools/loader/get_all_tools_schema.d.ts

@@ -0,0 +1 @@
+export declare const getAllToolsSchema: import("../..").ToolInfo;

package/lib/tools/loader/get_all_tools_schema.js

@@ -0,0 +1,33 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getAllToolsSchema = void 0;
+const zod_1 = require("zod");
+const zod_to_json_schema_1 = __importDefault(require("zod-to-json-schema"));
+const createTool_1 = require("../../utils/createTool");
+const generateToolMarkdown_1 = require("../../utils/generateToolMarkdown");
+const kit_1 = require("../../utils/kit");
+exports.getAllToolsSchema = (0, createTool_1.createTool)({
+    name: 'get_all_tools_schema',
+    description: 'Use this tool when you encounter a "tool not found" error or are unsure about the parameter schema. It retrieves the full definitions and JSON schemas for all available tools in the current environment.',
+    parameters: zod_1.z.object({
+        toolName: zod_1.z.string().optional().describe('Optional: Specify a tool name to get its detailed schema. If omitted, returns all available tools.')
+    }),
+    // 增加第二个参数 context
+    handler: async ({ toolName }, context) => {
+        // 这里的 context.allTools 是在运行时从 Agent 实例传入的
+        const availableTools = (0, kit_1.getArray)(context?.allTools);
+        const targetTools = toolName
+            ? availableTools.filter(t => t.function.name === toolName)
+            : availableTools;
+        return (0, generateToolMarkdown_1.generateToolMarkdown)(targetTools.map(item => ({
+            ...item,
+            function: {
+                ...item.function,
+                parameters: (0, zod_to_json_schema_1.default)(item.function.parameters)
+            }
+        })));
+    }
+});

package/lib/types/type.d.ts

@@ -13,7 +13,9 @@ export interface ToolInfo {
         description?: string;
         parameters: z.ZodObject<any>;
     };
-    _handler?: (args: any) => Promise<any>;
+    _handler?: (args: any, context: {
+        allTools: ToolInfo[];
+    }) => Promise<any>;
     _client?: Client;
     _originalName?: string;
 }

package/lib/utils/convertToLangChainTool.d.ts

@@ -1,6 +1,8 @@
 import { DynamicStructuredTool } from '@langchain/core/tools';
 import { ToolInfo } from '../types/type';
-export declare function convertToLangChainTool(info: ToolInfo): DynamicStructuredTool<import("zod").ZodObject<any, import("zod").UnknownKeysParam, import("zod").ZodTypeAny, {
+export declare function convertToLangChainTool(info: ToolInfo, context: {
+    allTools: ToolInfo[];
+}): DynamicStructuredTool<import("zod").ZodObject<any, import("zod").UnknownKeysParam, import("zod").ZodTypeAny, {
     [x: string]: any;
 }, {
     [x: string]: any;

package/lib/utils/convertToLangChainTool.js

@@ -2,14 +2,14 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.convertToLangChainTool = convertToLangChainTool;
 const tools_1 = require("@langchain/core/tools");
-function convertToLangChainTool(info) {
+function convertToLangChainTool(info, context) {
     return new tools_1.DynamicStructuredTool({
         name: info.function.name,
         description: info.function.description || '',
         schema: info.function.parameters,
         func: async (args) => {
             if (info._handler)
-                return await info._handler(args);
+                return await info._handler(args, context);
             if (info._client && info._originalName) {
                 const result = await info._client.callTool({
                     name: info._originalName,

package/lib/utils/createTool.d.ts

@@ -7,6 +7,8 @@ export interface CreateToolOptions {
      * zod@3.25
      */
     parameters: z.ZodObject<any>;
-    handler: (args: any) => Promise<string>;
+    handler: (args: any, context: {
+        allTools: ToolInfo[];
+    }) => Promise<string>;
 }
 export declare function createTool(options: CreateToolOptions): ToolInfo;

package/lib/utils/createTool.js

@@ -9,7 +9,7 @@ function createTool(options) {
             description: options.description,
             parameters: options.parameters,
         },
-        _handler: async (input) => {
+        _handler: async (input, context) => {
             // 兼容处理：如果 input 是字符串，尝试解析为 JSON 对象
             let args = input;
             if (typeof input === 'string') {
@@ -20,7 +20,7 @@ function createTool(options) {
                     args = input;
                 }
             }
-            return options.handler(args);
+            return options.handler(args, context);
         },
     };
 }

package/lib/utils/getRepoMapPrompt.d.ts

@@ -0,0 +1 @@
+export declare const getIncrementalRepoMapPrompt: (targetDir: string, isFirstTurn: boolean) => string;

package/lib/utils/getRepoMapPrompt.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getIncrementalRepoMapPrompt = void 0;
+const ts_context_mcp_1 = require("@saber2pr/ts-context-mcp");
+let lastRepoMapContent = '';
+const getIncrementalRepoMapPrompt = (targetDir, isFirstTurn) => {
+    try {
+        const engine = new ts_context_mcp_1.PromptEngine(targetDir);
+        const currentContent = engine.getRepoMap();
+        // 1. 第一轮对话：必须全量发送
+        if (isFirstTurn || !lastRepoMapContent) {
+            lastRepoMapContent = currentContent;
+            return `\n# Initial Project Repository Map\n\`\`\`text\n${currentContent}\n\`\`\``;
+        }
+        // 2. 如果完全没变：发送静默占位符
+        if (currentContent === lastRepoMapContent) {
+            return "\n\n(Note: Project structure remains unchanged.)";
+        }
+        // 3. 如果变了：计算增量 (Incremental Diff)
+        const lastLines = lastRepoMapContent.split('\n');
+        const currentLines = currentContent.split('\n');
+        // 找出发生变化的行 (简单示例：找出在 current 中但不在 last 中的行)
+        const changes = currentLines.filter(line => !lastLines.includes(line));
+        lastRepoMapContent = currentContent;
+        if (changes.length > 0) {
+            return `
+# 🆕 Project Map Updates (Incremental)
+The project structure has changed. Key updates:
+\`\`\`text
+${changes.join('\n')}
+\`\`\`
+*(Refer to previous messages for the rest of the map)*`;
+        }
+        return "\n\n(Note: Minor internal structure changes detected.)";
+    }
+    catch (error) {
+        return '';
+    }
+};
+exports.getIncrementalRepoMapPrompt = getIncrementalRepoMapPrompt;

package/package.json

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