@saber2pr/ai-agent 0.0.5 → 0.0.6

package/lib/agent.d.ts

@@ -1,3 +1,16 @@
+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;
+}
 export default class McpAgent {
     private openai;
     private modelName;
@@ -6,12 +19,14 @@ export default class McpAgent {
     private messages;
     private engine;
     private encoder;
-    constructor();
+    private extraTools;
+    constructor(options?: AgentOptions);
     /**
      * 计算当前消息列表的总 Token 消耗
      * 兼容多模态内容 (Content Parts) 和 工具调用 (Tool Calls)
      */
     private calculateTokens;
+    private injectCustomTools;
     /**
      * 核心功能：内置代码分析工具
      * 这里的逻辑直接调用 PromptEngine，不走网络请求，效率极高
@@ -25,5 +40,11 @@ export default class McpAgent {
      * 简易 Loading 动画辅助函数
      */
     private showLoading;
+    /**
+     * 编程式对话入口
+     * @param input 用户指令
+     * @returns AI 的最终答复内容
+     */
+    chat(input: string): Promise<string>;
     start(): Promise<void>;
 }

package/lib/agent.js

@@ -47,24 +47,34 @@ 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() {
+    constructor(options) {
         this.modelName = "";
         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({
-            role: "system",
-            content: `你是一个专业的 AI 代码架构师。
+        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。`,
+请优先使用 read_skeleton 查看结构，只有在必要时才使用 read_full_code 或 get_method_body。`;
+        // 2. 拼接额外指令
+        if (options === null || options === void 0 ? void 0 : options.extraSystemPrompt) {
+            const extra = typeof options.extraSystemPrompt === 'string'
+                ? options.extraSystemPrompt
+                : JSON.stringify(options.extraSystemPrompt, null, 2);
+            baseSystemPrompt += `\n\n[额外执行指令]:\n${extra}`;
+        }
+        this.messages.push({
+            role: "system",
+            content: baseSystemPrompt,
         });
         // 初始化内置工具
         this.registerBuiltinTools();
+        this.injectCustomTools(); // 注入外部工具
     }
     /**
      * 计算当前消息列表的总 Token 消耗
@@ -108,6 +118,19 @@ class McpAgent {
         }
         return total;
     }
+    injectCustomTools() {
+        for (const tool of this.extraTools) {
+            this.allTools.push({
+                type: "function",
+                function: {
+                    name: tool.name,
+                    description: tool.description,
+                    parameters: tool.parameters,
+                },
+                _handler: tool.handler,
+            });
+        }
+    }
     /**
      * 核心功能：内置代码分析工具
      * 这里的逻辑直接调用 PromptEngine，不走网络请求，效率极高
@@ -218,7 +241,11 @@ class McpAgent {
                         if (!fs_1.default.existsSync(fullPath)) {
                             return `Error: 文件不存在: ${filePath}`;
                         }
-                        return fs_1.default.readFileSync(fullPath, "utf-8");
+                        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}`;
@@ -376,8 +403,27 @@ class McpAgent {
             process.stdout.write('\r\x1b[K'); // 清除当前行
         };
     }
+    /**
+     * 编程式对话入口
+     * @param input 用户指令
+     * @returns AI 的最终答复内容
+     */
+    async chat(input) {
+        if (!this.openai) {
+            await this.init();
+        }
+        // 调用现有的处理逻辑
+        // 假设你的 processChat 已经处理了所有的 tool_calls 循环
+        await this.processChat(input);
+        // 返回消息列表中的最后一条 AI 回复
+        const lastMsg = this.messages[this.messages.length - 1];
+        return lastMsg.role === 'assistant' ? lastMsg.content : '';
+    }
+    // 修改原来的 start 方法，使其内部也调用 chat
     async start() {
-        await this.init();
+        if (!this.openai) {
+            await this.init();
+        }
         const rl = readline.createInterface({
             input: process.stdin,
             output: process.stdout,
@@ -388,7 +434,8 @@ class McpAgent {
                 if (input.toLowerCase() === "exit")
                     process.exit(0);
                 try {
-                    await this.processChat(input);
+                    // 这里统一调用 chat 或核心逻辑
+                    await this.chat(input);
                 }
                 catch (err) {
                     console.error("\n❌ 系统错误:", err.message);

package/lib/cli.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/lib/cli.js

@@ -0,0 +1,9 @@
+#!/usr/bin/env node
+"use strict";
+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();

package/lib/index.d.ts

@@ -1,2 +1,2 @@
-#!/usr/bin/env node
-export {};
+export * from './agent';
+export { default } from './agent';

package/lib/index.js

@@ -1,9 +1,23 @@
-#!/usr/bin/env node
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
 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();
+exports.default = void 0;
+__exportStar(require("./agent"), exports);
+var agent_1 = require("./agent");
+Object.defineProperty(exports, "default", { enumerable: true, get: function () { return __importDefault(agent_1).default; } });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saber2pr/ai-agent",
-  "version": "0.0.5",
+  "version": "0.0.6",
   "description": "AI Assistant CLI.",
   "author": "saber2pr",
   "license": "ISC",
@@ -8,7 +8,7 @@
     "lib"
   ],
   "bin": {
-    "sagent": "./lib/index.js"
+    "sagent": "./lib/cli.js"
   },
   "publishConfig": {
     "access": "public",