@ai-zen/agents-sdk 0.2.0-alpha.2

package/dist/capabilities/implements/builtin/exec.js

@@ -0,0 +1,31 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import { exec } from "child_process";
+export const execTool = new CallbackTool({
+    function: {
+        name: "exec",
+        description: "执行命令",
+        parameters: {
+            type: "object",
+            properties: {
+                command: {
+                    type: "string",
+                    description: "要执行的命令",
+                },
+            },
+            required: ["command"],
+            additionalProperties: false,
+        },
+    },
+    async callback(input) {
+        const result = await new Promise((resolve) => {
+            exec(input.command, (error, stdout, stderr) => {
+                resolve({
+                    stdout,
+                    stderr,
+                    exitCode: error?.code ?? (error ? 1 : null),
+                });
+            });
+        });
+        return JSON.stringify(result);
+    },
+});

package/dist/capabilities/implements/builtin/exist.d.ts

@@ -0,0 +1,2 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+export declare const existTool: CallbackTool;

package/dist/capabilities/implements/builtin/exist.js

@@ -0,0 +1,25 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import * as fsp from "fs/promises";
+export const existTool = new CallbackTool({
+    function: {
+        name: "exist",
+        description: "检查文件或目录是否存在",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "文件或目录路径",
+                },
+            },
+            required: ["path"],
+            additionalProperties: false,
+        },
+    },
+    async callback(input) {
+        return await fsp
+            .access(input.path)
+            .then(() => "true")
+            .catch(() => "false");
+    },
+});

package/dist/capabilities/implements/builtin/findText.d.ts

@@ -0,0 +1,2 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+export declare const findTextTool: CallbackTool;

package/dist/capabilities/implements/builtin/findText.js

@@ -0,0 +1,86 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import * as fsp from "fs/promises";
+import * as path from "path";
+export const findTextTool = new CallbackTool({
+    function: {
+        name: "findText",
+        description: "查找文本出现的位置，支持普通文本或正则匹配，返回文件名及具体行号、行内容",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "glob cwd",
+                },
+                pattern: {
+                    type: "string",
+                    description: "glob pattern",
+                },
+                text: {
+                    type: "string",
+                    description: "要查找的文本（与 regex 二选一）",
+                },
+                regex: {
+                    type: "string",
+                    description: "正则表达式（与 text 二选一，例如 \\bconst\\s+\\w+）",
+                },
+                exclude: {
+                    type: "array",
+                    description: "要排除的 glob 模式数组",
+                    items: {
+                        type: "string",
+                    },
+                },
+            },
+            required: ["path", "pattern"],
+        },
+    },
+    async callback(input) {
+        try {
+            const result = [];
+            const text = input.text;
+            const regexStr = input.regex;
+            if (!text && !regexStr) {
+                return "请提供 text 或 regex 参数";
+            }
+            const regex = regexStr ? new RegExp(regexStr) : null;
+            for await (const file of fsp.glob(input.pattern, {
+                cwd: input.path,
+                exclude: input.exclude || ["**/node_modules/**"],
+            })) {
+                const fullPath = path.join(input.path, file);
+                const stats = await fsp.stat(fullPath);
+                if (stats.isFile()) {
+                    const content = await fsp.readFile(fullPath, "utf-8");
+                    const lines = content.split("\n");
+                    const matches = [];
+                    lines.forEach((lineContent, index) => {
+                        if (regex) {
+                            const matchResult = lineContent.match(regex);
+                            if (matchResult) {
+                                matches.push({
+                                    line: index + 1,
+                                    content: lineContent,
+                                    match: matchResult[0],
+                                });
+                            }
+                        }
+                        else if (lineContent.includes(text)) {
+                            matches.push({
+                                line: index + 1,
+                                content: lineContent,
+                            });
+                        }
+                    });
+                    if (matches.length > 0) {
+                        result.push({ file, matches });
+                    }
+                }
+            }
+            return JSON.stringify(result);
+        }
+        catch (error) {
+            return error?.message;
+        }
+    },
+});

package/dist/capabilities/implements/builtin/generateImage.d.ts

@@ -0,0 +1,9 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import type { AppConfig } from "../../../types/index.js";
+/**
+ * 根据文字描述生成图片。返回图片 URL，可通过 downloadFile 工具保存到本地。
+ *
+ * 依赖 AppConfig 中的 imageModels 和 defaultImageModel 配置。
+ * 如果未配置图片模型，工具会返回错误提示。
+ */
+export declare function createGenerateImageTool(config: AppConfig): CallbackTool;

package/dist/capabilities/implements/builtin/generateImage.js

@@ -0,0 +1,112 @@
+import { CallbackTool, OpenAI, ZhipuImage } from "@ai-zen/agents-core";
+/**
+ * 根据文字描述生成图片。返回图片 URL，可通过 downloadFile 工具保存到本地。
+ *
+ * 依赖 AppConfig 中的 imageModels 和 defaultImageModel 配置。
+ * 如果未配置图片模型，工具会返回错误提示。
+ */
+export function createGenerateImageTool(config) {
+    return new CallbackTool({
+        function: {
+            name: "generateImage",
+            description: "根据文字描述生成图片。返回图片 URL，可通过 downloadFile 工具保存到本地。",
+            parameters: {
+                type: "object",
+                properties: {
+                    prompt: {
+                        type: "string",
+                        description: "所需图像的文本描述，应详细描述画面内容、风格、构图等",
+                    },
+                    model: {
+                        type: "string",
+                        description: "图片模型 ID，不指定则使用配置中的默认图片模型。配置中的图片模型可通过 'aiz config show' 查看。",
+                    },
+                    size: {
+                        type: "string",
+                        description: '图片尺寸。不指定则使用模型的默认尺寸。cogview系列: 1024x1024, 768x1344, 864x1152, 1344x768, 1152x864, 1440x720, 720x1440。glm-image: 1280x1280, 1568x1056, 1056x1568, 1472x1088, 1088x1472, 1728x960, 960x1728。',
+                    },
+                    quality: {
+                        type: "string",
+                        description: "图像质量。hd: 精细(约20秒), standard: 快速(约5-10秒)",
+                        enum: ["hd", "standard"],
+                    },
+                },
+                required: ["prompt"],
+                additionalProperties: false,
+            },
+        },
+        async callback(input) {
+            try {
+                const prompt = input.prompt;
+                if (!prompt || !prompt.trim()) {
+                    return JSON.stringify({ success: false, error: "prompt 不能为空" });
+                }
+                const imageModels = config.imageModels;
+                const defaultImageModel = config.defaultImageModel;
+                if (!imageModels || imageModels.length === 0) {
+                    return JSON.stringify({
+                        success: false,
+                        error: "未配置图片生成模型。请在配置中添加 imageModels。",
+                    });
+                }
+                const modelId = input.model || defaultImageModel;
+                const imageModel = modelId
+                    ? imageModels.find((m) => m.id === modelId)
+                    : imageModels[0];
+                if (!imageModel) {
+                    return JSON.stringify({
+                        success: false,
+                        error: `图片模型 "${modelId}" 不存在。可用的图片模型: ${imageModels.map((m) => m.id).join(", ")}`,
+                    });
+                }
+                const endpoint = config.endpoints.find((e) => e.id === imageModel.endpointId);
+                if (!endpoint) {
+                    return JSON.stringify({
+                        success: false,
+                        error: `图片模型 "${imageModel.name}" 对应的端点 "${imageModel.endpointId}" 未配置`,
+                    });
+                }
+                if (!endpoint.apiKey) {
+                    return JSON.stringify({
+                        success: false,
+                        error: `端点 "${endpoint.name}" 的 API Key 未设置`,
+                    });
+                }
+                const ep = new OpenAI({
+                    openai_endpoint: endpoint.baseUrl,
+                    api_key: endpoint.apiKey,
+                });
+                const modelName = input.model || imageModel.modelName;
+                const model = new ZhipuImage({
+                    request_config: await ep.imageGeneration(modelName),
+                });
+                const result = await model.generate({
+                    prompt: prompt.trim(),
+                    model: modelName,
+                    size: input.size || imageModel.defaultSize || undefined,
+                    quality: input.quality || imageModel.defaultQuality || undefined,
+                });
+                const images = result.data.map((img, i) => ({
+                    index: i,
+                    url: img.url,
+                    description: `图片 ${i + 1}`,
+                }));
+                return JSON.stringify({
+                    success: true,
+                    model: imageModel.name,
+                    modelId: imageModel.id,
+                    created: result.created,
+                    images,
+                    content_filter: result.content_filter || [],
+                    note: "图片临时链接有效期为30天，请及时转存图片。",
+                }, null, 2);
+            }
+            catch (error) {
+                return JSON.stringify({
+                    success: false,
+                    error: error instanceof Error ? error.message : "图片生成失败",
+                });
+            }
+        },
+    });
+}

package/dist/capabilities/implements/builtin/glob.d.ts

@@ -0,0 +1,2 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+export declare const globTool: CallbackTool;

package/dist/capabilities/implements/builtin/glob.js

@@ -0,0 +1,45 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import * as fsp from "fs/promises";
+export const globTool = new CallbackTool({
+    function: {
+        name: "glob",
+        description: "使用 glob 模式递归扫描和查找文件。这是进行文件系统搜索的首选工具，功能远优于简单的 'ls' 列表命令。当你需要查找特定类型的文件、遍历目录树或需要排除特定文件时，请优先使用此函数。",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "glob cwd",
+                },
+                pattern: {
+                    type: "string",
+                    description: "glob pattern",
+                },
+                exclude: {
+                    type: "array",
+                    description: "glob pattern to exclude",
+                    items: {
+                        type: "string",
+                    },
+                },
+            },
+            required: ["pattern"],
+            additionalProperties: false,
+        },
+    },
+    async callback(input) {
+        try {
+            const result = [];
+            for await (const file of fsp.glob(input.pattern, {
+                exclude: input.exclude || ["**/node_modules/**"],
+                cwd: input.path,
+            })) {
+                result.push(file);
+            }
+            return JSON.stringify(result);
+        }
+        catch (error) {
+            return error?.message;
+        }
+    },
+});

package/dist/capabilities/implements/builtin/index.d.ts

@@ -0,0 +1,19 @@
+import type { Tool } from "@ai-zen/agents-core";
+import { cwdTool } from "./cwd.js";
+import { readFileTool } from "./readFile.js";
+import { writeFileTool } from "./writeFile.js";
+import { execTool } from "./exec.js";
+import { mkdirTool } from "./mkdir.js";
+import { rmTool } from "./rm.js";
+import { globTool } from "./glob.js";
+import { lsTool } from "./ls.js";
+import { existTool } from "./exist.js";
+import { findTextTool } from "./findText.js";
+import { downloadFileTool } from "./downloadFile.js";
+import { renameTool } from "./rename.js";
+import { copyTool } from "./copy.js";
+import { batchEditTool } from "./batchEdit.js";
+import { editTool } from "./edit.js";
+/** 15 个内置工具实例，注册与实现一体。 */
+export declare const BUILTIN_TOOLS: Tool[];
+export { cwdTool, readFileTool, writeFileTool, execTool, mkdirTool, rmTool, globTool, lsTool, existTool, findTextTool, downloadFileTool, renameTool, copyTool, batchEditTool, editTool };

package/dist/capabilities/implements/builtin/index.js

@@ -0,0 +1,34 @@
+import { cwdTool } from "./cwd.js";
+import { readFileTool } from "./readFile.js";
+import { writeFileTool } from "./writeFile.js";
+import { execTool } from "./exec.js";
+import { mkdirTool } from "./mkdir.js";
+import { rmTool } from "./rm.js";
+import { globTool } from "./glob.js";
+import { lsTool } from "./ls.js";
+import { existTool } from "./exist.js";
+import { findTextTool } from "./findText.js";
+import { downloadFileTool } from "./downloadFile.js";
+import { renameTool } from "./rename.js";
+import { copyTool } from "./copy.js";
+import { batchEditTool } from "./batchEdit.js";
+import { editTool } from "./edit.js";
+/** 15 个内置工具实例，注册与实现一体。 */
+export const BUILTIN_TOOLS = [
+    cwdTool,
+    readFileTool,
+    writeFileTool,
+    execTool,
+    mkdirTool,
+    rmTool,
+    globTool,
+    lsTool,
+    existTool,
+    findTextTool,
+    downloadFileTool,
+    renameTool,
+    copyTool,
+    batchEditTool,
+    editTool,
+];
+export { cwdTool, readFileTool, writeFileTool, execTool, mkdirTool, rmTool, globTool, lsTool, existTool, findTextTool, downloadFileTool, renameTool, copyTool, batchEditTool, editTool };

package/dist/capabilities/implements/builtin/ls.d.ts

@@ -0,0 +1,2 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+export declare const lsTool: CallbackTool;

package/dist/capabilities/implements/builtin/ls.js

@@ -0,0 +1,28 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import * as fsp from "fs/promises";
+export const lsTool = new CallbackTool({
+    function: {
+        name: "ls",
+        description: "列出目录",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "目录路径",
+                },
+            },
+            required: ["path"],
+            additionalProperties: false,
+        },
+    },
+    async callback(input) {
+        try {
+            const result = await fsp.readdir(input.path);
+            return JSON.stringify(result);
+        }
+        catch (error) {
+            return error?.message;
+        }
+    },
+});

package/dist/capabilities/implements/builtin/mkdir.d.ts

@@ -0,0 +1,2 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+export declare const mkdirTool: CallbackTool;

package/dist/capabilities/implements/builtin/mkdir.js

@@ -0,0 +1,34 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import * as fsp from "fs/promises";
+export const mkdirTool = new CallbackTool({
+    function: {
+        name: "mkdir",
+        description: "创建目录",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "目录路径",
+                },
+                recursive: {
+                    type: "boolean",
+                    description: "是否递归创建子目录",
+                },
+            },
+            required: ["path"],
+            additionalProperties: false,
+        },
+    },
+    async callback(input) {
+        try {
+            await fsp.mkdir(input.path, {
+                recursive: input.recursive,
+            });
+            return "success";
+        }
+        catch (error) {
+            return error?.message;
+        }
+    },
+});

package/dist/capabilities/implements/builtin/readFile.d.ts

@@ -0,0 +1,2 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+export declare const readFileTool: CallbackTool;

package/dist/capabilities/implements/builtin/readFile.js

@@ -0,0 +1,31 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import * as fsp from "fs/promises";
+export const readFileTool = new CallbackTool({
+    function: {
+        name: "readFile",
+        description: "读取文件",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "文件路径",
+                },
+            },
+            required: ["path"],
+            additionalProperties: false,
+        },
+    },
+    async callback(input) {
+        try {
+            const stats = await fsp.stat(input.path);
+            if (stats.size > 300 * 1024) {
+                throw new Error(`文件过大，无法读取，当前文件大小 ${stats.size} 字节`);
+            }
+            return await fsp.readFile(input.path, "utf-8");
+        }
+        catch (error) {
+            return error?.message;
+        }
+    },
+});

package/dist/capabilities/implements/builtin/rename.d.ts

@@ -0,0 +1,2 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+export declare const renameTool: CallbackTool;

package/dist/capabilities/implements/builtin/rename.js

@@ -0,0 +1,32 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import * as fsp from "fs/promises";
+export const renameTool = new CallbackTool({
+    function: {
+        name: "rename",
+        description: "重命名或移动文件/目录（相当于 mv 命令）",
+        parameters: {
+            type: "object",
+            properties: {
+                oldPath: {
+                    type: "string",
+                    description: "原路径",
+                },
+                newPath: {
+                    type: "string",
+                    description: "新路径",
+                },
+            },
+            required: ["oldPath", "newPath"],
+            additionalProperties: false,
+        },
+    },
+    async callback(input) {
+        try {
+            await fsp.rename(input.oldPath, input.newPath);
+            return "success";
+        }
+        catch (error) {
+            return error?.message;
+        }
+    },
+});

package/dist/capabilities/implements/builtin/rm.d.ts

@@ -0,0 +1,2 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+export declare const rmTool: CallbackTool;

package/dist/capabilities/implements/builtin/rm.js

@@ -0,0 +1,34 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import * as fsp from "fs/promises";
+export const rmTool = new CallbackTool({
+    function: {
+        name: "rm",
+        description: "删除文件或目录",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "文件或目录路径",
+                },
+                recursive: {
+                    type: "boolean",
+                    description: "是否递归删除子目录",
+                },
+            },
+            required: ["path"],
+            additionalProperties: false,
+        },
+    },
+    async callback(input) {
+        try {
+            await fsp.rm(input.path, {
+                recursive: input.recursive ?? false,
+            });
+            return "success";
+        }
+        catch (error) {
+            return error?.message;
+        }
+    },
+});

package/dist/capabilities/implements/builtin/writeFile.d.ts

@@ -0,0 +1,2 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+export declare const writeFileTool: CallbackTool;

package/dist/capabilities/implements/builtin/writeFile.js

@@ -0,0 +1,34 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import * as fsp from "fs/promises";
+import * as path from "path";
+export const writeFileTool = new CallbackTool({
+    function: {
+        name: "writeFile",
+        description: "写入文件",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "文件路径",
+                },
+                content: {
+                    type: "string",
+                    description: "文件内容",
+                },
+            },
+            required: ["path", "content"],
+            additionalProperties: false,
+        },
+    },
+    async callback(input) {
+        try {
+            await fsp.mkdir(path.dirname(input.path), { recursive: true });
+            await fsp.writeFile(input.path, input.content);
+            return "success";
+        }
+        catch (error) {
+            return error?.message;
+        }
+    },
+});

package/dist/capabilities/implements/mcpTools.d.ts

@@ -0,0 +1,20 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import type { McpConnectionManager } from "../../runtime/McpConnectionManager.js";
+import type { McpServerConfig } from "../../types/index.js";
+/**
+ * 创建 load_mcp 工具。
+ * server 枚举由 filteredMcps （完整 McpServerConfig[]）自动推导。
+ */
+export declare function createLoadMcpTool(mcpManager: McpConnectionManager, mcpConfigs: Map<string, {
+    name: string;
+    config: McpServerConfig;
+}>, filteredMcps: McpServerConfig[]): CallbackTool;
+/**
+ * 创建 call_mcp_tool 工具。
+ * 通过官方 Client.callTool() API 调用 MCP 服务器上的工具。
+ */
+export declare function createCallMcpTool(mcpManager: McpConnectionManager): CallbackTool;
+/**
+ * 创建 read_mcp_resource 工具。
+ */
+export declare function createReadMcpResourceTool(mcpManager: McpConnectionManager): CallbackTool;