@axiom-lattice/client-sdk 1.0.43 → 1.0.44

package/dist/ChunkMessageMerger.d.ts

@@ -0,0 +1,18 @@
+/**
+ * ChunkMessageMerger
+ *
+ * A utility for handling streaming message chunks and merging them into complete messages.
+ */
+import { MessageChunk } from "@axiom-lattice/protocols";
+/**
+ * Creates a simple message merger for handling streaming message chunks
+ * @returns An object with methods to push chunks and retrieve merged messages
+ */
+export declare function createSimpleMessageMerger(): {
+    push: (chunk: MessageChunk) => void;
+    initialMessages: (msgs: any[]) => void;
+    getMessages: () => any[];
+    getMessagesWithoutToolCalls: () => any[];
+    reset: () => void;
+};
+//# sourceMappingURL=ChunkMessageMerger.d.ts.map

package/dist/ChunkMessageMerger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChunkMessageMerger.d.ts","sourceRoot":"","sources":["../src/ChunkMessageMerger.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAW,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAEjE;;;GAGG;AACH,wBAAgB,yBAAyB;kBAwIlB,YAAY;4BATF,GAAG,EAAE;uBAmGZ,GAAG,EAAE;uCAiDW,GAAG,EAAE;;EAwB9C"}

package/dist/ChunkMessageMerger.js

@@ -0,0 +1,264 @@
+/**
+ * ChunkMessageMerger
+ *
+ * A utility for handling streaming message chunks and merging them into complete messages.
+ */
+/**
+ * Creates a simple message merger for handling streaming message chunks
+ * @returns An object with methods to push chunks and retrieve merged messages
+ */
+export function createSimpleMessageMerger() {
+    let messages = [];
+    const messageMap = new Map();
+    // Tool call builder: messageId -> index -> build state
+    const toolBuilders = new Map();
+    /**
+     * Normalizes role string to standard role
+     * @param type - Role string from the input
+     * @returns Normalized role
+     */
+    function normalizeRole(type) {
+        switch (type) {
+            case "ai":
+                return "ai";
+            case "human":
+            case "user":
+                return "human";
+            case "system":
+                return "system";
+            case "tool":
+                return "tool";
+            case "developer":
+                return "developer";
+            default:
+                return "ai";
+        }
+    }
+    /**
+     * Safely parses JSON string
+     * @param str - JSON string to parse
+     * @returns Parsed object or original string if parsing fails
+     */
+    function safeJsonParse(str) {
+        try {
+            return JSON.parse(str);
+        }
+        catch {
+            return str; // Return original string if parsing fails
+        }
+    }
+    /**
+     * Ensures a message exists, creating it if needed
+     * @param id - Message ID
+     * @param role - Message role
+     * @returns The message object
+     */
+    function ensureMessage(id, role) {
+        const index = messageMap.get(id);
+        if (index !== undefined) {
+            return messages[index];
+        }
+        // Create new message
+        let newMessage;
+        switch (role) {
+            case "human":
+                newMessage = { id, role: "human", content: "" };
+                break;
+            case "system":
+                newMessage = { id, role: "system", content: "" };
+                break;
+            case "tool":
+                newMessage = { id, role: "tool", content: "", tool_call_id: "" };
+                break;
+            case "developer":
+                newMessage = { id, role: "developer", content: "" };
+                break;
+            default:
+                newMessage = { id, role: "ai", content: "" };
+        }
+        messages.push(newMessage);
+        messageMap.set(id, messages.length - 1);
+        return newMessage;
+    }
+    /**
+     * Updates tool calls for a message
+     * @param message - Message to update
+     */
+    function updateToolCalls(message) {
+        if (message.role !== "ai")
+            return;
+        const builders = toolBuilders.get(message.id);
+        if (!builders || builders.size === 0)
+            return;
+        const toolCalls = [];
+        for (const [index, builder] of Array.from(builders.entries())) {
+            if (builder.name && builder.id) {
+                // Try to parse arguments as JSON object
+                const args = builder.args.trim();
+                const parsedArgs = safeJsonParse(args);
+                // Create ToolCall object
+                toolCalls.push({
+                    id: builder.id,
+                    name: builder.name,
+                    args: typeof parsedArgs === "object" ? parsedArgs : { value: parsedArgs },
+                    type: "tool_call",
+                });
+            }
+        }
+        // Only set tool_calls if there are any
+        if (toolCalls.length > 0) {
+            message.tool_calls = toolCalls;
+        }
+    }
+    /**
+     * Initialize messages
+     * @param msgs - Initial messages
+     */
+    function initialMessages(msgs) {
+        // Just store the messages as-is
+        messages = msgs;
+    }
+    /**
+     * Processes a message chunk
+     * @param chunk - Message chunk to process
+     */
+    function push(chunk) {
+        const role = normalizeRole(chunk.type);
+        const message = ensureMessage(chunk.data.id, role);
+        // Accumulate content
+        if (chunk.data.content) {
+            message.content = (message.content || "") + chunk.data.content;
+        }
+        // Handle tool call ID (for tool messages)
+        if (message.role === "tool" && chunk.data.tool_call_id) {
+            message.tool_call_id = chunk.data.tool_call_id;
+        }
+        // Handle tool call chunks (incremental)
+        if (chunk.data.tool_call_chunks && message.role === "ai") {
+            let builders = toolBuilders.get(message.id);
+            if (!builders) {
+                builders = new Map();
+                toolBuilders.set(message.id, builders);
+            }
+            for (const chunk_item of chunk.data.tool_call_chunks) {
+                let builder = builders.get(chunk_item.index);
+                if (!builder) {
+                    builder = { args: "" };
+                    builders.set(chunk_item.index, builder);
+                }
+                if (chunk_item.name)
+                    builder.name = chunk_item.name;
+                if (chunk_item.id)
+                    builder.id = chunk_item.id;
+                if (chunk_item.args)
+                    builder.args += chunk_item.args;
+            }
+            updateToolCalls(message);
+        }
+        // Handle complete tool calls (overwrite)
+        if (chunk.data.tool_calls && message.role === "ai") {
+            const toolCalls = chunk.data.tool_calls
+                .filter((tc) => tc.name && tc.id)
+                .map((tc) => ({
+                id: tc.id,
+                name: tc.name,
+                args: tc.args,
+                type: "tool_call",
+                response: tc.response,
+                status: "success",
+            }));
+            if (toolCalls.length > 0) {
+                message.tool_calls = toolCalls;
+            }
+        }
+        // Handle tool calls in additional_kwargs
+        if (chunk.data.additional_kwargs?.tool_calls &&
+            message.role === "ai" &&
+            !message.tool_calls) {
+            const toolCalls = [];
+            for (const tc of chunk.data.additional_kwargs.tool_calls) {
+                if (tc.function?.name && tc.id) {
+                    // Convert to our format
+                    const args = typeof tc.function.arguments === "string"
+                        ? safeJsonParse(tc.function.arguments)
+                        : tc.function.arguments || {};
+                    toolCalls.push({
+                        id: tc.id,
+                        name: tc.function.name,
+                        args,
+                        type: "tool_call",
+                    });
+                }
+            }
+            if (toolCalls.length > 0) {
+                message.tool_calls = toolCalls;
+            }
+        }
+    }
+    /**
+     * Gets all messages with tool responses merged into tool calls
+     * @returns Array of messages with tool responses merged into tool calls
+     */
+    function getMessages() {
+        // Create a map to store tool responses by their tool_call_id
+        const toolResponsesMap = {};
+        // First pass: collect all tool responses
+        messages.forEach((message) => {
+            if (message.role === "tool" && message.tool_call_id) {
+                toolResponsesMap[message.tool_call_id] = message;
+            }
+        });
+        // Second pass: merge tool responses with their corresponding AI messages
+        return messages
+            .filter((message) => {
+            // Keep all messages except tool messages
+            return message.role !== "tool";
+        })
+            .map((message) => {
+            if (message.role === "ai" && message.tool_calls) {
+                // Create a new message object with merged tool responses
+                const mergedMessage = { ...message };
+                // Add tool responses to the corresponding tool calls
+                if (mergedMessage.tool_calls) {
+                    mergedMessage.tool_calls = mergedMessage.tool_calls.map((toolCall) => {
+                        const toolResponse = toolResponsesMap[toolCall.id];
+                        if (toolResponse) {
+                            return {
+                                ...toolCall,
+                                response: toolResponse.content,
+                                status: "success",
+                            };
+                        }
+                        return toolCall;
+                    });
+                }
+                return mergedMessage;
+            }
+            return message;
+        });
+    }
+    /**
+     * Gets messages without tool calls
+     * @returns Array of messages without tool calls
+     */
+    function getMessagesWithoutToolCalls() {
+        return messages.filter((message) => message.role !== "tool" &&
+            (message.role !== "ai" || !message.tool_calls));
+    }
+    /**
+     * Resets the message merger state
+     */
+    function reset() {
+        messages.length = 0;
+        messageMap.clear();
+        toolBuilders.clear();
+    }
+    return {
+        push,
+        initialMessages,
+        getMessages,
+        getMessagesWithoutToolCalls,
+        reset,
+    };
+}
+//# sourceMappingURL=ChunkMessageMerger.js.map

package/dist/ChunkMessageMerger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChunkMessageMerger.js","sourceRoot":"","sources":["../src/ChunkMessageMerger.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH;;;GAGG;AACH,MAAM,UAAU,yBAAyB;IACvC,IAAI,QAAQ,GAAc,EAAE,CAAC;IAC7B,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;IAE7C,uDAAuD;IACvD,MAAM,YAAY,GAAG,IAAI,GAAG,EAUzB,CAAC;IAEJ;;;;OAIG;IACH,SAAS,aAAa,CAAC,IAAY;QACjC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,IAAI;gBACP,OAAO,IAAI,CAAC;YACd,KAAK,OAAO,CAAC;YACb,KAAK,MAAM;gBACT,OAAO,OAAO,CAAC;YACjB,KAAK,QAAQ;gBACX,OAAO,QAAQ,CAAC;YAClB,KAAK,MAAM;gBACT,OAAO,MAAM,CAAC;YAChB,KAAK,WAAW;gBACd,OAAO,WAAW,CAAC;YACrB;gBACE,OAAO,IAAI,CAAC;QAChB,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,SAAS,aAAa,CAAC,GAAW;QAChC,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,GAAG,CAAC,CAAC,0CAA0C;QACxD,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,SAAS,aAAa,CAAC,EAAU,EAAE,IAAY;QAC7C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;QAED,qBAAqB;QACrB,IAAI,UAAe,CAAC;QACpB,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,OAAO;gBACV,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;gBAChD,MAAM;YACR,KAAK,QAAQ;gBACX,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;gBACjD,MAAM;YACR,KAAK,MAAM;gBACT,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;gBACjE,MAAM;YACR,KAAK,WAAW;gBACd,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;gBACpD,MAAM;YACR;gBACE,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACjD,CAAC;QAED,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1B,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACxC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;OAGG;IACH,SAAS,eAAe,CAAC,OAAY;QACnC,IAAI,OAAO,CAAC,IAAI,KAAK,IAAI;YAAE,OAAO;QAElC,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9C,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QAE7C,MAAM,SAAS,GAAU,EAAE,CAAC;QAC5B,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YAC9D,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;gBAC/B,wCAAwC;gBACxC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACjC,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;gBAEvC,yBAAyB;gBACzB,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,OAAO,CAAC,EAAE;oBACd,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,IAAI,EACF,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE;oBACrE,IAAI,EAAE,WAAW;iBAClB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,uCAAuC;QACvC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;QACjC,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,SAAS,eAAe,CAAC,IAAW;QAClC,gCAAgC;QAChC,QAAQ,GAAG,IAAI,CAAC;IAClB,CAAC;IAED;;;OAGG;IACH,SAAS,IAAI,CAAC,KAAmB;QAC/B,MAAM,IAAI,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,OAAO,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAEnD,qBAAqB;QACrB,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACvB,OAAO,CAAC,OAAO,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;QACjE,CAAC;QAED,0CAA0C;QAC1C,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvD,OAAO,CAAC,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC;QACjD,CAAC;QAED,wCAAwC;QACxC,IAAI,KAAK,CAAC,IAAI,CAAC,gBAAgB,IAAI,OAAO,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACzD,IAAI,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC5C,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;gBACrB,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;YACzC,CAAC;YAED,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACrD,IAAI,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,OAAO,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;oBACvB,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;gBAC1C,CAAC;gBAED,IAAI,UAAU,CAAC,IAAI;oBAAE,OAAO,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;gBACpD,IAAI,UAAU,CAAC,EAAE;oBAAE,OAAO,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;gBAC9C,IAAI,UAAU,CAAC,IAAI;oBAAE,OAAO,CAAC,IAAI,IAAI,UAAU,CAAC,IAAI,CAAC;YACvD,CAAC;YAED,eAAe,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;QAED,yCAAyC;QACzC,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,IAAI,OAAO,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACnD,MAAM,SAAS,GAAU,KAAK,CAAC,IAAI,CAAC,UAAU;iBAC3C,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,CAAC;iBAChC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACZ,EAAE,EAAE,EAAE,CAAC,EAAE;gBACT,IAAI,EAAE,EAAE,CAAC,IAAI;gBACb,IAAI,EAAE,EAAE,CAAC,IAAI;gBACb,IAAI,EAAE,WAAW;gBACjB,QAAQ,EAAE,EAAE,CAAC,QAAQ;gBACrB,MAAM,EAAE,SAAS;aAClB,CAAC,CAAC,CAAC;YAEN,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;YACjC,CAAC;QACH,CAAC;QAED,yCAAyC;QACzC,IACE,KAAK,CAAC,IAAI,CAAC,iBAAiB,EAAE,UAAU;YACxC,OAAO,CAAC,IAAI,KAAK,IAAI;YACrB,CAAC,OAAO,CAAC,UAAU,EACnB,CAAC;YACD,MAAM,SAAS,GAAU,EAAE,CAAC;YAE5B,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC;gBACzD,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC/B,wBAAwB;oBACxB,MAAM,IAAI,GACR,OAAO,EAAE,CAAC,QAAQ,CAAC,SAAS,KAAK,QAAQ;wBACvC,CAAC,CAAC,aAAa,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;wBACtC,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC;oBAElC,SAAS,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,EAAE,CAAC,EAAE;wBACT,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;wBACtB,IAAI;wBACJ,IAAI,EAAE,WAAW;qBAClB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;YACjC,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,SAAS,WAAW;QAClB,6DAA6D;QAC7D,MAAM,gBAAgB,GAAwB,EAAE,CAAC;QAEjD,yCAAyC;QACzC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;gBACpD,gBAAgB,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC;YACnD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,yEAAyE;QACzE,OAAO,QAAQ;aACZ,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;YAClB,yCAAyC;YACzC,OAAO,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC;QACjC,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;YACf,IAAI,OAAO,CAAC,IAAI,KAAK,IAAI,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;gBAChD,yDAAyD;gBACzD,MAAM,aAAa,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;gBAErC,qDAAqD;gBACrD,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;oBAC7B,aAAa,CAAC,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC,GAAG,CACrD,CAAC,QAAa,EAAE,EAAE;wBAChB,MAAM,YAAY,GAAG,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;wBACnD,IAAI,YAAY,EAAE,CAAC;4BACjB,OAAO;gCACL,GAAG,QAAQ;gCACX,QAAQ,EAAE,YAAY,CAAC,OAAO;gCAC9B,MAAM,EAAE,SAAS;6BAClB,CAAC;wBACJ,CAAC;wBACD,OAAO,QAAQ,CAAC;oBAClB,CAAC,CACF,CAAC;gBACJ,CAAC;gBAED,OAAO,aAAa,CAAC;YACvB,CAAC;YACD,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC,CAAC;IACP,CAAC;IAED;;;OAGG;IACH,SAAS,2BAA2B;QAClC,OAAO,QAAQ,CAAC,MAAM,CACpB,CAAC,OAAO,EAAE,EAAE,CACV,OAAO,CAAC,IAAI,KAAK,MAAM;YACvB,CAAC,OAAO,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CACjD,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,SAAS,KAAK;QACZ,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QACpB,UAAU,CAAC,KAAK,EAAE,CAAC;QACnB,YAAY,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED,OAAO;QACL,IAAI;QACJ,eAAe;QACf,WAAW;QACX,2BAA2B;QAC3B,KAAK;KACN,CAAC;AACJ,CAAC"}

package/dist/abstract-client.d.ts

@@ -0,0 +1,129 @@
+import { Message, MessageChunk } from "@axiom-lattice/protocols";
+import { AgentState, ChatResponse, ChatSendOptions, ChatStreamOptions, ClientConfig, CreateThreadOptions, GetMessagesOptions, ListThreadsOptions, RegisterToolOptions, RunOptions, Thread } from "./types";
+/**
+ * Abstract client class for interacting with the Axiom Lattice Agent Service API
+ * Provides common functionality for different client implementations
+ */
+export declare abstract class AbstractClient {
+    protected config: ClientConfig;
+    protected assistantId: string;
+    protected tenantId: string;
+    protected registeredTools: Map<string, RegisterToolOptions>;
+    /**
+     * Creates a new AbstractClient instance
+     * @param config - Configuration options for the client
+     */
+    constructor(config: ClientConfig);
+    /**
+     * Set tenant ID for multi-tenant environments
+     * @param tenantId - Tenant identifier
+     */
+    abstract setTenantId(tenantId: string): void;
+    /**
+     * Abstract method for making API requests
+     * To be implemented by concrete client classes
+     */
+    protected abstract makeRequest<T>(url: string, options?: {
+        method?: string;
+        body?: any;
+        headers?: Record<string, string>;
+    }): Promise<T>;
+    /**
+     * 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
+     */
+    protected buildStreamRequestParams(options: RunOptions): {
+        url: string;
+        method: string;
+        body: any;
+    };
+    protected abstract streamRequest(options: RunOptions, onEvent: (event: MessageChunk) => void, onComplete?: (state?: AgentState) => void, onError?: (error: Error) => void): () => void;
+    /**
+     * Creates a new thread
+     * @param options - Options for creating a thread
+     * @returns A promise that resolves to the thread ID
+     */
+    createThread(options: CreateThreadOptions): Promise<string>;
+    /**
+     * Retrieves thread information
+     * @param threadId - Thread identifier
+     * @returns A promise that resolves to the thread information
+     */
+    getThread(threadId: string): Promise<Thread>;
+    /**
+     * Lists all threads
+     * @param options - Options for listing threads
+     * @returns A promise that resolves to an array of threads
+     */
+    listThreads(options?: ListThreadsOptions): Promise<Thread[]>;
+    /**
+     * Deletes a thread
+     * @param threadId - Thread identifier
+     * @returns A promise that resolves when the thread is deleted
+     */
+    deleteThread(threadId: string): Promise<void>;
+    /**
+     * Retrieves messages from a thread
+     * @param options - Options for retrieving messages
+     * @returns A promise that resolves to an array of messages
+     */
+    getMessages(options: GetMessagesOptions): Promise<Message[]>;
+    /**
+     * Retrieves agent state
+     * @param threadId - Thread identifier
+     * @returns A promise that resolves to the agent state
+     */
+    getAgentState(threadId: string): Promise<AgentState>;
+    /**
+     * Gets agent graph visualization
+     * @returns A promise that resolves to the graph visualization data
+     */
+    getAgentGraph(): Promise<string>;
+    /**
+     * Run agent with options
+     * @param options - Options for running the agent
+     * @returns A promise that resolves to the run result
+     */
+    run(options: RunOptions): Promise<ChatResponse>;
+    /**
+     * Chat namespace for sending messages and streaming responses
+     */
+    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: (options: ChatSendOptions) => Promise<ChatResponse>;
+        /**
+         * 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: ChatStreamOptions, onEvent: (event: MessageChunk) => void, onComplete?: (state?: AgentState) => void, onError?: (error: Error) => void) => (() => void);
+    };
+    /**
+     * Tools namespace for registering and unregistering client-side tools
+     */
+    tools: {
+        /**
+         * Registers a client-side tool
+         * @param options - Options for registering a tool
+         */
+        register: (options: RegisterToolOptions) => void;
+        /**
+         * Unregisters a client-side tool
+         * @param name - Tool name
+         */
+        unregister: (name: string) => void;
+    };
+}
+//# sourceMappingURL=abstract-client.d.ts.map

package/dist/abstract-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"abstract-client.d.ts","sourceRoot":"","sources":["../src/abstract-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EACL,UAAU,EAEV,YAAY,EACZ,eAAe,EACf,iBAAiB,EACjB,YAAY,EACZ,mBAAmB,EACnB,kBAAkB,EAClB,kBAAkB,EAClB,mBAAmB,EACnB,UAAU,EACV,MAAM,EACP,MAAM,SAAS,CAAC;AAEjB;;;GAGG;AACH,8BAAsB,cAAc;IAClC,SAAS,CAAC,MAAM,EAAE,YAAY,CAAC;IAC/B,SAAS,CAAC,WAAW,EAAE,MAAM,CAAC;IAC9B,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAM;IAChC,SAAS,CAAC,eAAe,EAAE,GAAG,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAa;IAExE;;;OAGG;gBACS,MAAM,EAAE,YAAY;IAShC;;;OAGG;IACH,QAAQ,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAE5C;;;OAGG;IACH,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,EAC9B,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;IAEb;;;OAGG;IACH;;;;OAIG;IACH,SAAS,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,GAAG;QACvD,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,GAAG,CAAC;KACX;IAmBD,SAAS,CAAC,QAAQ,CAAC,aAAa,CAC9B,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;IAEb;;;;OAIG;IACG,YAAY,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC;IAejE;;;;OAIG;IACG,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IASlD;;;;OAIG;IACG,WAAW,CAAC,OAAO,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAmBlE;;;;OAIG;IACG,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASnD;;;;OAIG;IACG,WAAW,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IA8BlE;;;;OAIG;IACG,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;IAU1D;;;OAGG;IACG,aAAa,IAAI,OAAO,CAAC,MAAM,CAAC;IAWtC;;;;OAIG;IACG,GAAG,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,YAAY,CAAC;IA6BrD;;OAEG;IACH,IAAI;QACF;;;;WAIG;wBACmB,eAAe,KAAG,OAAO,CAAC,YAAY,CAAC;QAyB7D;;;;;;;WAOG;0BAEQ,iBAAiB,WACjB,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,eACzB,CAAC,KAAK,CAAC,EAAE,UAAU,KAAK,IAAI,YAC/B,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,KAC/B,CAAC,MAAM,IAAI,CAAC;MAuBf;IAEF;;OAEG;IACH,KAAK;QACH;;;WAGG;4BACiB,mBAAmB,KAAG,IAAI;QAS9C;;;WAGG;2BACgB,MAAM,KAAG,IAAI;MAGhC;CACH"}