@axiom-lattice/client-sdk 1.0.43 → 1.0.44

package/dist/abstract-client.js

@@ -0,0 +1,279 @@
+/**
+ * Abstract client class for interacting with the Axiom Lattice Agent Service API
+ * Provides common functionality for different client implementations
+ */
+export class AbstractClient {
+    /**
+     * Creates a new AbstractClient instance
+     * @param config - Configuration options for the client
+     */
+    constructor(config) {
+        this.tenantId = "";
+        this.registeredTools = new Map();
+        /**
+         * Chat namespace for sending messages and streaming responses
+         */
+        this.chat = {
+            /**
+             * Sends a message to a thread and receives a response
+             * @param options - Options for sending a message
+             * @returns A promise that resolves to the chat response
+             */
+            send: async (options) => {
+                try {
+                    // Extract the message from the messages array (assuming the last message is the one to send)
+                    const message = options.messages[options.messages.length - 1];
+                    const { command, threadId, files, ...rest } = options;
+                    // Use the run method to send the message
+                    const result = await this.run({
+                        threadId: threadId,
+                        message: typeof message.content === "string"
+                            ? message.content
+                            : JSON.stringify(message.content),
+                        streaming: false,
+                        command: command,
+                        files: files,
+                        ...rest,
+                    });
+                    return result;
+                }
+                catch (error) {
+                    throw error;
+                }
+            },
+            /**
+             * Sends a message to a thread and streams the response
+             * @param options - Options for streaming a message
+             * @param onEvent - Callback function that receives stream events
+             * @param onComplete - Optional callback function called when streaming completes
+             * @param onError - Optional callback function called when an error occurs
+             * @returns A function that can be called to stop the stream
+             */
+            stream: (options, onEvent, onComplete, onError) => {
+                // Extract the message from the messages array (assuming the last message is the one to send)
+                const message = options.messages[options.messages.length - 1];
+                // Use the streamRequest method to stream the response
+                return this.streamRequest({
+                    threadId: options.threadId,
+                    message: typeof message.content === "string"
+                        ? message.content
+                        : JSON.stringify(message.content),
+                    streaming: true,
+                    command: options.command,
+                    background: options.background,
+                    enableReturnStateWhenSteamCompleted: options.enableReturnStateWhenSteamCompleted,
+                }, onEvent, onComplete, onError);
+            },
+        };
+        /**
+         * Tools namespace for registering and unregistering client-side tools
+         */
+        this.tools = {
+            /**
+             * Registers a client-side tool
+             * @param options - Options for registering a tool
+             */
+            register: (options) => {
+                if (this.config.transport !== "ws") {
+                    throw new Error("Client-side tools are only supported with WebSocket transport");
+                }
+                this.registeredTools.set(options.name, options);
+            },
+            /**
+             * Unregisters a client-side tool
+             * @param name - Tool name
+             */
+            unregister: (name) => {
+                this.registeredTools.delete(name);
+            },
+        };
+        this.config = {
+            timeout: 300000, // Default timeout
+            ...config,
+        };
+        this.assistantId = config.assistantId;
+    }
+    /**
+     * Abstract method for streaming API requests
+     * To be implemented by concrete client classes
+     */
+    /**
+     * Helper method to build stream request parameters
+     * @param options - Options for running the agent
+     * @returns The formatted request parameters
+     */
+    buildStreamRequestParams(options) {
+        const { command, threadId, message, files, background, ...rest } = options;
+        return {
+            url: "/api/runs",
+            method: "POST",
+            body: {
+                assistant_id: this.assistantId,
+                thread_id: threadId,
+                message: message,
+                files: files,
+                command: command,
+                streaming: true,
+                background: background || false,
+                ...rest,
+            },
+        };
+    }
+    /**
+     * Creates a new thread
+     * @param options - Options for creating a thread
+     * @returns A promise that resolves to the thread ID
+     */
+    async createThread(options) {
+        try {
+            const data = await this.makeRequest("/threads", {
+                method: "POST",
+                body: {
+                    ...options,
+                    assistantId: this.assistantId,
+                },
+            });
+            return data.id;
+        }
+        catch (error) {
+            throw error;
+        }
+    }
+    /**
+     * Retrieves thread information
+     * @param threadId - Thread identifier
+     * @returns A promise that resolves to the thread information
+     */
+    async getThread(threadId) {
+        try {
+            const url = `/threads/${threadId}?assistantId=${this.assistantId}`;
+            return await this.makeRequest(url);
+        }
+        catch (error) {
+            throw error;
+        }
+    }
+    /**
+     * Lists all threads
+     * @param options - Options for listing threads
+     * @returns A promise that resolves to an array of threads
+     */
+    async listThreads(options) {
+        try {
+            let url = `/threads?assistantId=${this.assistantId}`;
+            if (options?.limit) {
+                url += `&limit=${options.limit}`;
+            }
+            if (options?.offset !== undefined) {
+                url += `&offset=${options.offset}`;
+            }
+            const data = await this.makeRequest(url);
+            return data.threads;
+        }
+        catch (error) {
+            throw error;
+        }
+    }
+    /**
+     * Deletes a thread
+     * @param threadId - Thread identifier
+     * @returns A promise that resolves when the thread is deleted
+     */
+    async deleteThread(threadId) {
+        try {
+            const url = `/threads/${threadId}?assistantId=${this.assistantId}`;
+            await this.makeRequest(url, { method: "DELETE" });
+        }
+        catch (error) {
+            throw error;
+        }
+    }
+    /**
+     * Retrieves messages from a thread
+     * @param options - Options for retrieving messages
+     * @returns A promise that resolves to an array of messages
+     */
+    async getMessages(options) {
+        try {
+            let url = `/api/assistants/${this.assistantId}/${options.threadId}/memory`;
+            if (options.limit) {
+                url += `?limit=${options.limit}`;
+            }
+            if (options.after) {
+                url += url.includes("?")
+                    ? `&after=${options.after}`
+                    : `?after=${options.after}`;
+            }
+            if (options.reverse !== undefined) {
+                url += url.includes("?")
+                    ? `&reverse=${options.reverse}`
+                    : `?reverse=${options.reverse}`;
+            }
+            url += url.includes("?")
+                ? `&assistantId=${this.assistantId}`
+                : `?assistantId=${this.assistantId}`;
+            return await this.makeRequest(url);
+        }
+        catch (error) {
+            throw error;
+        }
+    }
+    /**
+     * Retrieves agent state
+     * @param threadId - Thread identifier
+     * @returns A promise that resolves to the agent state
+     */
+    async getAgentState(threadId) {
+        try {
+            return await this.makeRequest(`/api/assistants/${this.assistantId}/${threadId}/state`);
+        }
+        catch (error) {
+            throw error;
+        }
+    }
+    /**
+     * Gets agent graph visualization
+     * @returns A promise that resolves to the graph visualization data
+     */
+    async getAgentGraph() {
+        try {
+            const data = await this.makeRequest(`/api/assistants/${this.assistantId}/graph`);
+            return data.image;
+        }
+        catch (error) {
+            throw error;
+        }
+    }
+    /**
+     * Run agent with options
+     * @param options - Options for running the agent
+     * @returns A promise that resolves to the run result
+     */
+    async run(options) {
+        try {
+            const { command, threadId, message, files, background, ...rest } = options;
+            if (options.streaming) {
+                throw new Error("Streaming without callbacks is not supported. Use chat.stream with callbacks instead.");
+            }
+            else {
+                return await this.makeRequest("/api/runs", {
+                    method: "POST",
+                    body: {
+                        assistant_id: this.assistantId,
+                        thread_id: threadId,
+                        message: message,
+                        files: files,
+                        command: command,
+                        streaming: false,
+                        background: background || false,
+                        ...rest,
+                    },
+                });
+            }
+        }
+        catch (error) {
+            throw error;
+        }
+    }
+}
+//# sourceMappingURL=abstract-client.js.map

package/dist/abstract-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"abstract-client.js","sourceRoot":"","sources":["../src/abstract-client.ts"],"names":[],"mappings":"AAgBA;;;GAGG;AACH,MAAM,OAAgB,cAAc;IAMlC;;;OAGG;IACH,YAAY,MAAoB;QAPtB,aAAQ,GAAW,EAAE,CAAC;QACtB,oBAAe,GAAqC,IAAI,GAAG,EAAE,CAAC;QAoPxE;;WAEG;QACH,SAAI,GAAG;YACL;;;;eAIG;YACH,IAAI,EAAE,KAAK,EAAE,OAAwB,EAAyB,EAAE;gBAC9D,IAAI,CAAC;oBACH,6FAA6F;oBAC7F,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBAC9D,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;oBAEtD,yCAAyC;oBACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC;wBAC5B,QAAQ,EAAE,QAAQ;wBAClB,OAAO,EACL,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;4BACjC,CAAC,CAAC,OAAO,CAAC,OAAO;4BACjB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;wBACrC,SAAS,EAAE,KAAK;wBAChB,OAAO,EAAE,OAAO;wBAChB,KAAK,EAAE,KAAK;wBACZ,GAAG,IAAI;qBACR,CAAC,CAAC;oBAEH,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;YAED;;;;;;;eAOG;YACH,MAAM,EAAE,CACN,OAA0B,EAC1B,OAAsC,EACtC,UAAyC,EACzC,OAAgC,EAClB,EAAE;gBAChB,6FAA6F;gBAC7F,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAE9D,sDAAsD;gBACtD,OAAO,IAAI,CAAC,aAAa,CACvB;oBACE,QAAQ,EAAE,OAAO,CAAC,QAAQ;oBAC1B,OAAO,EACL,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;wBACjC,CAAC,CAAC,OAAO,CAAC,OAAO;wBACjB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;oBACrC,SAAS,EAAE,IAAI;oBACf,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,UAAU,EAAE,OAAO,CAAC,UAAU;oBAC9B,mCAAmC,EACjC,OAAO,CAAC,mCAAmC;iBAC9C,EACD,OAAO,EACP,UAAU,EACV,OAAO,CACR,CAAC;YACJ,CAAC;SACF,CAAC;QAEF;;WAEG;QACH,UAAK,GAAG;YACN;;;eAGG;YACH,QAAQ,EAAE,CAAC,OAA4B,EAAQ,EAAE;gBAC/C,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,KAAK,IAAI,EAAE,CAAC;oBACnC,MAAM,IAAI,KAAK,CACb,+DAA+D,CAChE,CAAC;gBACJ,CAAC;gBACD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YAClD,CAAC;YAED;;;eAGG;YACH,UAAU,EAAE,CAAC,IAAY,EAAQ,EAAE;gBACjC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACpC,CAAC;SACF,CAAC;QA7UA,IAAI,CAAC,MAAM,GAAG;YACZ,OAAO,EAAE,MAAM,EAAE,kBAAkB;YACnC,GAAG,MAAM;SACV,CAAC;QAEF,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;IACxC,CAAC;IAqBD;;;OAGG;IACH;;;;OAIG;IACO,wBAAwB,CAAC,OAAmB;QAKpD,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;QAE3E,OAAO;YACL,GAAG,EAAE,WAAW;YAChB,MAAM,EAAE,MAAM;YACd,IAAI,EAAE;gBACJ,YAAY,EAAE,IAAI,CAAC,WAAW;gBAC9B,SAAS,EAAE,QAAQ;gBACnB,OAAO,EAAE,OAAO;gBAChB,KAAK,EAAE,KAAK;gBACZ,OAAO,EAAE,OAAO;gBAChB,SAAS,EAAE,IAAI;gBACf,UAAU,EAAE,UAAU,IAAI,KAAK;gBAC/B,GAAG,IAAI;aACR;SACF,CAAC;IACJ,CAAC;IASD;;;;OAIG;IACH,KAAK,CAAC,YAAY,CAAC,OAA4B;QAC7C,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAiB,UAAU,EAAE;gBAC9D,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE;oBACJ,GAAG,OAAO;oBACV,WAAW,EAAE,IAAI,CAAC,WAAW;iBAC9B;aACF,CAAC,CAAC;YACH,OAAO,IAAI,CAAC,EAAE,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,SAAS,CAAC,QAAgB;QAC9B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,YAAY,QAAQ,gBAAgB,IAAI,CAAC,WAAW,EAAE,CAAC;YACnE,OAAO,MAAM,IAAI,CAAC,WAAW,CAAS,GAAG,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW,CAAC,OAA4B;QAC5C,IAAI,CAAC;YACH,IAAI,GAAG,GAAG,wBAAwB,IAAI,CAAC,WAAW,EAAE,CAAC;YAErD,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;gBACnB,GAAG,IAAI,UAAU,OAAO,CAAC,KAAK,EAAE,CAAC;YACnC,CAAC;YAED,IAAI,OAAO,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;gBAClC,GAAG,IAAI,WAAW,OAAO,CAAC,MAAM,EAAE,CAAC;YACrC,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAwB,GAAG,CAAC,CAAC;YAChE,OAAO,IAAI,CAAC,OAAO,CAAC;QACtB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,YAAY,CAAC,QAAgB;QACjC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,YAAY,QAAQ,gBAAgB,IAAI,CAAC,WAAW,EAAE,CAAC;YACnE,MAAM,IAAI,CAAC,WAAW,CAAO,GAAG,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC1D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW,CAAC,OAA2B;QAC3C,IAAI,CAAC;YACH,IAAI,GAAG,GAAG,mBAAmB,IAAI,CAAC,WAAW,IAAI,OAAO,CAAC,QAAQ,SAAS,CAAC;YAE3E,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAClB,GAAG,IAAI,UAAU,OAAO,CAAC,KAAK,EAAE,CAAC;YACnC,CAAC;YAED,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAClB,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC;oBACtB,CAAC,CAAC,UAAU,OAAO,CAAC,KAAK,EAAE;oBAC3B,CAAC,CAAC,UAAU,OAAO,CAAC,KAAK,EAAE,CAAC;YAChC,CAAC;YAED,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;gBAClC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC;oBACtB,CAAC,CAAC,YAAY,OAAO,CAAC,OAAO,EAAE;oBAC/B,CAAC,CAAC,YAAY,OAAO,CAAC,OAAO,EAAE,CAAC;YACpC,CAAC;YAED,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC;gBACtB,CAAC,CAAC,gBAAgB,IAAI,CAAC,WAAW,EAAE;gBACpC,CAAC,CAAC,gBAAgB,IAAI,CAAC,WAAW,EAAE,CAAC;YAEvC,OAAO,MAAM,IAAI,CAAC,WAAW,CAAY,GAAG,CAAC,CAAC;QAChD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,aAAa,CAAC,QAAgB;QAClC,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,WAAW,CAC3B,mBAAmB,IAAI,CAAC,WAAW,IAAI,QAAQ,QAAQ,CACxD,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,aAAa;QACjB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CACjC,mBAAmB,IAAI,CAAC,WAAW,QAAQ,CAC5C,CAAC;YACF,OAAO,IAAI,CAAC,KAAK,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,GAAG,CAAC,OAAmB;QAC3B,IAAI,CAAC;YACH,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,IAAI,EAAE,GAC9D,OAAO,CAAC;YAEV,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CACb,uFAAuF,CACxF,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,OAAO,MAAM,IAAI,CAAC,WAAW,CAAe,WAAW,EAAE;oBACvD,MAAM,EAAE,MAAM;oBACd,IAAI,EAAE;wBACJ,YAAY,EAAE,IAAI,CAAC,WAAW;wBAC9B,SAAS,EAAE,QAAQ;wBACnB,OAAO,EAAE,OAAO;wBAChB,KAAK,EAAE,KAAK;wBACZ,OAAO,EAAE,OAAO;wBAChB,SAAS,EAAE,KAAK;wBAChB,UAAU,EAAE,UAAU,IAAI,KAAK;wBAC/B,GAAG,IAAI;qBACR;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CAmGF"}

package/dist/client.d.ts

@@ -0,0 +1,54 @@
+import { MessageChunk } from "@axiom-lattice/protocols";
+import { AgentState, ClientConfig, RunOptions } from "./types";
+import { AbstractClient } from "./abstract-client";
+/**
+ * Web client class for interacting with the Axiom Lattice Agent Service API
+ */
+export declare class Client extends AbstractClient {
+    private headers;
+    /**
+     * Creates a new Client instance
+     * @param config - Configuration options for the client
+     */
+    constructor(config: ClientConfig);
+    /**
+     * Helper method to handle fetch responses and errors
+     * @private
+     */
+    private handleResponse;
+    /**
+     * Helper method to make fetch requests
+     * @private
+     */
+    private fetchWithTimeout;
+    /**
+     * Set tenant ID for multi-tenant environments
+     * @param tenantId - Tenant identifier
+     */
+    setTenantId(tenantId: string): void;
+    /**
+     * Implementation of the abstract makeRequest method for web clients
+     * @param url - The URL to make the request to
+     * @param options - Request options
+     * @returns A promise that resolves to the response data
+     */
+    protected makeRequest<T>(url: string, options?: {
+        method?: string;
+        body?: any;
+        headers?: Record<string, string>;
+    }): Promise<T>;
+    /**
+     * Implementation of the abstract streamRequest method for web clients
+     */
+    protected streamRequest(options: RunOptions, onEvent: (event: MessageChunk) => void, onComplete?: (state?: AgentState) => void, onError?: (error: Error) => void): () => void;
+    /**
+     * Stream run results
+     * @param options - Options for streaming run results
+     * @param onEvent - Callback function that receives stream events
+     * @param onComplete - Optional callback function called when streaming completes
+     * @param onError - Optional callback function called when an error occurs
+     * @returns A function that can be called to stop the stream
+     */
+    private streamRun;
+}
+//# sourceMappingURL=client.d.ts.map

package/dist/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAW,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EACL,UAAU,EAMV,YAAY,EAMZ,UAAU,EAEX,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD;;GAEG;AACH,qBAAa,MAAO,SAAQ,cAAc;IACxC,OAAO,CAAC,OAAO,CAAyB;IAExC;;;OAGG;gBACS,MAAM,EAAE,YAAY;IAUhC;;;OAGG;YACW,cAAc;IA4B5B;;;OAGG;YACW,gBAAgB;IAqC9B;;;OAGG;IACH,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAKnC;;;;;OAKG;cACa,WAAW,CAAC,CAAC,EAC3B,GAAG,EAAE,MAAM,EACX,OAAO,CAAC,EAAE;QACR,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,GAAG,CAAC;QACX,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAClC,GACA,OAAO,CAAC,CAAC,CAAC;IAmBb;;OAEG;IACH,SAAS,CAAC,aAAa,CACrB,OAAO,EAAE,UAAU,EACnB,OAAO,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,EACtC,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,UAAU,KAAK,IAAI,EACzC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,GAC/B,MAAM,IAAI;IAIb;;;;;;;OAOG;IACH,OAAO,CAAC,SAAS;CAuHlB"}

package/dist/client.js

@@ -0,0 +1,224 @@
+import { ApiError, AuthenticationError, NetworkError, } from "./types";
+import { AbstractClient } from "./abstract-client";
+/**
+ * Web client class for interacting with the Axiom Lattice Agent Service API
+ */
+export class Client extends AbstractClient {
+    /**
+     * Creates a new Client instance
+     * @param config - Configuration options for the client
+     */
+    constructor(config) {
+        super(config);
+        this.headers = {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${this.config.apiKey}`,
+            ...this.config.headers,
+        };
+    }
+    /**
+     * Helper method to handle fetch responses and errors
+     * @private
+     */
+    async handleResponse(response) {
+        if (!response.ok) {
+            if (response.status === 401) {
+                throw new AuthenticationError("Authentication failed");
+            }
+            else {
+                const errorData = (await response.json().catch(() => ({})));
+                throw new ApiError(errorData.message || "API Error", response.status, errorData);
+            }
+        }
+        // For 204 No Content responses
+        if (response.status === 204) {
+            return {};
+        }
+        try {
+            return (await response.json());
+        }
+        catch (error) {
+            throw new Error(`Failed to parse response: ${error}`);
+        }
+    }
+    /**
+     * Helper method to make fetch requests
+     * @private
+     */
+    async fetchWithTimeout(url, options = {}) {
+        const controller = new AbortController();
+        const timeoutId = setTimeout(() => controller.abort(), this.config.timeout);
+        try {
+            const fullUrl = url.startsWith("http")
+                ? url
+                : `${this.config.baseURL}${url}`;
+            const response = await fetch(fullUrl, {
+                ...options,
+                headers: {
+                    ...this.headers,
+                    ...(options.headers || {}),
+                },
+                signal: controller.signal,
+            });
+            return await this.handleResponse(response);
+        }
+        catch (error) {
+            if (error instanceof DOMException && error.name === "AbortError") {
+                throw new NetworkError("Request timed out");
+            }
+            else if (error instanceof TypeError &&
+                error.message.includes("fetch")) {
+                throw new NetworkError("Network request failed");
+            }
+            throw error;
+        }
+        finally {
+            clearTimeout(timeoutId);
+        }
+    }
+    /**
+     * Set tenant ID for multi-tenant environments
+     * @param tenantId - Tenant identifier
+     */
+    setTenantId(tenantId) {
+        this.tenantId = tenantId;
+        this.headers["x-tenant-id"] = tenantId;
+    }
+    /**
+     * Implementation of the abstract makeRequest method for web clients
+     * @param url - The URL to make the request to
+     * @param options - Request options
+     * @returns A promise that resolves to the response data
+     */
+    async makeRequest(url, options) {
+        const method = options?.method || "GET";
+        const requestHeaders = {
+            ...this.headers,
+            ...(options?.headers || {}),
+        };
+        const requestOptions = {
+            method,
+            headers: requestHeaders,
+        };
+        if (options?.body) {
+            requestOptions.body = JSON.stringify(options.body);
+        }
+        return this.fetchWithTimeout(url, requestOptions);
+    }
+    /**
+     * Implementation of the abstract streamRequest method for web clients
+     */
+    streamRequest(options, onEvent, onComplete, onError) {
+        return this.streamRun(options, onEvent, onComplete, onError);
+    }
+    /**
+     * Stream run results
+     * @param options - Options for streaming run results
+     * @param onEvent - Callback function that receives stream events
+     * @param onComplete - Optional callback function called when streaming completes
+     * @param onError - Optional callback function called when an error occurs
+     * @returns A function that can be called to stop the stream
+     */
+    streamRun(options, onEvent, onComplete, onError) {
+        const headers = {
+            "Content-Type": "application/json",
+            Accept: "text/event-stream",
+            ...this.headers,
+        };
+        if (this.tenantId) {
+            headers["x-tenant-id"] = this.tenantId;
+        }
+        // Create abort controller for cancellation
+        const controller = new AbortController();
+        const { signal } = controller;
+        // Start the streaming request
+        (async () => {
+            try {
+                // Get request parameters from the abstract method
+                const requestParams = this.buildStreamRequestParams(options);
+                const response = await fetch(`${this.config.baseURL}${requestParams.url}`, {
+                    method: requestParams.method,
+                    headers,
+                    body: JSON.stringify(requestParams.body),
+                    signal,
+                });
+                if (!response.ok) {
+                    throw new Error(`HTTP error! Status: ${response.status}`);
+                }
+                if (!response.body) {
+                    throw new Error("Response body is null");
+                }
+                const reader = response.body.getReader();
+                const decoder = new TextDecoder();
+                let buffer = "";
+                while (true) {
+                    const { done, value } = await reader.read();
+                    if (done)
+                        break;
+                    // Decode the chunk and append to buffer
+                    const chunk = decoder.decode(value, { stream: true });
+                    buffer += chunk;
+                    // Process complete lines
+                    const lines = buffer.split("\n");
+                    // Keep the last potentially incomplete line in the buffer
+                    buffer = lines.pop() || "";
+                    for (const line of lines) {
+                        if (line.trim().startsWith("data: ")) {
+                            try {
+                                const eventData = JSON.parse(line.trim().slice(6));
+                                onEvent(eventData);
+                            }
+                            catch (error) {
+                                console.error("Error parsing SSE data:", line, error);
+                                if (onError) {
+                                    onError(error instanceof Error ? error : new Error(String(error)));
+                                }
+                            }
+                        }
+                    }
+                }
+                // Process any remaining data in the buffer
+                if (buffer && buffer.trim().startsWith("data: ")) {
+                    try {
+                        const eventData = JSON.parse(buffer.trim().slice(6));
+                        onEvent(eventData);
+                    }
+                    catch (error) {
+                        console.error("Error parsing SSE data:", buffer, error);
+                    }
+                }
+                if (onComplete) {
+                    // Fetch agent state before calling onComplete if enabled
+                    if (options.enableReturnStateWhenSteamCompleted) {
+                        try {
+                            const state = await this.getAgentState(options.threadId);
+                            onComplete(state);
+                        }
+                        catch (error) {
+                            // If getting state fails, still call onComplete without state
+                            onComplete();
+                        }
+                    }
+                    else {
+                        // Just call onComplete without state if not enabled
+                        onComplete();
+                    }
+                }
+            }
+            catch (error) {
+                if (error instanceof DOMException && error.name === "AbortError") {
+                    // Request was cancelled, no need to call onError
+                    return;
+                }
+                if (onError) {
+                    onError(error instanceof Error ? error : new Error(String(error)));
+                }
+            }
+        })();
+        // Return a function to cancel the fetch request
+        return () => {
+            controller.abort();
+        };
+    }
+}
+//# sourceMappingURL=client.js.map

package/dist/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AACA,OAAO,EAEL,QAAQ,EACR,mBAAmB,EAQnB,YAAY,GAIb,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD;;GAEG;AACH,MAAM,OAAO,MAAO,SAAQ,cAAc;IAGxC;;;OAGG;IACH,YAAY,MAAoB;QAC9B,KAAK,CAAC,MAAM,CAAC,CAAC;QAEd,IAAI,CAAC,OAAO,GAAG;YACb,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;YAC7C,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO;SACvB,CAAC;IACJ,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,cAAc,CAAI,QAAkB;QAChD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,MAAM,IAAI,mBAAmB,CAAC,uBAAuB,CAAC,CAAC;YACzD,CAAC;iBAAM,CAAC;gBACN,MAAM,SAAS,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAEzD,CAAC;gBACF,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,OAAO,IAAI,WAAW,EAChC,QAAQ,CAAC,MAAM,EACf,SAAS,CACV,CAAC;YACJ,CAAC;QACH,CAAC;QAED,+BAA+B;QAC/B,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YAC5B,OAAO,EAAO,CAAC;QACjB,CAAC;QAED,IAAI,CAAC;YACH,OAAO,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAM,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,6BAA6B,KAAK,EAAE,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,gBAAgB,CAC5B,GAAW,EACX,UAAuB,EAAE;QAEzB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAE5E,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC;gBACpC,CAAC,CAAC,GAAG;gBACL,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC;YAEnC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,OAAO,EAAE;gBACpC,GAAG,OAAO;gBACV,OAAO,EAAE;oBACP,GAAG,IAAI,CAAC,OAAO;oBACf,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;iBAC3B;gBACD,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,OAAO,MAAM,IAAI,CAAC,cAAc,CAAI,QAAQ,CAAC,CAAC;QAChD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACjE,MAAM,IAAI,YAAY,CAAC,mBAAmB,CAAC,CAAC;YAC9C,CAAC;iBAAM,IACL,KAAK,YAAY,SAAS;gBAC1B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,EAC/B,CAAC;gBACD,MAAM,IAAI,YAAY,CAAC,wBAAwB,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,SAAS,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,WAAW,CAAC,QAAgB;QAC1B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,QAAQ,CAAC;IACzC,CAAC;IAED;;;;;OAKG;IACO,KAAK,CAAC,WAAW,CACzB,GAAW,EACX,OAIC;QAED,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,KAAK,CAAC;QACxC,MAAM,cAAc,GAAG;YACrB,GAAG,IAAI,CAAC,OAAO;YACf,GAAG,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;SAC5B,CAAC;QAEF,MAAM,cAAc,GAAgB;YAClC,MAAM;YACN,OAAO,EAAE,cAAc;SACxB,CAAC;QAEF,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;YAClB,cAAc,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACrD,CAAC;QAED,OAAO,IAAI,CAAC,gBAAgB,CAAI,GAAG,EAAE,cAAc,CAAC,CAAC;IACvD,CAAC;IAED;;OAEG;IACO,aAAa,CACrB,OAAmB,EACnB,OAAsC,EACtC,UAAyC,EACzC,OAAgC;QAEhC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IAC/D,CAAC;IAED;;;;;;;OAOG;IACK,SAAS,CACf,OAAmB,EACnB,OAAsC,EACtC,UAAyC,EACzC,OAAgC;QAEhC,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,kBAAkB;YAClC,MAAM,EAAE,mBAAmB;YAC3B,GAAG,IAAI,CAAC,OAAO;SAChB,CAAC;QAEF,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC;QACzC,CAAC;QAED,2CAA2C;QAC3C,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC;QAE9B,8BAA8B;QAC9B,CAAC,KAAK,IAAI,EAAE;YACV,IAAI,CAAC;gBACH,kDAAkD;gBAClD,MAAM,aAAa,GAAG,IAAI,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;gBAC7D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,aAAa,CAAC,GAAG,EAAE,EAC5C;oBACE,MAAM,EAAE,aAAa,CAAC,MAAM;oBAC5B,OAAO;oBACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC;oBACxC,MAAM;iBACP,CACF,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,MAAM,IAAI,KAAK,CAAC,uBAAuB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC5D,CAAC;gBAED,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACnB,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;gBAC3C,CAAC;gBAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACzC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;gBAClC,IAAI,MAAM,GAAG,EAAE,CAAC;gBAEhB,OAAO,IAAI,EAAE,CAAC;oBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;oBAC5C,IAAI,IAAI;wBAAE,MAAM;oBAEhB,wCAAwC;oBACxC,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;oBACtD,MAAM,IAAI,KAAK,CAAC;oBAEhB,yBAAyB;oBACzB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACjC,0DAA0D;oBAC1D,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;oBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;wBACzB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;4BACrC,IAAI,CAAC;gCACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gCACnD,OAAO,CAAC,SAAS,CAAC,CAAC;4BACrB,CAAC;4BAAC,OAAO,KAAK,EAAE,CAAC;gCACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;gCACtD,IAAI,OAAO,EAAE,CAAC;oCACZ,OAAO,CACL,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAC1D,CAAC;gCACJ,CAAC;4BACH,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,2CAA2C;gBAC3C,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACjD,IAAI,CAAC;wBACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;wBACrD,OAAO,CAAC,SAAS,CAAC,CAAC;oBACrB,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;oBAC1D,CAAC;gBACH,CAAC;gBAED,IAAI,UAAU,EAAE,CAAC;oBACf,yDAAyD;oBACzD,IAAI,OAAO,CAAC,mCAAmC,EAAE,CAAC;wBAChD,IAAI,CAAC;4BACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;4BACzD,UAAU,CAAC,KAAK,CAAC,CAAC;wBACpB,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,8DAA8D;4BAC9D,UAAU,EAAE,CAAC;wBACf,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,oDAAoD;wBACpD,UAAU,EAAE,CAAC;oBACf,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBACjE,iDAAiD;oBACjD,OAAO;gBACT,CAAC;gBAED,IAAI,OAAO,EAAE,CAAC;oBACZ,OAAO,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACrE,CAAC;YACH,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,gDAAgD;QAChD,OAAO,GAAG,EAAE;YACV,UAAU,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC,CAAC;IACJ,CAAC;CACF"}