@ai-zen/agents-sdk 0.5.6 → 0.6.0

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

@@ -1,25 +1,23 @@
 import * as fsp from "fs/promises";
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 export class ExistTool extends SdkCallbackTool {
-    constructor(env) {
-        super({
-            function: {
-                name: "exist",
-                description: "检查文件或目录是否存在",
-                parameters: {
-                    type: "object",
-                    properties: {
-                        path: {
-                            type: "string",
-                            description: "文件或目录路径",
-                        },
-                    },
-                    required: ["path"],
-                    additionalProperties: false,
+    function = {
+        name: "exist",
+        description: "检查文件或目录是否存在",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "文件或目录路径",
                 },
             },
-            env,
-        });
+            required: ["path"],
+            additionalProperties: false,
+        },
+    };
+    constructor(env) {
+        super({ env });
     }
     async call(input) {
         return await fsp

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

@@ -1,6 +1,8 @@
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 import type { ToolEnv } from "../../../types/index.js";
+import type { AgentNS, ToolCallContext } from "@ai-zen/agents-core";
 export declare class FindTextTool extends SdkCallbackTool {
+    function: AgentNS.FunctionDefine;
     constructor(env: ToolEnv);
     call(input: {
         path?: string;
@@ -8,5 +10,5 @@ export declare class FindTextTool extends SdkCallbackTool {
         text?: string;
         regex?: string;
         exclude?: string[];
-    }): Promise<string>;
+    }, ctx?: ToolCallContext): Promise<string>;
 }

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

@@ -2,45 +2,44 @@ import * as fsp from "fs/promises";
 import * as path from "path";
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 export class FindTextTool extends SdkCallbackTool {
-    constructor(env) {
-        super({
-            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",
-                            },
-                        },
+    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"],
                 },
             },
-            env,
-        });
+            required: ["path", "pattern"],
+        },
+    };
+    constructor(env) {
+        super({ env });
     }
-    async call(input) {
+    async call(input, ctx) {
+        const signal = ctx?.signal;
         try {
             const cwd = input.path ? this.resolve(input.path) : this.env.cwd;
             const result = [];
@@ -54,10 +53,16 @@ export class FindTextTool extends SdkCallbackTool {
                 cwd,
                 exclude: input.exclude || ["**/node_modules/**"],
             })) {
+                if (signal?.aborted)
+                    break; // 中断：停止继续遍历/消费
                 const fullPath = path.join(cwd, file);
                 const stats = await fsp.stat(fullPath);
+                if (signal?.aborted)
+                    break; // 中断：stat 后再次检查
                 if (stats.isFile()) {
                     const content = await fsp.readFile(fullPath, "utf-8");
+                    if (signal?.aborted)
+                        break; // 中断：readFile 后再次检查
                     const lines = content.split("\n");
                     const matches = [];
                     lines.forEach((lineContent, index) => {
@@ -83,6 +88,9 @@ export class FindTextTool extends SdkCallbackTool {
                     }
                 }
             }
+            if (signal?.aborted) {
+                return JSON.stringify({ aborted: true, results: result });
+            }
             return JSON.stringify(result);
         }
         catch (error) {

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

@@ -1,3 +1,4 @@
+import type { AgentNS, ToolCallContext } from "@ai-zen/agents-core";
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 import type { ToolEnv } from "../../../types/index.js";
 /**
@@ -11,11 +12,12 @@ import type { ToolEnv } from "../../../types/index.js";
  * 不加入 BUILTIN_TOOL_CLASSES 静态注册表。
  */
 export declare class GenerateImageTool extends SdkCallbackTool {
+    function: AgentNS.FunctionDefine;
     constructor(env: ToolEnv);
     call(input: {
         prompt: string;
         model?: string;
         size?: string;
         quality?: string;
-    }): Promise<string>;
+    }, ctx?: ToolCallContext): Promise<string>;
 }

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

@@ -11,40 +11,39 @@ import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
  * 不加入 BUILTIN_TOOL_CLASSES 静态注册表。
  */
 export class GenerateImageTool extends SdkCallbackTool {
-    constructor(env) {
-        super({
-            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,
+    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"],
                 },
             },
-            env,
-        });
+            required: ["prompt"],
+            additionalProperties: false,
+        },
+    };
+    constructor(env) {
+        super({ env });
     }
-    async call(input) {
+    async call(input, ctx) {
+        const signal = ctx?.signal;
         try {
             const prompt = input.prompt;
             if (!prompt || !prompt.trim()) {
@@ -95,6 +94,7 @@ export class GenerateImageTool extends SdkCallbackTool {
                 model: modelName,
                 size: input.size || imageModel.defaultSize || undefined,
                 quality: input.quality || imageModel.defaultQuality || undefined,
+                signal,
             });
             const images = result.data.map((img, i) => ({
                 index: i,
@@ -112,6 +112,14 @@ export class GenerateImageTool extends SdkCallbackTool {
             }, null, 2);
         }
         catch (error) {
+            // 中断（abort）时返回明确的中断结果
+            if (signal?.aborted) {
+                return JSON.stringify({
+                    success: false,
+                    aborted: true,
+                    error: "图片生成被中断（aborted）",
+                });
+            }
             return JSON.stringify({
                 success: false,
                 error: error instanceof Error ? error.message : "图片生成失败",

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

@@ -1,10 +1,12 @@
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 import type { ToolEnv } from "../../../types/index.js";
+import type { AgentNS, ToolCallContext } from "@ai-zen/agents-core";
 export declare class GlobTool extends SdkCallbackTool {
+    function: AgentNS.FunctionDefine;
     constructor(env: ToolEnv);
     call(input: {
         path?: string;
         pattern: string;
         exclude?: string[];
-    }): Promise<string>;
+    }, ctx?: ToolCallContext): Promise<string>;
 }

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

@@ -1,38 +1,37 @@
 import * as fsp from "fs/promises";
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 export class GlobTool extends SdkCallbackTool {
-    constructor(env) {
-        super({
-            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",
-                            },
-                        },
+    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,
                 },
             },
-            env,
-        });
+            required: ["pattern"],
+            additionalProperties: false,
+        },
+    };
+    constructor(env) {
+        super({ env });
     }
-    async call(input) {
+    async call(input, ctx) {
+        const signal = ctx?.signal;
         try {
             const cwd = input.path ? this.resolve(input.path) : this.env.cwd;
             const result = [];
@@ -40,8 +39,13 @@ export class GlobTool extends SdkCallbackTool {
                 exclude: input.exclude || ["**/node_modules/**"],
                 cwd,
             })) {
+                if (signal?.aborted)
+                    break; // 中断：停止继续遍历/消费
                 result.push(file);
             }
+            if (signal?.aborted) {
+                return JSON.stringify({ aborted: true, files: result });
+            }
             return JSON.stringify(result);
         }
         catch (error) {

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

@@ -1,6 +1,8 @@
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 import type { ToolEnv } from "../../../types/index.js";
+import type { AgentNS } from "@ai-zen/agents-core";
 export declare class LsTool extends SdkCallbackTool {
+    function: AgentNS.FunctionDefine;
     constructor(env: ToolEnv);
     call(input: {
         path: string;

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

@@ -1,25 +1,23 @@
 import * as fsp from "fs/promises";
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 export class LsTool extends SdkCallbackTool {
-    constructor(env) {
-        super({
-            function: {
-                name: "ls",
-                description: "列出目录",
-                parameters: {
-                    type: "object",
-                    properties: {
-                        path: {
-                            type: "string",
-                            description: "目录路径",
-                        },
-                    },
-                    required: ["path"],
-                    additionalProperties: false,
+    function = {
+        name: "ls",
+        description: "列出目录",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "目录路径",
                 },
             },
-            env,
-        });
+            required: ["path"],
+            additionalProperties: false,
+        },
+    };
+    constructor(env) {
+        super({ env });
     }
     async call(input) {
         try {

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

@@ -1,6 +1,8 @@
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 import type { ToolEnv } from "../../../types/index.js";
+import type { AgentNS } from "@ai-zen/agents-core";
 export declare class MkdirTool extends SdkCallbackTool {
+    function: AgentNS.FunctionDefine;
     constructor(env: ToolEnv);
     call(input: {
         path: string;

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

@@ -1,29 +1,27 @@
 import * as fsp from "fs/promises";
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 export class MkdirTool extends SdkCallbackTool {
-    constructor(env) {
-        super({
-            function: {
-                name: "mkdir",
-                description: "创建目录",
-                parameters: {
-                    type: "object",
-                    properties: {
-                        path: {
-                            type: "string",
-                            description: "目录路径",
-                        },
-                        recursive: {
-                            type: "boolean",
-                            description: "是否递归创建子目录",
-                        },
-                    },
-                    required: ["path"],
-                    additionalProperties: false,
+    function = {
+        name: "mkdir",
+        description: "创建目录",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "目录路径",
+                },
+                recursive: {
+                    type: "boolean",
+                    description: "是否递归创建子目录",
                 },
             },
-            env,
-        });
+            required: ["path"],
+            additionalProperties: false,
+        },
+    };
+    constructor(env) {
+        super({ env });
     }
     async call(input) {
         try {

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

@@ -1,6 +1,8 @@
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 import type { ToolEnv } from "../../../types/index.js";
+import type { AgentNS } from "@ai-zen/agents-core";
 export declare class ReadFileTool extends SdkCallbackTool {
+    function: AgentNS.FunctionDefine;
     constructor(env: ToolEnv);
     call(input: {
         path: string;

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

@@ -1,25 +1,23 @@
 import * as fsp from "fs/promises";
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 export class ReadFileTool extends SdkCallbackTool {
-    constructor(env) {
-        super({
-            function: {
-                name: "readFile",
-                description: "读取文件",
-                parameters: {
-                    type: "object",
-                    properties: {
-                        path: {
-                            type: "string",
-                            description: "文件路径",
-                        },
-                    },
-                    required: ["path"],
-                    additionalProperties: false,
+    function = {
+        name: "readFile",
+        description: "读取文件",
+        parameters: {
+            type: "object",
+            properties: {
+                path: {
+                    type: "string",
+                    description: "文件路径",
                 },
             },
-            env,
-        });
+            required: ["path"],
+            additionalProperties: false,
+        },
+    };
+    constructor(env) {
+        super({ env });
     }
     async call(input) {
         try {

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

@@ -1,6 +1,8 @@
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 import type { ToolEnv } from "../../../types/index.js";
+import type { AgentNS } from "@ai-zen/agents-core";
 export declare class RenameTool extends SdkCallbackTool {
+    function: AgentNS.FunctionDefine;
     constructor(env: ToolEnv);
     call(input: {
         oldPath: string;

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

@@ -1,29 +1,27 @@
 import * as fsp from "fs/promises";
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 export class RenameTool extends SdkCallbackTool {
-    constructor(env) {
-        super({
-            function: {
-                name: "rename",
-                description: "重命名或移动文件/目录（相当于 mv 命令）",
-                parameters: {
-                    type: "object",
-                    properties: {
-                        oldPath: {
-                            type: "string",
-                            description: "原路径",
-                        },
-                        newPath: {
-                            type: "string",
-                            description: "新路径",
-                        },
-                    },
-                    required: ["oldPath", "newPath"],
-                    additionalProperties: false,
+    function = {
+        name: "rename",
+        description: "重命名或移动文件/目录（相当于 mv 命令）",
+        parameters: {
+            type: "object",
+            properties: {
+                oldPath: {
+                    type: "string",
+                    description: "原路径",
+                },
+                newPath: {
+                    type: "string",
+                    description: "新路径",
                 },
             },
-            env,
-        });
+            required: ["oldPath", "newPath"],
+            additionalProperties: false,
+        },
+    };
+    constructor(env) {
+        super({ env });
     }
     async call(input) {
         try {

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

@@ -1,6 +1,8 @@
 import { SdkCallbackTool } from "../../../runtime/SdkCallbackTool.js";
 import type { ToolEnv } from "../../../types/index.js";
+import type { AgentNS } from "@ai-zen/agents-core";
 export declare class RmTool extends SdkCallbackTool {
+    function: AgentNS.FunctionDefine;
     constructor(env: ToolEnv);
     call(input: {
         path: string;