@saber2pr/ai-agent 0.0.16 → 0.0.18

package/README.md

@@ -48,7 +48,7 @@ npm run build
 Best for quick scripts and simple chat interactions. It uses a straightforward message-loop logic.
 
 ```javascript
-const McpAgent = require("./lib/agent").default;
+import McpAgent from "@saber2pr/ai-agent";
 
 const agent = new McpAgent({
   targetDir: "/path/to/project"
@@ -63,10 +63,10 @@ await agent.chat("Analyze the directory structure.");
 Best for complex tasks like "Audit the whole project and fix bugs." It supports autonomous tool usage.
 
 ```javascript
-const McpAgent = require("./lib/agent-chain").default;
-const { MyPrivateLLM } = require("./your-custom-llm");
+import { McpChainAgent } from "@saber2pr/ai-agent";
+import { MyPrivateLLM } from "./your-custom-llm";
 
-const agent = new McpAgent({
+const agent = new McpChainAgent({
   apiModel: new MyPrivateLLM(), // Inject custom LLM
   maxIterations: 15,
   targetDir: "/path/to/project"
@@ -85,7 +85,7 @@ await agent.chat("Scan for hardcoded colors and submit a review report.");
 For LangChain mode, extend `AgentChainModel` which provides a simplified interface for integrating custom LLMs:
 
 ```javascript
-const { AgentChainModel } = require("@saber2pr/ai-agent");
+import { AgentChainModel } from "@saber2pr/ai-agent";
 
 class MyPrivateLLM extends AgentChainModel {
   constructor(fields) { 
@@ -175,6 +175,8 @@ The toolkit provides a comprehensive set of built-in tools organized into two ca
 You can pass structured rules via the `extraSystemPrompt`:
 
 ```javascript
+import McpAgent from "@saber2pr/ai-agent";
+
 const agent = new McpAgent({
   extraSystemPrompt: {
     role: "Code Auditor",

package/lib/core/agent-chain.js

@@ -54,9 +54,9 @@ class McpChainAgent {
         this.extraTools = [];
         this.runningTokenCounter = 0;
         this.extraTools = (options === null || options === void 0 ? void 0 : options.tools) || [];
-        this.maxTokens = (options === null || options === void 0 ? void 0 : options.maxTokens) || 100000;
+        this.maxTokens = (options === null || options === void 0 ? void 0 : options.maxTokens) || 8000;
         this.apiConfig = options === null || options === void 0 ? void 0 : options.apiConfig;
-        this.maxIterations = (options === null || options === void 0 ? void 0 : options.maxIterations) || 20;
+        this.maxIterations = (options === null || options === void 0 ? void 0 : options.maxIterations) || 50;
         this.apiModel = options === null || options === void 0 ? void 0 : options.apiModel;
         this.verbose = (options === null || options === void 0 ? void 0 : options.verbose) || false;
         const baseSystemPrompt = `你是一个专业的代码架构师。
@@ -114,7 +114,7 @@ class McpChainAgent {
         // maxTokenLimit 决定了当对话历史超过多少 Token 时触发“自动总结”
         this.memory = new memory_1.ConversationSummaryBufferMemory({
             llm: model,
-            maxTokenLimit: 2000,
+            maxTokenLimit: this.maxTokens,
             memoryKey: "chat_history",
             returnMessages: true,
             // 必须添加下面这两行显式声明：

package/lib/tools/filesystem/index.js

@@ -7,10 +7,10 @@ exports.getFilesystemTools = void 0;
 const zod_1 = require("zod");
 const promises_1 = __importDefault(require("fs/promises"));
 const path_1 = __importDefault(require("path"));
-const zod_to_json_schema_1 = require("zod-to-json-schema");
 const createTool_1 = require("../../utils/createTool");
 const minimatch_1 = require("minimatch");
 const lib_1 = require("./lib");
+const jsonSchemaToZod_1 = require("../../utils/jsonSchemaToZod");
 // Schema definitions
 const ReadTextFileArgsSchema = zod_1.z.object({
     path: zod_1.z.string(),
@@ -90,7 +90,7 @@ const getFilesystemTools = (targetDir) => {
             "the contents of a single file. Use the 'head' parameter to read only " +
             "the first N lines of a file, or the 'tail' parameter to read only " +
             "the last N lines of a file. Operates on the file as text regardless of extension.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(ReadTextFileArgsSchema, 'ReadTextFileArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(ReadTextFileArgsSchema, 'ReadTextFileArgsSchema'),
         validateParams: ["path"],
         handler: readTextFileHandler
     });
@@ -101,7 +101,7 @@ const getFilesystemTools = (targetDir) => {
             "or compare multiple files. Each file's content is returned with its " +
             "path as a reference. Failed reads for individual files won't stop " +
             "the entire operation. Only works within allowed directories.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(ReadMultipleFilesArgsSchema, 'ReadMultipleFilesArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(ReadMultipleFilesArgsSchema, 'ReadMultipleFilesArgsSchema'),
         validateParams: ["paths"],
         handler: async (args) => {
             const results = await Promise.all(args.paths.map(async (filePath) => {
@@ -124,7 +124,7 @@ const getFilesystemTools = (targetDir) => {
         description: "Create a new file or completely overwrite an existing file with new content. " +
             "Use with caution as it will overwrite existing files without warning. " +
             "Handles text content with proper encoding. Only works within allowed directories.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(WriteFileArgsSchema, 'WriteFileArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(WriteFileArgsSchema, 'WriteFileArgsSchema'),
         validateParams: ["path", "content"],
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
@@ -138,7 +138,7 @@ const getFilesystemTools = (targetDir) => {
         description: "Make line-based edits to a text file. Each edit replaces exact line sequences " +
             "with new content. Returns a git-style diff showing the changes made. " +
             "Only works within allowed directories.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(EditFileArgsSchema, 'EditFileArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(EditFileArgsSchema, 'EditFileArgsSchema'),
         validateParams: ["path", "edits"],
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
@@ -152,7 +152,7 @@ const getFilesystemTools = (targetDir) => {
             "nested directories in one operation. If the directory already exists, " +
             "this operation will succeed silently. Perfect for setting up directory " +
             "structures for projects or ensuring required paths exist. Only works within allowed directories.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(CreateDirectoryArgsSchema, 'CreateDirectoryArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(CreateDirectoryArgsSchema, 'CreateDirectoryArgsSchema'),
         validateParams: ["path"],
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
@@ -167,7 +167,7 @@ const getFilesystemTools = (targetDir) => {
             "Results clearly distinguish between files and directories with [FILE] and [DIR] " +
             "prefixes. This tool is essential for understanding directory structure and " +
             "finding specific files within a directory. Only works within allowed directories.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(ListDirectoryArgsSchema, 'ListDirectoryArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(ListDirectoryArgsSchema, 'ListDirectoryArgsSchema'),
         validateParams: ["path"],
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
@@ -184,7 +184,7 @@ const getFilesystemTools = (targetDir) => {
             "Results clearly distinguish between files and directories with [FILE] and [DIR] " +
             "prefixes. This tool is useful for understanding directory structure and " +
             "finding specific files within a directory. Only works within allowed directories.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(ListDirectoryWithSizesArgsSchema, 'ListDirectoryWithSizesArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(ListDirectoryWithSizesArgsSchema, 'ListDirectoryWithSizesArgsSchema'),
         validateParams: ["path"],
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
@@ -239,7 +239,7 @@ const getFilesystemTools = (targetDir) => {
             "Each entry includes 'name', 'type' (file/directory), and 'children' for directories. " +
             "Files have no children array, while directories always have a children array (which may be empty). " +
             "The output is formatted with 2-space indentation for readability. Only works within allowed directories.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(DirectoryTreeArgsSchema, 'DirectoryTreeArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(DirectoryTreeArgsSchema, 'DirectoryTreeArgsSchema'),
         validateParams: ["path"],
         handler: async (args) => {
             const rootPath = args.path;
@@ -284,7 +284,7 @@ const getFilesystemTools = (targetDir) => {
             "and rename them in a single operation. If the destination exists, the " +
             "operation will fail. Works across different directories and can be used " +
             "for simple renaming within the same directory. Both source and destination must be within allowed directories.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(MoveFileArgsSchema, 'MoveFileArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(MoveFileArgsSchema, 'MoveFileArgsSchema'),
         validateParams: ["source", "destination"],
         handler: async (args) => {
             const validSourcePath = await (0, lib_1.validatePath)(targetDir, args.source);
@@ -298,7 +298,7 @@ const getFilesystemTools = (targetDir) => {
         name: "search_files",
         description: "Search for files matching a specific pattern in a specified path. " +
             "Returns a list of files that match the pattern. Only works within allowed directories.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(SearchFilesArgsSchema, 'SearchFilesArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(SearchFilesArgsSchema, 'SearchFilesArgsSchema'),
         validateParams: ["path", "pattern"],
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
@@ -311,7 +311,7 @@ const getFilesystemTools = (targetDir) => {
         name: "get_file_info",
         description: "Get detailed information about a file, including its size, last modified time, and type. " +
             "Only works within allowed directories.",
-        parameters: (0, zod_to_json_schema_1.zodToJsonSchema)(GetFileInfoArgsSchema, 'GetFileInfoArgsSchema'),
+        parameters: (0, jsonSchemaToZod_1.zodObjectToJsonSchema)(GetFileInfoArgsSchema, 'GetFileInfoArgsSchema'),
         validateParams: ["path"],
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);

package/lib/utils/jsonSchemaToZod.d.ts

@@ -1,2 +1,12 @@
 import { z } from "zod";
+export declare const zodObjectToJsonSchema: (zodObject: z.ZodObject<any>, name: string) => z.ZodObject<any, z.UnknownKeysParam, z.ZodTypeAny, {
+    [x: string]: any;
+}, {
+    [x: string]: any;
+}> | {
+    title?: string;
+    default?: any;
+    description?: string;
+    markdownDescription?: string;
+};
 export declare function jsonSchemaToZod(parameters: any): z.ZodObject<any>;

package/lib/utils/jsonSchemaToZod.js

@@ -1,7 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.zodObjectToJsonSchema = void 0;
 exports.jsonSchemaToZod = jsonSchemaToZod;
 const zod_1 = require("zod");
+const zod_to_json_schema_1 = require("zod-to-json-schema");
+const zodObjectToJsonSchema = (zodObject, name) => {
+    var _a, _b;
+    return ((_b = (_a = (0, zod_to_json_schema_1.zodToJsonSchema)(zodObject, name)) === null || _a === void 0 ? void 0 : _a.definitions) === null || _b === void 0 ? void 0 : _b[name]) || zodObject;
+};
+exports.zodObjectToJsonSchema = zodObjectToJsonSchema;
 function jsonSchemaToZod(parameters) {
     var _a;
     if (!parameters || typeof parameters !== 'object') {

package/package.json

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