@ax-llm/ax 9.0.24 → 9.0.26

package/funcs/code.d.ts

@@ -1,9 +1,16 @@
-import type { AxFunction } from '../ai/index.js';
-export declare const axJSInterpreterFunction: (permissions?: readonly AxCodeInterpreterPermission[]) => AxFunction;
-export declare enum AxCodeInterpreterPermission {
+import type { AxFunction } from '../ai/types.js';
+export declare enum AxJSInterpreterPermission {
     FS = "node:fs",
     NET = "net",
     OS = "os",
     CRYPTO = "crypto",
     PROCESS = "process"
 }
+export declare class AxJSInterpreter {
+    private permissions;
+    constructor({ permissions }?: Readonly<{
+        permissions?: readonly AxJSInterpreterPermission[];
+    }> | undefined);
+    private codeInterpreterJavascript;
+    toFunction(): AxFunction;
+}

package/funcs/code.js

@@ -5,59 +5,55 @@ import * as _fs from 'node:fs';
 import * as _os from 'os';
 import * as _process from 'process';
 import { runInNewContext } from 'vm';
-export 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));
-        });
+export var AxJSInterpreterPermission;
+(function (AxJSInterpreterPermission) {
+    AxJSInterpreterPermission["FS"] = "node:fs";
+    AxJSInterpreterPermission["NET"] = "net";
+    AxJSInterpreterPermission["OS"] = "os";
+    AxJSInterpreterPermission["CRYPTO"] = "crypto";
+    AxJSInterpreterPermission["PROCESS"] = "process";
+})(AxJSInterpreterPermission || (AxJSInterpreterPermission = {}));
+export class AxJSInterpreter {
+    permissions;
+    constructor({ permissions = [] } = {}) {
+        this.permissions = permissions ?? [];
     }
-});
-export var AxCodeInterpreterPermission;
-(function (AxCodeInterpreterPermission) {
-    AxCodeInterpreterPermission["FS"] = "node:fs";
-    AxCodeInterpreterPermission["NET"] = "net";
-    AxCodeInterpreterPermission["OS"] = "os";
-    AxCodeInterpreterPermission["CRYPTO"] = "crypto";
-    AxCodeInterpreterPermission["PROCESS"] = "process";
-})(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;
+    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 runInNewContext(`(function() { ${code} })()`, context);
     }
-    if (permissions.includes(AxCodeInterpreterPermission.NET)) {
-        context.http = _http;
-        context.https = _https;
+    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: ({ code }) => this.codeInterpreterJavascript(code)
+        };
     }
-    if (permissions.includes(AxCodeInterpreterPermission.OS)) {
-        context.os = _os;
-    }
-    if (permissions.includes(AxCodeInterpreterPermission.CRYPTO)) {
-        context.crypto = _crypto;
-    }
-    if (permissions.includes(AxCodeInterpreterPermission.PROCESS)) {
-        context.process = _process;
-    }
-    // executing code within the sandbox
-    return runInNewContext(`(function() { ${code} })()`, context);
-};
+}
 //# sourceMappingURL=code.js.map

package/funcs/code.js.map

@@ -1 +1 @@
-{"version":3,"file":"code.js","sourceRoot":"","sources":["../../funcs/code.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,OAAO,MAAM,QAAQ,CAAC;AAClC,OAAO,KAAK,KAAK,MAAM,MAAM,CAAC;AAC9B,OAAO,KAAK,MAAM,MAAM,OAAO,CAAC;AAChC,OAAO,KAAK,GAAG,MAAM,SAAS,CAAC;AAC/B,OAAO,KAAK,GAAG,MAAM,IAAI,CAAC;AAC1B,OAAO,KAAK,QAAQ,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC;AAIrC,MAAM,CAAC,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;AAEH,MAAM,CAAN,IAAY,2BAMX;AAND,WAAY,2BAA2B;IACrC,6CAAc,CAAA;IACd,0CAAW,CAAA;IACX,wCAAS,CAAA;IACT,gDAAiB,CAAA;IACjB,kDAAmB,CAAA;AACrB,CAAC,EANW,2BAA2B,KAA3B,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,eAAe,CAAC,iBAAiB,IAAI,OAAO,EAAE,OAAO,CAAC,CAAC;AAChE,CAAC,CAAC"}
+{"version":3,"file":"code.js","sourceRoot":"","sources":["../../funcs/code.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,OAAO,MAAM,QAAQ,CAAC;AAClC,OAAO,KAAK,KAAK,MAAM,MAAM,CAAC;AAC9B,OAAO,KAAK,MAAM,MAAM,OAAO,CAAC;AAChC,OAAO,KAAK,GAAG,MAAM,SAAS,CAAC;AAC/B,OAAO,KAAK,GAAG,MAAM,IAAI,CAAC;AAC1B,OAAO,KAAK,QAAQ,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC;AAIrC,MAAM,CAAN,IAAY,yBAMX;AAND,WAAY,yBAAyB;IACnC,2CAAc,CAAA;IACd,wCAAW,CAAA;IACX,sCAAS,CAAA;IACT,8CAAiB,CAAA;IACjB,gDAAmB,CAAA;AACrB,CAAC,EANW,yBAAyB,KAAzB,yBAAyB,QAMpC;AAED,MAAM,OAAO,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,eAAe,CAAC,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,CAAC,EAAE,IAAI,EAA8B,EAAE,EAAE,CAC7C,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC;SACvC,CAAC;IACJ,CAAC;CACF"}

package/funcs/docker.d.ts

@@ -0,0 +1,101 @@
+import type { AxFunction } from '../ai/types.js';
+export interface AxDockerContainer {
+    Id: string;
+    Names: string[];
+    Image: string;
+    ImageID: string;
+    Command: string;
+    Created: number;
+    State: {
+        Status: string;
+        Running: boolean;
+        Paused: boolean;
+        Restarting: boolean;
+        OOMKilled: boolean;
+        Dead: boolean;
+        Pid: number;
+        ExitCode: number;
+        Error: string;
+        StartedAt: Date;
+        FinishedAt: Date;
+    };
+    Status: string;
+    Ports: Array<{
+        IP: string;
+        PrivatePort: number;
+        PublicPort: number;
+        Type: string;
+    }>;
+    Labels: {
+        [key: string]: string;
+    };
+    SizeRw: number;
+    SizeRootFs: number;
+    HostConfig: {
+        NetworkMode: string;
+    };
+    NetworkSettings: {
+        Networks: {
+            [key: string]: {
+                IPAddress: string;
+                IPPrefixLen: number;
+                Gateway: string;
+                MacAddress: string;
+            };
+        };
+    };
+    Mounts: Array<{
+        Type: string;
+        Source: string;
+        Destination: string;
+        Mode: string;
+        RW: boolean;
+        Propagation: string;
+    }>;
+}
+export declare class AxDockerSession {
+    private readonly apiUrl;
+    private containerId;
+    constructor(apiUrl?: string);
+    pullImage(imageName: string): Promise<void>;
+    createContainer({ imageName, volumes, doNotPullImage, tag }: Readonly<{
+        imageName: string;
+        volumes?: Array<{
+            hostPath: string;
+            containerPath: string;
+        }>;
+        doNotPullImage?: boolean;
+        tag?: string;
+    }>): Promise<{
+        Id: string;
+    }>;
+    findOrCreateContainer({ imageName, volumes, doNotPullImage, tag }: Readonly<{
+        imageName: string;
+        volumes?: Array<{
+            hostPath: string;
+            containerPath: string;
+        }>;
+        doNotPullImage?: boolean;
+        tag: string;
+    }>): Promise<{
+        Id: string;
+        isNew: boolean;
+    }>;
+    startContainer(): Promise<void>;
+    connectToContainer(containerId: string): Promise<void>;
+    stopContainers({ tag, remove, timeout }: Readonly<{
+        tag?: string;
+        remove?: boolean;
+        timeout?: number;
+    }>): Promise<Array<{
+        Id: string;
+        Action: 'stopped' | 'removed';
+    }>>;
+    listContainers(all?: boolean): Promise<AxDockerContainer[]>;
+    getContainerLogs(): Promise<string>;
+    executeCommand(command: string): Promise<string>;
+    private getContainerInfo;
+    private waitForContainerToBeRunning;
+    private fetchDockerAPI;
+    toFunction(): AxFunction;
+}

package/funcs/docker.js

@@ -0,0 +1,213 @@
+export 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 = [], doNotPullImage, tag }) {
+        const binds = volumes.map((v) => `${v.hostPath}:${v.containerPath}`);
+        if (!doNotPullImage) {
+            await this.pullImage(imageName);
+        }
+        const containerConfig = {
+            Image: imageName,
+            Tty: true,
+            OpenStdin: false,
+            AttachStdin: false,
+            AttachStdout: false,
+            AttachStderr: false,
+            HostConfig: { Binds: binds },
+            Labels: {}
+        };
+        if (tag) {
+            containerConfig.Labels['com.example.tag'] = tag;
+        }
+        const response = await this.fetchDockerAPI(`/containers/create`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify(containerConfig)
+        });
+        if (!response.ok) {
+            throw new Error(`Failed to create container: ${response.statusText}`);
+        }
+        const data = (await response.json());
+        this.containerId = data.Id;
+        return data;
+    }
+    async findOrCreateContainer({ imageName, volumes = [], doNotPullImage, tag }) {
+        // First, try to find existing containers with the given tag
+        const existingContainers = await this.listContainers(true);
+        const matchingContainers = existingContainers.filter((container) => container.Labels && container.Labels['com.example.tag'] === tag);
+        if (matchingContainers && matchingContainers.length > 0) {
+            // Randomly select a container from the matching ones
+            const randomIndex = Math.floor(Math.random() * matchingContainers.length);
+            const selectedContainer = matchingContainers[randomIndex];
+            if (selectedContainer) {
+                // Connect to the selected container
+                await this.connectToContainer(selectedContainer.Id);
+                return { Id: selectedContainer.Id, isNew: false };
+            }
+        }
+        // If no container with the tag exists, create a new one
+        const newContainer = await this.createContainer({
+            imageName,
+            volumes,
+            doNotPullImage,
+            tag
+        });
+        return { Id: newContainer.Id, isNew: true };
+    }
+    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 stopContainers({ tag, remove, timeout = 10 }) {
+        const results = [];
+        // List all containers
+        const containers = await this.listContainers(true);
+        // Filter containers by tag if provided
+        const targetContainers = tag
+            ? containers.filter((container) => container.Labels['com.example.tag'] === tag)
+            : containers;
+        for (const container of targetContainers) {
+            // Stop the container if it's running
+            if (container.State.Status === 'running') {
+                const stopResponse = await this.fetchDockerAPI(`/containers/${container.Id}/stop?t=${timeout}`, { method: 'POST' });
+                if (!stopResponse.ok) {
+                    console.warn(`Failed to stop container ${container.Id}: ${stopResponse.statusText}`);
+                    continue;
+                }
+                results.push({ Id: container.Id, Action: 'stopped' });
+            }
+            // Remove the container if the remove flag is set
+            if (remove) {
+                const removeResponse = await this.fetchDockerAPI(`/containers/${container.Id}`, { method: 'DELETE' });
+                if (!removeResponse.ok) {
+                    console.warn(`Failed to remove container ${container.Id}: ${removeResponse.statusText}`);
+                    continue;
+                }
+                results.push({ Id: container.Id, Action: 'removed' });
+            }
+        }
+        return results;
+    }
+    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 executeCommand(command) {
+        console.log('Executing command:', command);
+        if (!this.containerId) {
+            throw new Error('No container created or connected');
+        }
+        // Check container state
+        const containerInfo = await this.getContainerInfo(this.containerId);
+        if (containerInfo.State.Status !== 'running') {
+            await this.startContainer();
+            // Wait for the container to be in the "running" state
+            await this.waitForContainerToBeRunning(this.containerId);
+        }
+        // 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();
+    }
+    // Add these new methods to the class:
+    async getContainerInfo(containerId) {
+        const response = await this.fetchDockerAPI(`/containers/${containerId}/json`);
+        if (!response.ok) {
+            throw new Error(`Failed to get container info: ${response.statusText}`);
+        }
+        return response.json();
+    }
+    async waitForContainerToBeRunning(containerId, timeout = 30000) {
+        const startTime = Date.now();
+        while (Date.now() - startTime < timeout) {
+            const containerInfo = await this.getContainerInfo(containerId);
+            if (containerInfo.State.Status === 'running') {
+                return;
+            }
+            await new Promise((resolve) => setTimeout(resolve, 1000)); // Wait for 1 second before checking again
+        }
+        throw new Error('Timeout waiting for container to start');
+    }
+    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: async ({ command }) => await this.executeCommand(command)
+        };
+    }
+}
+//# sourceMappingURL=docker.js.map

package/funcs/docker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"docker.js","sourceRoot":"","sources":["../../funcs/docker.ts"],"names":[],"mappings":"AAuDA,MAAM,OAAO,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,EACZ,cAAc,EACd,GAAG,EAMH;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,eAAe,GAAG;YACtB,KAAK,EAAE,SAAS;YAChB,GAAG,EAAE,IAAI;YACT,SAAS,EAAE,KAAK;YAChB,WAAW,EAAE,KAAK;YAClB,YAAY,EAAE,KAAK;YACnB,YAAY,EAAE,KAAK;YACnB,UAAU,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;YAC5B,MAAM,EAAE,EAA4B;SACrC,CAAC;QAEF,IAAI,GAAG,EAAE,CAAC;YACR,eAAe,CAAC,MAAM,CAAC,iBAAiB,CAAC,GAAG,GAAG,CAAC;QAClD,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,eAAe,CAAC;SACtC,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,qBAAqB,CAAC,EAC1B,SAAS,EACT,OAAO,GAAG,EAAE,EACZ,cAAc,EACd,GAAG,EAMH;QACA,4DAA4D;QAC5D,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAC3D,MAAM,kBAAkB,GAAG,kBAAkB,CAAC,MAAM,CAClD,CAAC,SAAS,EAAE,EAAE,CACZ,SAAS,CAAC,MAAM,IAAI,SAAS,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,GAAG,CAClE,CAAC;QAEF,IAAI,kBAAkB,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,qDAAqD;YACrD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC1E,MAAM,iBAAiB,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC;YAE1D,IAAI,iBAAiB,EAAE,CAAC;gBACtB,oCAAoC;gBACpC,MAAM,IAAI,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;gBACpD,OAAO,EAAE,EAAE,EAAE,iBAAiB,CAAC,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;YACpD,CAAC;QACH,CAAC;QAED,wDAAwD;QACxD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC;YAC9C,SAAS;YACT,OAAO;YACP,cAAc;YACd,GAAG;SACJ,CAAC,CAAC;QAEH,OAAO,EAAE,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IAC9C,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,EACnB,GAAG,EACH,MAAM,EACN,OAAO,GAAG,EAAE,EACmD;QAG/D,MAAM,OAAO,GAAyD,EAAE,CAAC;QAEzE,sBAAsB;QACtB,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAEnD,uCAAuC;QACvC,MAAM,gBAAgB,GAAG,GAAG;YAC1B,CAAC,CAAC,UAAU,CAAC,MAAM,CACf,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,GAAG,CAC3D;YACH,CAAC,CAAC,UAAU,CAAC;QAEf,KAAK,MAAM,SAAS,IAAI,gBAAgB,EAAE,CAAC;YACzC,qCAAqC;YACrC,IAAI,SAAS,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBACzC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,cAAc,CAC5C,eAAe,SAAS,CAAC,EAAE,WAAW,OAAO,EAAE,EAC/C,EAAE,MAAM,EAAE,MAAM,EAAE,CACnB,CAAC;gBAEF,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,CAAC;oBACrB,OAAO,CAAC,IAAI,CACV,4BAA4B,SAAS,CAAC,EAAE,KAAK,YAAY,CAAC,UAAU,EAAE,CACvE,CAAC;oBACF,SAAS;gBACX,CAAC;gBAED,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;YACxD,CAAC;YAED,iDAAiD;YACjD,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,cAAc,CAC9C,eAAe,SAAS,CAAC,EAAE,EAAE,EAC7B,EAAE,MAAM,EAAE,QAAQ,EAAE,CACrB,CAAC;gBAEF,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC;oBACvB,OAAO,CAAC,IAAI,CACV,8BAA8B,SAAS,CAAC,EAAE,KAAK,cAAc,CAAC,UAAU,EAAE,CAC3E,CAAC;oBACF,SAAS;gBACX,CAAC;gBAED,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,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,EAAkC,CAAC;IACzD,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;IAED,KAAK,CAAC,cAAc,CAAC,OAAe;QAClC,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC;QAE3C,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,wBAAwB;QACxB,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAEpE,IAAI,aAAa,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC7C,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;YAE5B,sDAAsD;YACtD,MAAM,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC3D,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,sCAAsC;IAE9B,KAAK,CAAC,gBAAgB,CAC5B,WAAmB;QAEnB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CACxC,eAAe,WAAW,OAAO,CAClC,CAAC;QACF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,iCAAiC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QAC1E,CAAC;QACD,OAAO,QAAQ,CAAC,IAAI,EAAgC,CAAC;IACvD,CAAC;IAEO,KAAK,CAAC,2BAA2B,CACvC,WAAmB,EACnB,UAAkB,KAAK;QAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,OAAO,EAAE,CAAC;YACxC,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAC/D,IAAI,aAAa,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC7C,OAAO;YACT,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,0CAA0C;QACvG,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,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,KAAK,EAAE,EAAE,OAAO,EAAiC,EAAE,EAAE,CACzD,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC;SACrC,CAAC;IACJ,CAAC;CACF"}

package/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/funcs/embed.js

@@ -1,29 +1,38 @@
-export 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));
-        });
+export class AxEmbeddingAdapter {
+    aiService;
+    info;
+    func;
+    constructor({ ai, info, func }) {
+        this.aiService = ai;
+        this.info = info;
+        this.func = func;
     }
-});
-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: ({ text }, options) => this.embedAdapter(text, options)
+        };
+    }
+}
 //# sourceMappingURL=embed.js.map

package/funcs/embed.js.map

@@ -1 +1 @@
-{"version":3,"file":"embed.js","sourceRoot":"","sources":["../../funcs/embed.ts"],"names":[],"mappings":"AAOA,MAAM,CAAC,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;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,MAAM,OAAO,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,CAAC,EAAE,IAAI,EAA8B,EAAE,OAAO,EAAE,EAAE,CACtD,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC;SACnC,CAAC;IACJ,CAAC;CACF"}

package/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/funcs/index.js

@@ -1,4 +1,4 @@
 export * from './code.js';
 export * from './embed.js';
-export * from './functions.js';
+export * from './docker.js';
 //# sourceMappingURL=index.js.map

package/funcs/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../funcs/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC;AAC1B,cAAc,YAAY,CAAC;AAC3B,cAAc,gBAAgB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../funcs/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC;AAC1B,cAAc,YAAY,CAAC;AAC3B,cAAc,aAAa,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ax-llm/ax",
-  "version": "9.0.24",
+  "version": "9.0.26",
   "type": "module",
   "description": "The best library to work with LLMs",
   "repository": {

package/prompts/react.js

@@ -5,8 +5,13 @@ export class AxReAct extends 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 AxSignature(signature);
         sig.setDescription(`Use the provided functions ${funcList} to complete the task and return the result if any.`);
         // sig.addInputField({

package/prompts/react.js.map

@@ -1 +1 @@
-{"version":3,"file":"react.js","sourceRoot":"","sources":["../../prompts/react.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,OAAO,OAGX,SAAQ,gBAAyB;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,WAAW,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"}
+{"version":3,"file":"react.js","sourceRoot":"","sources":["../../prompts/react.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,OAAO,OAGX,SAAQ,gBAAyB;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,WAAW,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"}

package/cjs/funcs/functions.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"functions.js","sourceRoot":"","sources":["../../../funcs/functions.ts"],"names":[],"mappings":";;;;;;AAAA,kDAA0B;AAI1B,mDAAqD;AAarD,MAAa,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,IAAA,kCAAkB,EAAC,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,eAAK,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;AAxED,kDAwEC"}

package/cjs/funcs/jsonschema.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"jsonschema.js","sourceRoot":"","sources":["../../../funcs/jsonschema.ts"],"names":[],"mappings":";;;AAEA,gEAAgE;AACzD,MAAM,kBAAkB,GAAG,CAChC,MAAsC,EAChC,EAAE;IACR,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,MAAM,oBAAoB,GAAG,CAC3B,MAAsC,EACtC,OAAe,EAAE,EACX,EAAE;QACR,MAAM,UAAU,GAAG;YACjB,OAAO;YACP,SAAS;YACT,QAAQ;YACR,QAAQ;YACR,SAAS;YACT,MAAM;YACN,QAAQ;SACT,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,MAAM,CAAC,IAAI,CAAC,iBAAiB,MAAM,CAAC,IAAI,QAAQ,IAAI,IAAI,MAAM,EAAE,CAAC,CAAC;YAClE,OAAO;QACT,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YAClD,IACE,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ;gBACrC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EAChC,CAAC;gBACD,MAAM,CAAC,IAAI,CAAC,gCAAgC,IAAI,IAAI,MAAM,EAAE,CAAC,CAAC;YAChE,CAAC;iBAAM,CAAC;gBACN,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;oBACpC,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;oBACrC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;wBAC9B,MAAM,CAAC,IAAI,CAAC,4BAA4B,IAAI,GAAG,GAAG,EAAE,CAAC,CAAC;wBACtD,SAAS;oBACX,CAAC;oBACD,oBAAoB,CAAC,KAAK,EAAE,GAAG,IAAI,GAAG,GAAG,GAAG,CAAC,CAAC;gBAChD,CAAC;YACH,CAAC;YAED,IAAI,MAAM,CAAC,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACvD,MAAM,CAAC,IAAI,CAAC,oCAAoC,IAAI,IAAI,MAAM,EAAE,CAAC,CAAC;YACpE,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAC5C,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACrC,MAAM,CAAC,IAAI,CAAC,2BAA2B,IAAI,IAAI,MAAM,EAAE,CAAC,CAAC;YAC3D,CAAC;iBAAM,CAAC;gBACN,oBAAoB,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,IAAI,QAAQ,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IAEF,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAE7B,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACrC,CAAC;AACH,CAAC,CAAC;AA5DW,QAAA,kBAAkB,sBA4D7B;AAEF,iBAAiB;AAEjB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA4CE"}