@ax-llm/ax 9.0.23 → 9.0.25

package/cjs/funcs/code.js

@@ -23,7 +23,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AxCodeInterpreterPermission = exports.axJSInterpreterFunction = void 0;
+exports.AxJSInterpreter = exports.AxJSInterpreterPermission = void 0;
 const _crypto = __importStar(require("crypto"));
 const _http = __importStar(require("http"));
 const _https = __importStar(require("https"));
@@ -31,60 +31,56 @@ const _fs = __importStar(require("node:fs"));
 const _os = __importStar(require("os"));
 const _process = __importStar(require("process"));
 const vm_1 = require("vm");
-const axJSInterpreterFunction = (permissions = []) => ({
-    name: 'jsInterpreter',
-    description: 'Run Javascript code',
-    parameters: {
-        type: 'object',
-        properties: {
-            code: {
-                type: 'string',
-                description: 'JS code with a return value in the end.'
-            }
-        },
-        required: ['code']
-    },
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    func: ({ code }) => {
-        return new Promise((resolve) => {
-            resolve(codeInterpreterJavascript(code, permissions));
-        });
+var AxJSInterpreterPermission;
+(function (AxJSInterpreterPermission) {
+    AxJSInterpreterPermission["FS"] = "node:fs";
+    AxJSInterpreterPermission["NET"] = "net";
+    AxJSInterpreterPermission["OS"] = "os";
+    AxJSInterpreterPermission["CRYPTO"] = "crypto";
+    AxJSInterpreterPermission["PROCESS"] = "process";
+})(AxJSInterpreterPermission || (exports.AxJSInterpreterPermission = AxJSInterpreterPermission = {}));
+class AxJSInterpreter {
+    permissions;
+    constructor({ permissions = [] } = {}) {
+        this.permissions = permissions ?? [];
     }
-});
-exports.axJSInterpreterFunction = axJSInterpreterFunction;
-var AxCodeInterpreterPermission;
-(function (AxCodeInterpreterPermission) {
-    AxCodeInterpreterPermission["FS"] = "node:fs";
-    AxCodeInterpreterPermission["NET"] = "net";
-    AxCodeInterpreterPermission["OS"] = "os";
-    AxCodeInterpreterPermission["CRYPTO"] = "crypto";
-    AxCodeInterpreterPermission["PROCESS"] = "process";
-})(AxCodeInterpreterPermission || (exports.AxCodeInterpreterPermission = AxCodeInterpreterPermission = {}));
-const codeInterpreterJavascript = (code, permissions = []
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-) => {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const context = {
-        // require: require,
-        console: console
-    };
-    if (permissions.includes(AxCodeInterpreterPermission.FS)) {
-        context.fs = _fs;
-    }
-    if (permissions.includes(AxCodeInterpreterPermission.NET)) {
-        context.http = _http;
-        context.https = _https;
-    }
-    if (permissions.includes(AxCodeInterpreterPermission.OS)) {
-        context.os = _os;
+    codeInterpreterJavascript(code) {
+        const context = { console };
+        if (this.permissions.includes(AxJSInterpreterPermission.FS)) {
+            context.fs = _fs;
+        }
+        if (this.permissions.includes(AxJSInterpreterPermission.NET)) {
+            context.http = _http;
+            context.https = _https;
+        }
+        if (this.permissions.includes(AxJSInterpreterPermission.OS)) {
+            context.os = _os;
+        }
+        if (this.permissions.includes(AxJSInterpreterPermission.CRYPTO)) {
+            context.crypto = _crypto;
+        }
+        if (this.permissions.includes(AxJSInterpreterPermission.PROCESS)) {
+            context.process = _process;
+        }
+        return (0, vm_1.runInNewContext)(`(function() { ${code} })()`, context);
     }
-    if (permissions.includes(AxCodeInterpreterPermission.CRYPTO)) {
-        context.crypto = _crypto;
+    toFunction() {
+        return {
+            name: 'javascriptInterpreter',
+            description: 'Use this function to run Javascript code and get any expected return value.',
+            parameters: {
+                type: 'object',
+                properties: {
+                    code: {
+                        type: 'string',
+                        description: 'JS code with a return value in the end.'
+                    }
+                },
+                required: ['code']
+            },
+            func: this.codeInterpreterJavascript
+        };
     }
-    if (permissions.includes(AxCodeInterpreterPermission.PROCESS)) {
-        context.process = _process;
-    }
-    // executing code within the sandbox
-    return (0, vm_1.runInNewContext)(`(function() { ${code} })()`, context);
-};
+}
+exports.AxJSInterpreter = AxJSInterpreter;
 //# sourceMappingURL=code.js.map

package/cjs/funcs/code.js.map

@@ -1 +1 @@
-{"version":3,"file":"code.js","sourceRoot":"","sources":["../../../funcs/code.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,gDAAkC;AAClC,4CAA8B;AAC9B,8CAAgC;AAChC,6CAA+B;AAC/B,wCAA0B;AAC1B,kDAAoC;AACpC,2BAAqC;AAI9B,MAAM,uBAAuB,GAAG,CACrC,cAAsD,EAAE,EAC5C,EAAE,CAAC,CAAC;IAChB,IAAI,EAAE,eAAe;IACrB,WAAW,EAAE,qBAAqB;IAClC,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yCAAyC;aACvD;SACF;QACD,QAAQ,EAAE,CAAC,MAAM,CAAC;KACnB;IAED,8DAA8D;IAC9D,IAAI,EAAE,CAAC,EAAE,IAAI,EAA8B,EAAgB,EAAE;QAC3D,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,OAAO,CAAC,yBAAyB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC;CACF,CAAC,CAAC;AAtBU,QAAA,uBAAuB,2BAsBjC;AAEH,IAAY,2BAMX;AAND,WAAY,2BAA2B;IACrC,6CAAc,CAAA;IACd,0CAAW,CAAA;IACX,wCAAS,CAAA;IACT,gDAAiB,CAAA;IACjB,kDAAmB,CAAA;AACrB,CAAC,EANW,2BAA2B,2CAA3B,2BAA2B,QAMtC;AAED,MAAM,yBAAyB,GAAG,CAChC,IAAY,EACZ,cAAsD,EAAE;AACxD,8DAA8D;EACzD,EAAE;IACP,8DAA8D;IAC9D,MAAM,OAAO,GAA2B;QACtC,oBAAoB;QACpB,OAAO,EAAE,OAAO;KACjB,CAAC;IAEF,IAAI,WAAW,CAAC,QAAQ,CAAC,2BAA2B,CAAC,EAAE,CAAC,EAAE,CAAC;QACzD,OAAO,CAAC,EAAE,GAAG,GAAG,CAAC;IACnB,CAAC;IAED,IAAI,WAAW,CAAC,QAAQ,CAAC,2BAA2B,CAAC,GAAG,CAAC,EAAE,CAAC;QAC1D,OAAO,CAAC,IAAI,GAAG,KAAK,CAAC;QACrB,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,IAAI,WAAW,CAAC,QAAQ,CAAC,2BAA2B,CAAC,EAAE,CAAC,EAAE,CAAC;QACzD,OAAO,CAAC,EAAE,GAAG,GAAG,CAAC;IACnB,CAAC;IAED,IAAI,WAAW,CAAC,QAAQ,CAAC,2BAA2B,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7D,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC;IAC3B,CAAC;IAED,IAAI,WAAW,CAAC,QAAQ,CAAC,2BAA2B,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9D,OAAO,CAAC,OAAO,GAAG,QAAQ,CAAC;IAC7B,CAAC;IAED,oCAAoC;IACpC,OAAO,IAAA,oBAAe,EAAC,iBAAiB,IAAI,OAAO,EAAE,OAAO,CAAC,CAAC;AAChE,CAAC,CAAC"}
+{"version":3,"file":"code.js","sourceRoot":"","sources":["../../../funcs/code.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,gDAAkC;AAClC,4CAA8B;AAC9B,8CAAgC;AAChC,6CAA+B;AAC/B,wCAA0B;AAC1B,kDAAoC;AACpC,2BAAqC;AAIrC,IAAY,yBAMX;AAND,WAAY,yBAAyB;IACnC,2CAAc,CAAA;IACd,wCAAW,CAAA;IACX,sCAAS,CAAA;IACT,8CAAiB,CAAA;IACjB,gDAAmB,CAAA;AACrB,CAAC,EANW,yBAAyB,yCAAzB,yBAAyB,QAMpC;AAED,MAAa,eAAe;IAClB,WAAW,CAAuC;IAE1D,YAAY,EACV,WAAW,GAAG,EAAE,KAGF,EAAE;QAChB,IAAI,CAAC,WAAW,GAAG,WAAW,IAAI,EAAE,CAAC;IACvC,CAAC;IAEO,yBAAyB,CAAC,IAAY;QAC5C,MAAM,OAAO,GAA+B,EAAE,OAAO,EAAE,CAAC;QAExD,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,CAAC,EAAE,CAAC;YAC5D,OAAO,CAAC,EAAE,GAAG,GAAG,CAAC;QACnB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,GAAG,CAAC,EAAE,CAAC;YAC7D,OAAO,CAAC,IAAI,GAAG,KAAK,CAAC;YACrB,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC;QACzB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,CAAC,EAAE,CAAC;YAC5D,OAAO,CAAC,EAAE,GAAG,GAAG,CAAC;QACnB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,MAAM,CAAC,EAAE,CAAC;YAChE,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC;QAC3B,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,OAAO,CAAC,EAAE,CAAC;YACjE,OAAO,CAAC,OAAO,GAAG,QAAQ,CAAC;QAC7B,CAAC;QAED,OAAO,IAAA,oBAAe,EAAC,iBAAiB,IAAI,OAAO,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC;IAEM,UAAU;QACf,OAAO;YACL,IAAI,EAAE,uBAAuB;YAC7B,WAAW,EACT,6EAA6E;YAC/E,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,IAAI,EAAE;wBACJ,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,yCAAyC;qBACvD;iBACF;gBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;aACnB;YAED,IAAI,EAAE,IAAI,CAAC,yBAAyB;SACrC,CAAC;IACJ,CAAC;CACF;AAzDD,0CAyDC"}

package/cjs/funcs/docker.d.ts

@@ -0,0 +1,26 @@
+import type { AxFunction } from '../ai/types.js';
+export declare class AxDockerSession {
+    private readonly apiUrl;
+    private containerId;
+    constructor(apiUrl?: string);
+    pullImage(imageName: string): Promise<void>;
+    createContainer({ imageName, volumes, // Example format: [{ hostPath: '/host/path', containerPath: '/container/path' }]
+    doNotPullImage }: Readonly<{
+        imageName: string;
+        volumes?: Array<{
+            hostPath: string;
+            containerPath: string;
+        }>;
+        doNotPullImage?: boolean;
+    }>): Promise<{
+        Id: string;
+    }>;
+    startContainer(): Promise<void>;
+    connectToContainer(containerId: string): Promise<void>;
+    executeCommand(command: string): Promise<string>;
+    stopContainer(): Promise<void>;
+    listContainers(all?: boolean): Promise<unknown>;
+    getContainerLogs(): Promise<string>;
+    private fetchDockerAPI;
+    toFunction(): AxFunction;
+}

package/cjs/funcs/docker.js

@@ -0,0 +1,144 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AxDockerSession = void 0;
+class AxDockerSession {
+    apiUrl;
+    containerId = null;
+    constructor(apiUrl = 'http://localhost:2375') {
+        this.apiUrl = apiUrl;
+    }
+    async pullImage(imageName) {
+        const response = await this.fetchDockerAPI(`/images/create?fromImage=${encodeURIComponent(imageName)}`, {
+            method: 'POST'
+        });
+        if (!response.ok) {
+            throw new Error(`Failed to pull image: ${response.statusText}`);
+        }
+        // Wait for the pull to complete
+        await response.text();
+    }
+    async createContainer({ imageName, volumes = [], // Example format: [{ hostPath: '/host/path', containerPath: '/container/path' }]
+    doNotPullImage }) {
+        const binds = volumes.map((v) => `${v.hostPath}:${v.containerPath}`);
+        if (!doNotPullImage) {
+            await this.pullImage(imageName);
+        }
+        const response = await this.fetchDockerAPI(`/containers/create`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                Image: imageName,
+                Tty: true,
+                OpenStdin: false,
+                AttachStdin: false,
+                AttachStdout: false,
+                AttachStderr: false,
+                HostConfig: { Binds: binds }
+            })
+        });
+        if (!response.ok) {
+            throw new Error(`Failed to create container: ${response.statusText}`);
+        }
+        const data = (await response.json());
+        this.containerId = data.Id;
+        return data;
+    }
+    async startContainer() {
+        if (!this.containerId) {
+            throw new Error('No container created or connected');
+        }
+        const response = await this.fetchDockerAPI(`/containers/${this.containerId}/start`, {
+            method: 'POST'
+        });
+        if (!response.ok) {
+            throw new Error(`Failed to start container: ${response.statusText}`);
+        }
+    }
+    async connectToContainer(containerId) {
+        const response = await this.fetchDockerAPI(`/containers/${containerId}/json`);
+        if (!response.ok) {
+            throw new Error(`Failed to connect to container: ${response.statusText}`);
+        }
+        this.containerId = containerId;
+    }
+    async executeCommand(command) {
+        if (!this.containerId) {
+            throw new Error('No container created or connected');
+        }
+        // Create exec instance
+        const createResponse = await this.fetchDockerAPI(`/containers/${this.containerId}/exec`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                Cmd: ['sh', '-c', command],
+                AttachStdout: true,
+                AttachStderr: true
+            })
+        });
+        if (!createResponse.ok) {
+            throw new Error(`Failed to create exec instance: ${createResponse.statusText}`);
+        }
+        const execData = (await createResponse.json());
+        // Start exec instance
+        const startResponse = await this.fetchDockerAPI(`/exec/${execData.Id}/start`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                Detach: false,
+                Tty: false
+            })
+        });
+        if (!startResponse.ok) {
+            throw new Error(`Failed to start exec instance: ${startResponse.statusText}`);
+        }
+        // Return the output
+        return await startResponse.text();
+    }
+    async stopContainer() {
+        if (!this.containerId) {
+            throw new Error('No container created or connected');
+        }
+        const response = await this.fetchDockerAPI(`/containers/${this.containerId}/stop`, {
+            method: 'POST'
+        });
+        if (!response.ok) {
+            throw new Error(`Failed to stop container: ${response.statusText}`);
+        }
+    }
+    async listContainers(all = false) {
+        const response = await this.fetchDockerAPI(`/containers/json?all=${all}`, {
+            method: 'GET'
+        });
+        return response.json();
+    }
+    async getContainerLogs() {
+        if (!this.containerId) {
+            throw new Error('No container created or connected');
+        }
+        const response = await this.fetchDockerAPI(`/containers/${this.containerId}/logs?stdout=true&stderr=true`, { method: 'GET' });
+        return response.text();
+    }
+    async fetchDockerAPI(endpoint, options) {
+        const url = new URL(endpoint, this.apiUrl).toString();
+        return await fetch(url, options);
+    }
+    toFunction() {
+        return {
+            name: 'commandExecution',
+            description: 'Use this function to execute shell commands, scripts, and programs. This function enables interaction with the file system, running system utilities, and performing tasks that require a shell interface.',
+            parameters: {
+                type: 'object',
+                properties: {
+                    command: {
+                        type: 'string',
+                        description: 'Shell command to execute. eg. `ls -l` or `echo "Hello, World!"`.'
+                    }
+                },
+                required: ['command']
+            },
+            func: this.executeCommand
+        };
+    }
+}
+exports.AxDockerSession = AxDockerSession;
+//# sourceMappingURL=docker.js.map

package/cjs/funcs/docker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"docker.js","sourceRoot":"","sources":["../../../funcs/docker.ts"],"names":[],"mappings":";;;AAEA,MAAa,eAAe;IACT,MAAM,CAAS;IACxB,WAAW,GAAkB,IAAI,CAAC;IAE1C,YAAY,SAAiB,uBAAuB;QAClD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,SAAiB;QAC/B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CACxC,4BAA4B,kBAAkB,CAAC,SAAS,CAAC,EAAE,EAC3D;YACE,MAAM,EAAE,MAAM;SACf,CACF,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,gCAAgC;QAChC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,EACpB,SAAS,EACT,OAAO,GAAG,EAAE,EAAE,iFAAiF;IAC/F,cAAc,EAKd;QACA,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC;QAErE,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,oBAAoB,EAAE;YAC/D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,SAAS;gBAChB,GAAG,EAAE,IAAI;gBACT,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,KAAK;gBAClB,YAAY,EAAE,KAAK;gBACnB,YAAY,EAAE,KAAK;gBACnB,UAAU,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;aAC7B,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+BAA+B,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAmB,CAAC;QACvD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC;QAE3B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CACxC,eAAe,IAAI,CAAC,WAAW,QAAQ,EACvC;YACE,MAAM,EAAE,MAAM;SACf,CACF,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,WAAmB;QAC1C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CACxC,eAAe,WAAW,OAAO,CAClC,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,mCAAmC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QAC5E,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,OAAe;QAClC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,uBAAuB;QACvB,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,cAAc,CAC9C,eAAe,IAAI,CAAC,WAAW,OAAO,EACtC;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,GAAG,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC;gBAC1B,YAAY,EAAE,IAAI;gBAClB,YAAY,EAAE,IAAI;aACnB,CAAC;SACH,CACF,CAAC;QAEF,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CACb,mCAAmC,cAAc,CAAC,UAAU,EAAE,CAC/D,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,CAAC,MAAM,cAAc,CAAC,IAAI,EAAE,CAAmB,CAAC;QAEjE,sBAAsB;QACtB,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,cAAc,CAC7C,SAAS,QAAQ,CAAC,EAAE,QAAQ,EAC5B;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,MAAM,EAAE,KAAK;gBACb,GAAG,EAAE,KAAK;aACX,CAAC;SACH,CACF,CAAC;QAEF,IAAI,CAAC,aAAa,CAAC,EAAE,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,kCAAkC,aAAa,CAAC,UAAU,EAAE,CAC7D,CAAC;QACJ,CAAC;QAED,oBAAoB;QACpB,OAAO,MAAM,aAAa,CAAC,IAAI,EAAE,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CACxC,eAAe,IAAI,CAAC,WAAW,OAAO,EACtC;YACE,MAAM,EAAE,MAAM;SACf,CACF,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,6BAA6B,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,MAAe,KAAK;QACvC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,wBAAwB,GAAG,EAAE,EAAE;YACxE,MAAM,EAAE,KAAK;SACd,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,gBAAgB;QACpB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CACxC,eAAe,IAAI,CAAC,WAAW,+BAA+B,EAC9D,EAAE,MAAM,EAAE,KAAK,EAAE,CAClB,CAAC;QACF,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;IACzB,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,QAAgB,EAChB,OAA+B;QAE/B,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QACtD,OAAO,MAAM,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;IAEM,UAAU;QACf,OAAO;YACL,IAAI,EAAE,kBAAkB;YACxB,WAAW,EACT,4MAA4M;YAC9M,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,OAAO,EAAE;wBACP,IAAI,EAAE,QAAQ;wBACd,WAAW,EACT,kEAAkE;qBACrE;iBACF;gBACD,QAAQ,EAAE,CAAC,SAAS,CAAC;aACtB;YAED,IAAI,EAAE,IAAI,CAAC,cAAc;SAC1B,CAAC;IACJ,CAAC;CACF;AA7MD,0CA6MC"}

package/cjs/funcs/embed.d.ts

@@ -1,6 +1,17 @@
 import type { AxAIService, AxAIServiceActionOptions, AxFunction } from '../ai/index.js';
-export declare const axEmbedAdapter: (ai: AxAIService, info: Readonly<{
-    name: string;
-    description: string;
-    argumentDescription: string;
-}>, func: (args: readonly number[], extra?: Readonly<AxAIServiceActionOptions>) => Promise<any>) => AxFunction;
+export declare class AxEmbeddingAdapter {
+    private aiService;
+    private info;
+    private func;
+    constructor({ ai, info, func }: Readonly<{
+        ai: AxAIService;
+        info: Readonly<{
+            name: string;
+            description: string;
+            argumentDescription: string;
+        }>;
+        func: (args: readonly number[], extra?: Readonly<AxAIServiceActionOptions>) => Promise<unknown>;
+    }>);
+    private embedAdapter;
+    toFunction(): AxFunction;
+}

package/cjs/funcs/embed.js

@@ -1,33 +1,42 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.axEmbedAdapter = void 0;
-const axEmbedAdapter = (ai, info, func) => ({
-    name: info.name,
-    description: info.description,
-    parameters: {
-        type: 'object',
-        properties: {
-            text: {
-                type: 'string',
-                description: info.argumentDescription
-            }
-        },
-        required: ['text']
-    },
-    func: ({ text }, extra
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    ) => {
-        return new Promise((resolve) => {
-            resolve(embedAdapter(ai, text, func, extra));
-        });
+exports.AxEmbeddingAdapter = void 0;
+class AxEmbeddingAdapter {
+    aiService;
+    info;
+    func;
+    constructor({ ai, info, func }) {
+        this.aiService = ai;
+        this.info = info;
+        this.func = func;
     }
-});
-exports.axEmbedAdapter = axEmbedAdapter;
-const embedAdapter = async (ai, text, func, extra
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-) => {
-    const embedRes = await ai.embed({ texts: [text] }, { sessionId: extra?.sessionId });
-    const embeds = embedRes.embeddings.at(0);
-    return func.length === 2 ? func(embeds, extra) : func(embeds);
-};
+    async embedAdapter(text, extra) {
+        const embedRes = await this.aiService.embed({ texts: [text] }, { sessionId: extra?.sessionId });
+        const embeds = embedRes.embeddings.at(0);
+        if (!embeds) {
+            throw new Error('Failed to embed text');
+        }
+        return this.func.length === 2
+            ? this.func(embeds, extra)
+            : this.func(embeds);
+    }
+    toFunction() {
+        return {
+            name: this.info.name,
+            description: this.info.description,
+            parameters: {
+                type: 'object',
+                properties: {
+                    text: {
+                        type: 'string',
+                        description: this.info.argumentDescription
+                    }
+                },
+                required: ['text']
+            },
+            func: this.embedAdapter
+        };
+    }
+}
+exports.AxEmbeddingAdapter = AxEmbeddingAdapter;
 //# sourceMappingURL=embed.js.map

package/cjs/funcs/embed.js.map

@@ -1 +1 @@
-{"version":3,"file":"embed.js","sourceRoot":"","sources":["../../../funcs/embed.ts"],"names":[],"mappings":";;;AAOO,MAAM,cAAc,GAAG,CAC5B,EAAe,EACf,IAIE,EACF,IAIiB,EACL,EAAE,CAAC,CAAC;IAChB,IAAI,EAAE,IAAI,CAAC,IAAI;IACf,WAAW,EAAE,IAAI,CAAC,WAAW;IAC7B,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,IAAI,CAAC,mBAAmB;aACtC;SACF;QACD,QAAQ,EAAE,CAAC,MAAM,CAAC;KACnB;IAED,IAAI,EAAE,CACJ,EAAE,IAAI,EAA8B,EACpC,KAA0C;IAC1C,8DAA8D;MAChD,EAAE;QAChB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,OAAO,CAAC,YAAY,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC;CACF,CAAC,CAAC;AAnCU,QAAA,cAAc,kBAmCxB;AAEH,MAAM,YAAY,GAAG,KAAK,EACxB,EAAe,EACf,IAAY,EACZ,IAAuB,EACvB,KAA0C;AAC1C,8DAA8D;EAChD,EAAE;IAChB,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,KAAK,CAC7B,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,EACjB,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,CAChC,CAAC;IACF,MAAM,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAEzC,OAAO,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAChE,CAAC,CAAC"}
+{"version":3,"file":"embed.js","sourceRoot":"","sources":["../../../funcs/embed.ts"],"names":[],"mappings":";;;AAMA,MAAa,kBAAkB;IACrB,SAAS,CAAc;IACvB,IAAI,CAIV;IACM,IAAI,CAGU;IAEtB,YAAY,EACV,EAAE,EACF,IAAI,EACJ,IAAI,EAYJ;QACA,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;QACpB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,IAAY,EACZ,KAA0C;QAE1C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CACzC,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,EACjB,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,CAChC,CAAC;QACF,MAAM,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAEzC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;YAC3B,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC;YAC1B,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACxB,CAAC;IAEM,UAAU;QACf,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI;YACpB,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW;YAClC,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,IAAI,EAAE;wBACJ,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,mBAAmB;qBAC3C;iBACF;gBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;aACnB;YACD,IAAI,EAAE,IAAI,CAAC,YAAY;SACxB,CAAC;IACJ,CAAC;CACF;AArED,gDAqEC"}

package/cjs/funcs/index.d.ts

@@ -1,3 +1,3 @@
 export * from './code.js';
 export * from './embed.js';
-export * from './functions.js';
+export * from './docker.js';

package/cjs/funcs/index.js

@@ -16,5 +16,5 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./code.js"), exports);
 __exportStar(require("./embed.js"), exports);
-__exportStar(require("./functions.js"), exports);
+__exportStar(require("./docker.js"), exports);
 //# sourceMappingURL=index.js.map

package/cjs/funcs/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../funcs/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,4CAA0B;AAC1B,6CAA2B;AAC3B,iDAA+B"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../funcs/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,4CAA0B;AAC1B,6CAA2B;AAC3B,8CAA4B"}

package/cjs/prompts/agent.d.ts

@@ -6,6 +6,7 @@ export interface AxAgentic extends AxTunable, AxUsable {
 }
 export type AxAgentOptions = Omit<AxGenerateOptions, 'functions' | 'functionCall'>;
 export declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAgentic {
+    private signature;
     private program;
     private name;
     private description;
@@ -18,7 +19,6 @@ export declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implement
         agents?: AxAgentic[];
         functions?: AxFunction[];
     }>, options?: Readonly<AxAgentOptions>);
-    getSignature(): AxSignature;
     setExamples(examples: Readonly<AxProgramExamples>): void;
     setId(id: string): void;
     setParentId(parentId: string): void;

package/cjs/prompts/agent.js

@@ -1,16 +1,19 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AxAgent = void 0;
-const index_js_1 = require("../trace/index.js");
+const index_js_1 = require("../dsp/index.js");
+const index_js_2 = require("../trace/index.js");
 const cot_js_1 = require("./cot.js");
 const react_js_1 = require("./react.js");
 class AxAgent {
+    signature;
     program;
     name;
     description;
     subAgentList;
     func;
     constructor(ai, { name, description, signature, agents, functions }, options) {
+        this.signature = new index_js_1.AxSignature(signature);
         const funcs = [
             ...(functions ?? []),
             ...(agents?.map((a) => a.getFunction()) ?? [])
@@ -21,8 +24,8 @@ class AxAgent {
         };
         this.program =
             funcs.length > 0
-                ? new react_js_1.AxReAct(ai, signature, opt)
-                : new cot_js_1.AxChainOfThought(ai, signature, opt);
+                ? new react_js_1.AxReAct(ai, this.signature, opt)
+                : new cot_js_1.AxChainOfThought(ai, this.signature, opt);
         if (!name || name.length < 5) {
             throw new Error('Agent name must be at least 10 characters (more descriptive): ' + name);
         }
@@ -36,16 +39,13 @@ class AxAgent {
         this.func = {
             name: toCamelCase(this.name),
             description: this.description,
-            parameters: this.program.getSignature().toJSONSchema(),
+            parameters: this.signature.toJSONSchema(),
             func: () => this.forward
         };
         for (const agent of agents ?? []) {
             this.program.register(agent);
         }
     }
-    getSignature() {
-        return this.program.getSignature();
-    }
     setExamples(examples) {
         this.program.setExamples(examples);
     }
@@ -80,7 +80,7 @@ class AxAgent {
             ['agent.subAgents']: this.subAgentList ?? 'none'
         };
         return await options?.tracer.startActiveSpan('Agent', {
-            kind: index_js_1.AxSpanKind.SERVER,
+            kind: index_js_2.AxSpanKind.SERVER,
             attributes
         }, async (span) => {
             const res = await this.program.forward(values, options);

package/cjs/prompts/agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../prompts/agent.ts"],"names":[],"mappings":";;;AAYA,gDAA+C;AAE/C,qCAA4C;AAC5C,yCAAqC;AAWrC,MAAa,OAAO;IAGV,OAAO,CAAqB;IAE5B,IAAI,CAAS;IACb,WAAW,CAAS;IACpB,YAAY,CAAU;IACtB,IAAI,CAAa;IAEzB,YACE,EAAe,EACf,EACE,IAAI,EACJ,WAAW,EACX,SAAS,EACT,MAAM,EACN,SAAS,EAOT,EACF,OAAkC;QAElC,MAAM,KAAK,GAAiB;YAC1B,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC;YACpB,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;SAC/C,CAAC;QAEF,MAAM,GAAG,GAAG;YACV,GAAG,OAAO;YACV,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,OAAO;YACV,KAAK,CAAC,MAAM,GAAG,CAAC;gBACd,CAAC,CAAC,IAAI,kBAAO,CAAU,EAAE,EAAE,SAAS,EAAE,GAAG,CAAC;gBAC1C,CAAC,CAAC,IAAI,yBAAgB,CAAU,EAAE,EAAE,SAAS,EAAE,GAAG,CAAC,CAAC;QAExD,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CACb,gEAAgE,GAAG,IAAI,CACxE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CACb,4FAA4F;gBAC1F,WAAW,CACd,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAExE,IAAI,CAAC,IAAI,GAAG;YACV,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;YAC5B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,YAAY,EAAE;YACtD,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO;SACzB,CAAC;QAEF,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEM,YAAY;QACjB,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;IACrC,CAAC;IAEM,WAAW,CAAC,QAAqC;QACtD,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAEM,KAAK,CAAC,EAAU;QACrB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC;IAEM,WAAW,CAAC,QAAgB;QACjC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAEM,SAAS;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC;IAEM,QAAQ,CAAC,KAAgC;QAC9C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAEM,QAAQ;QACb,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;IACjC,CAAC;IAEM,UAAU;QACf,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAC5B,CAAC;IAEM,WAAW;QAChB,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,OAAO,CAClB,MAAU,EACV,OAA2C;QAE3C,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,UAAU,GAAG;YACjB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,IAAI;YACzB,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,WAAW;YACvC,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,YAAY,IAAI,MAAM;SACjD,CAAC;QAEF,OAAO,MAAM,OAAO,EAAE,MAAM,CAAC,eAAe,CAC1C,OAAO,EACP;YACE,IAAI,EAAE,qBAAU,CAAC,MAAM;YACvB,UAAU;SACX,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACxD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF;AAtID,0BAsIC;AAED,SAAS,WAAW,CAAC,WAAmB;IACtC,8FAA8F;IAC9F,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAEhD,wFAAwF;IACxF,MAAM,eAAe,GAAG,KAAK;SAC1B,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACnB,qEAAqE;QACrE,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAErC,gEAAgE;QAChE,IAAI,KAAK,GAAG,CAAC,IAAI,SAAS,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3C,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;SACD,IAAI,CAAC,EAAE,CAAC,CAAC;IAEZ,OAAO,eAAe,CAAC;AACzB,CAAC"}
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../prompts/agent.ts"],"names":[],"mappings":";;;AACA,8CAAsE;AAWtE,gDAA+C;AAE/C,qCAA4C;AAC5C,yCAAqC;AAWrC,MAAa,OAAO;IAGV,SAAS,CAAc;IACvB,OAAO,CAAkC;IAEzC,IAAI,CAAS;IACb,WAAW,CAAS;IACpB,YAAY,CAAU;IACtB,IAAI,CAAa;IAEzB,YACE,EAAe,EACf,EACE,IAAI,EACJ,WAAW,EACX,SAAS,EACT,MAAM,EACN,SAAS,EAOT,EACF,OAAkC;QAElC,IAAI,CAAC,SAAS,GAAG,IAAI,sBAAW,CAAC,SAAS,CAAC,CAAC;QAE5C,MAAM,KAAK,GAAiB;YAC1B,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC;YACpB,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;SAC/C,CAAC;QAEF,MAAM,GAAG,GAAG;YACV,GAAG,OAAO;YACV,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,OAAO;YACV,KAAK,CAAC,MAAM,GAAG,CAAC;gBACd,CAAC,CAAC,IAAI,kBAAO,CAAU,EAAE,EAAE,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC;gBAC/C,CAAC,CAAC,IAAI,yBAAgB,CAAU,EAAE,EAAE,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QAE7D,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CACb,gEAAgE,GAAG,IAAI,CACxE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CACb,4FAA4F;gBAC1F,WAAW,CACd,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAExE,IAAI,CAAC,IAAI,GAAG;YACV,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;YAC5B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE;YACzC,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO;SACzB,CAAC;QAEF,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEM,WAAW,CAAC,QAAqC;QACtD,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAEM,KAAK,CAAC,EAAU;QACrB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC;IAEM,WAAW,CAAC,QAAgB;QACjC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAEM,SAAS;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC;IAEM,QAAQ,CAAC,KAAgC;QAC9C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAEM,QAAQ;QACb,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;IACjC,CAAC;IAEM,UAAU;QACf,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAC5B,CAAC;IAEM,WAAW;QAChB,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,OAAO,CAClB,MAAU,EACV,OAA2C;QAE3C,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,UAAU,GAAG;YACjB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,IAAI;YACzB,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,WAAW;YACvC,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,YAAY,IAAI,MAAM;SACjD,CAAC;QAEF,OAAO,MAAM,OAAO,EAAE,MAAM,CAAC,eAAe,CAC1C,OAAO,EACP;YACE,IAAI,EAAE,qBAAU,CAAC,MAAM;YACvB,UAAU;SACX,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACxD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF;AArID,0BAqIC;AAED,SAAS,WAAW,CAAC,WAAmB;IACtC,8FAA8F;IAC9F,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAEhD,wFAAwF;IACxF,MAAM,eAAe,GAAG,KAAK;SAC1B,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACnB,qEAAqE;QACrE,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAErC,gEAAgE;QAChE,IAAI,KAAK,GAAG,CAAC,IAAI,SAAS,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3C,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;SACD,IAAI,CAAC,EAAE,CAAC,CAAC;IAEZ,OAAO,eAAe,CAAC;AACzB,CAAC"}

package/cjs/prompts/react.js

@@ -8,8 +8,13 @@ class AxReAct extends cot_js_1.AxChainOfThought {
         if (!options?.functions || options.functions.length === 0) {
             throw new Error('No functions provided');
         }
-        const functions = [...options.functions];
-        const funcList = functions.map((f) => `'${f.name}'`).join(', ');
+        const fnNames = options.functions.map((f) => {
+            if ('toFunction' in f) {
+                return f.toFunction().name;
+            }
+            return f.name;
+        });
+        const funcList = fnNames.map((fname) => `'${fname}'`).join(', ');
         const sig = new sig_js_1.AxSignature(signature);
         sig.setDescription(`Use the provided functions ${funcList} to complete the task and return the result if any.`);
         // sig.addInputField({

package/cjs/prompts/react.js.map

@@ -1 +1 @@
-{"version":3,"file":"react.js","sourceRoot":"","sources":["../../../prompts/react.ts"],"names":[],"mappings":";;;AAGA,0CAA4C;AAE5C,qCAA4C;AAE5C,MAAa,OAGX,SAAQ,yBAAyB;IACjC,YACE,EAAe,EACf,SAAyC,EACzC,OAAoC;QAEpC,IAAI,CAAC,OAAO,EAAE,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1D,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,SAAS,GAAG,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;QACzC,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEhE,MAAM,GAAG,GAAG,IAAI,oBAAW,CAAC,SAAS,CAAC,CAAC;QACvC,GAAG,CAAC,cAAc,CAChB,8BAA8B,QAAQ,qDAAqD,CAC5F,CAAC;QAEF,sBAAsB;QACtB,yBAAyB;QACzB,wDAAwD;QACxD,sBAAsB;QACtB,4CAA4C;QAC5C,MAAM;QAEN,GAAG,CAAC,eAAe,CACjB,GAAG,CAAC,eAAe,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAC/D,CAAC;QAEF,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF;AAlCD,0BAkCC"}
+{"version":3,"file":"react.js","sourceRoot":"","sources":["../../../prompts/react.ts"],"names":[],"mappings":";;;AAGA,0CAA4C;AAE5C,qCAA4C;AAE5C,MAAa,OAGX,SAAQ,yBAAyB;IACjC,YACE,EAAe,EACf,SAAyC,EACzC,OAAoC;QAEpC,IAAI,CAAC,OAAO,EAAE,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1D,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YAC1C,IAAI,YAAY,IAAI,CAAC,EAAE,CAAC;gBACtB,OAAO,CAAC,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC;YAC7B,CAAC;YACD,OAAO,CAAC,CAAC,IAAI,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEjE,MAAM,GAAG,GAAG,IAAI,oBAAW,CAAC,SAAS,CAAC,CAAC;QACvC,GAAG,CAAC,cAAc,CAChB,8BAA8B,QAAQ,qDAAqD,CAC5F,CAAC;QAEF,sBAAsB;QACtB,yBAAyB;QACzB,wDAAwD;QACxD,sBAAsB;QACtB,4CAA4C;QAC5C,MAAM;QAEN,GAAG,CAAC,eAAe,CACjB,GAAG,CAAC,eAAe,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAC/D,CAAC;QAEF,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF;AAxCD,0BAwCC"}

package/dsp/functions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"functions.js","sourceRoot":"","sources":["../../dsp/functions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAI1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAarD,MAAM,OAAO,mBAAmB;IACtB,QAAQ,CAAwB;IAExC,YAAY,QAA+B;QACzC,QAAQ;aACL,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;aAC3B,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC,UAAW,CAAC,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAEO,eAAe,GAAG,KAAK,EAC7B,MAA4B,EAC5B,IAA0C,EAC1C,OAA4C,EACnB,EAAE;QAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,wBAAwB,MAAM,CAAC,IAAI,kBAAkB,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,IAAI,CAAC;QAET,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1D,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;aAAM,CAAC;YACN,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QAED,MAAM,GAAG,GAAG,OAAO;YACjB,CAAC,CAAC;gBACE,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;aACzB;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YACvB,MAAM,GAAG,GACP,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAE1E,OAAO;gBACL,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;aACrC,CAAC;QACJ,CAAC;QAED,MAAM,GAAG,GACP,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;YACtB,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,CAAC;YAC9B,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE9B,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;SACrC,CAAC;IACJ,CAAC,CAAC;IAEK,OAAO,GAAG,KAAK,EACpB,IAA0C,EAC1C,OAA4C,EACnB,EAAE;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAC7C,CAAC;QACF,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,sBAAsB,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC;QAED,+BAA+B;QAC/B,OAAO,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAC3D,CAAC,CAAC;CACH"}

package/dsp/generate.d.ts

@@ -1,9 +1,9 @@
 import type { AxAIService, AxChatRequest, AxFunction, AxRateLimiterFunction } from '../ai/types.js';
-import { type AxChatResponseFunctionCall } from '../funcs/functions.js';
 import { type AxAIMemory, AxMemory } from '../mem/index.js';
 import { type AxTracer } from '../trace/index.js';
 import { type AxAssertion, type AxStreamingAssertion } from './asserts.js';
-import { type AxGenIn, type AxGenOut, AxProgram, type AxProgramForwardOptions } from './program.js';
+import { type AxChatResponseFunctionCall } from './functions.js';
+import { type AxGenIn, type AxGenOut, type AxProgramForwardOptions, AxProgramWithSignature } from './program.js';
 import { AxPromptTemplate } from './prompt.js';
 import { AxSignature } from './sig.js';
 export interface AxGenerateOptions {
@@ -15,7 +15,9 @@ export interface AxGenerateOptions {
     rateLimiter?: AxRateLimiterFunction;
     stream?: boolean;
     debug?: boolean;
-    functions?: AxFunction[];
+    functions?: AxFunction[] | {
+        toFunction: () => AxFunction;
+    }[];
     functionCall?: AxChatRequest['functionCall'];
     promptTemplate?: typeof AxPromptTemplate;
     asserts?: AxAssertion[];
@@ -34,11 +36,13 @@ export interface AxResponseHandlerArgs<T> {
     sessionId?: string;
     traceId?: string;
 }
-export declare class AxGenerate<IN extends AxGenIn = AxGenIn, OUT extends AxGenerateResult<AxGenOut> = AxGenerateResult<AxGenOut>> extends AxProgram<IN, OUT> {
+export declare class AxGenerate<IN extends AxGenIn = AxGenIn, OUT extends AxGenerateResult<AxGenOut> = AxGenerateResult<AxGenOut>> extends AxProgramWithSignature<IN, OUT> {
+    private ai;
     private pt;
     private asserts;
     private streamingAsserts;
     private options?;
+    private functions?;
     private funcProc?;
     private functionList?;
     constructor(ai: AxAIService, signature: Readonly<AxSignature | string>, options?: Readonly<AxGenerateOptions>);