@ax-llm/ax 9.0.25 → 9.0.28

package/funcs/docker.d.ts

@@ -1,26 +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, // Example format: [{ hostPath: '/host/path', containerPath: '/container/path' }]
-    doNotPullImage }: Readonly<{
+    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>;
-    executeCommand(command: string): Promise<string>;
-    stopContainer(): Promise<void>;
-    listContainers(all?: boolean): Promise<unknown>;
+    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

@@ -14,24 +14,28 @@ export class AxDockerSession {
         // Wait for the pull to complete
         await response.text();
     }
-    async createContainer({ imageName, volumes = [], // Example format: [{ hostPath: '/host/path', containerPath: '/container/path' }]
-    doNotPullImage }) {
+    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({
-                Image: imageName,
-                Tty: true,
-                OpenStdin: false,
-                AttachStdin: false,
-                AttachStdout: false,
-                AttachStderr: false,
-                HostConfig: { Binds: binds }
-            })
+            body: JSON.stringify(containerConfig)
         });
         if (!response.ok) {
             throw new Error(`Failed to create container: ${response.statusText}`);
@@ -40,6 +44,29 @@ export class AxDockerSession {
         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');
@@ -58,10 +85,61 @@ export class AxDockerSession {
         }
         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',
@@ -91,29 +169,24 @@ export class AxDockerSession {
         // 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'
-        });
+    // 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 stop container: ${response.statusText}`);
+            throw new Error(`Failed to get container info: ${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');
+    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
         }
-        const response = await this.fetchDockerAPI(`/containers/${this.containerId}/logs?stdout=true&stderr=true`, { method: 'GET' });
-        return response.text();
+        throw new Error('Timeout waiting for container to start');
     }
     async fetchDockerAPI(endpoint, options) {
         const url = new URL(endpoint, this.apiUrl).toString();
@@ -133,7 +206,7 @@ export class AxDockerSession {
                 },
                 required: ['command']
             },
-            func: this.executeCommand
+            func: async ({ command }) => await this.executeCommand(command)
         };
     }
 }

package/funcs/docker.js.map

@@ -1 +1 @@
-{"version":3,"file":"docker.js","sourceRoot":"","sources":["../../funcs/docker.ts"],"names":[],"mappings":"AAEA,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,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"}
+{"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.js

@@ -31,7 +31,7 @@ export class AxEmbeddingAdapter {
                 },
                 required: ['text']
             },
-            func: this.embedAdapter
+            func: ({ text }, options) => this.embedAdapter(text, options)
         };
     }
 }

package/funcs/embed.js.map

@@ -1 +1 @@
-{"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,IAAI,CAAC,YAAY;SACxB,CAAC;IACJ,CAAC;CACF"}
+{"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/package.json

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