@axiom-lattice/client-sdk 2.1.30 → 2.1.32

package/dist/index.mjs

@@ -1876,13 +1876,14 @@ var AbstractClient = class {
       send: async (options) => {
         try {
           const message = options.messages[options.messages.length - 1];
-          const { command, threadId, files, assistantId, ...rest } = options;
+          const { command, threadId, files, assistantId, mode, ...rest } = options;
           const result = await this.run({
             threadId,
             message: typeof message.content === "string" ? message.content : JSON.stringify(message.content),
             streaming: false,
             command,
             files,
+            mode,
             ...rest
           });
           return result;
@@ -1906,6 +1907,7 @@ var AbstractClient = class {
           assistantId,
           background,
           enableReturnStateWhenSteamCompleted,
+          mode,
           ...rest
         } = options;
         const message = options.messages[options.messages.length - 1];
@@ -1914,16 +1916,53 @@ var AbstractClient = class {
             threadId,
             assistantId,
             message: typeof message.content === "string" ? message.content : JSON.stringify(message.content),
+            messageId: message.id,
             streaming: true,
             command,
             background,
             enableReturnStateWhenSteamCompleted,
+            mode,
             ...rest
           },
           onEvent,
           onComplete,
           onError
         );
+      },
+      /**
+       * Remove a pending message from the queue
+       * @param options - Options for removing the message
+       * @returns Promise with success status
+       */
+      removePendingMessage: async (options) => {
+        try {
+          const { assistantId, threadId, messageId } = options;
+          return await this.makeRequest(
+            `/api/assistants/${assistantId}/threads/${threadId}/pending-messages/${messageId}`,
+            {
+              method: "DELETE"
+            }
+          );
+        } catch (error) {
+          throw error;
+        }
+      },
+      /**
+       * Abort agent execution for a thread
+       * @param params - Parameters for aborting
+       * @returns Promise with success status
+       */
+      abort: async (params) => {
+        try {
+          return await this.makeRequest(
+            `/api/assistants/${params.assistantId}/threads/${params.threadId}/abort`,
+            {
+              method: "POST"
+            }
+          );
+        } catch (error) {
+          throw error;
+        }
       }
     };
     /**
@@ -2524,7 +2563,7 @@ var AbstractClient = class {
    * @returns The formatted request parameters
    */
   buildStreamRequestParams(options) {
-    const { command, threadId, message, files, background, custom_run_config, ...rest } = options;
+    const { command, threadId, message, messageId, files, background, custom_run_config, ...rest } = options;
     return {
       url: "/api/runs",
       method: "POST",
@@ -2532,6 +2571,7 @@ var AbstractClient = class {
         assistant_id: this.assistantId,
         thread_id: threadId,
         message,
+        message_id: messageId,
         files,
         command,
         streaming: true,
@@ -2837,14 +2877,15 @@ var _Client = class extends AbstractClient {
       ...workspaceHeaders,
       ...options?.headers || {}
     };
+    if (method === "DELETE" && !options?.body) {
+      delete requestHeaders["Content-Type"];
+    }
     const requestOptions = {
       method,
       headers: requestHeaders
     };
     if (options?.body) {
       requestOptions.body = JSON.stringify(options.body);
-    } else if (method !== "GET" && method !== "HEAD") {
-      requestOptions.body = JSON.stringify({});
     }
     return this.fetchWithTimeout(url, requestOptions);
   }
@@ -3620,6 +3661,7 @@ function createSimpleMessageMerger() {
       updateToolCalls(message);
     }
     if (message.role === "tool" && chunk.data.tool_call_id) {
+      const original_tool_message_id = chunk.data.id;
       const messageId = toolId_MessageIdMap.get(chunk.data.tool_call_id);
       if (!messageId)
         return;
@@ -3628,7 +3670,7 @@ function createSimpleMessageMerger() {
         const message2 = messages[messageIndex];
         message2.tool_calls = message2.tool_calls?.map((tc) => {
           if (tc.id === chunk.data.tool_call_id) {
-            return { ...tc, response: chunk.data.content, status: "success" };
+            return { ...tc, response: chunk.data.content, status: "success", original_tool_message_id };
           }
           return tc;
         });