@axiom-lattice/client-sdk 1.0.45 → 1.0.50

package/dist/index.mjs

@@ -1885,15 +1885,24 @@ var AbstractClient = class {
        * @returns A function that can be called to stop the stream
        */
       stream: (options, onEvent, onComplete, onError) => {
+        const {
+          command,
+          threadId,
+          files,
+          background,
+          enableReturnStateWhenSteamCompleted,
+          ...rest
+        } = options;
         const message = options.messages[options.messages.length - 1];
         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
+            command,
+            background,
+            enableReturnStateWhenSteamCompleted,
+            ...rest
           },
           onEvent,
           onComplete,
@@ -2211,6 +2220,98 @@ var Client = class extends AbstractClient {
   streamRequest(options, onEvent, onComplete, onError) {
     return this.streamRun(options, onEvent, onComplete, onError);
   }
+  /**
+   * Resume streaming from a known position
+   * @param options - Options for resuming the stream
+   * @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
+   */
+  resumeStream(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;
+    }
+    const controller = new AbortController();
+    const { signal } = controller;
+    (async () => {
+      try {
+        const response = await fetch(
+          `${this.config.baseURL}/api/resume_stream`,
+          {
+            method: "POST",
+            headers,
+            body: JSON.stringify({
+              thread_id: options.threadId,
+              message_id: options.messageId,
+              known_content: options.knownContent,
+              poll_interval: options.pollInterval || 100
+            }),
+            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;
+          const chunk = decoder.decode(value, { stream: true });
+          buffer += chunk;
+          const lines = buffer.split("\n");
+          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))
+                  );
+                }
+              }
+            }
+          }
+        }
+        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) {
+          onComplete();
+        }
+      } catch (error) {
+        if (error instanceof DOMException && error.name === "AbortError") {
+          return;
+        }
+        if (onError) {
+          onError(error instanceof Error ? error : new Error(String(error)));
+        }
+      }
+    })();
+    return () => {
+      controller.abort();
+    };
+  }
   /**
    * Stream run results
    * @param options - Options for streaming run results
@@ -2361,6 +2462,72 @@ var WeChatClient = class extends AbstractClient {
   streamRequest(options, onEvent, onComplete, onError) {
     return this.streamRun(options, onEvent, onComplete, onError);
   }
+  /**
+   * Resume streaming from a known position
+   * @param options - Options for resuming the stream
+   * @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
+   */
+  resumeStream(options, onEvent, onComplete, onError) {
+    const headers = {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${this.config.apiKey}`,
+      ...this.config.headers
+    };
+    if (this.tenantId) {
+      headers["x-tenant-id"] = this.tenantId;
+    }
+    const requestTask = wx.request({
+      url: `${this.config.baseURL}/api/resume_stream`,
+      method: "POST",
+      data: {
+        thread_id: options.threadId,
+        message_id: options.messageId,
+        known_content: options.knownContent,
+        poll_interval: options.pollInterval || 100
+      },
+      header: headers,
+      responseType: "text",
+      enableChunked: true,
+      success: () => {
+      },
+      fail: (err) => {
+        if (onError) {
+          onError(new Error(`Resume stream request failed: ${err.errMsg}`));
+        }
+      },
+      complete: () => {
+        if (onComplete) {
+          onComplete();
+        }
+      }
+    });
+    requestTask.onChunkReceived((res) => {
+      if (!res.data)
+        return;
+      const text = this.decodeUint8Array(res.data);
+      const lines = text.split("\n");
+      for (const line of lines) {
+        if (line.trim().startsWith("data: ")) {
+          try {
+            const eventData = JSON.parse(line.trim().slice(6));
+            onEvent(eventData);
+          } catch (error) {
+            if (onError) {
+              onError(
+                error instanceof Error ? error : new Error(String(error))
+              );
+            }
+          }
+        }
+      }
+    });
+    return () => {
+      requestTask.abort();
+    };
+  }
   /**
    * Helper method to make WeChat HTTP requests
    * @private
@@ -2694,7 +2861,7 @@ function createSimpleMessageMerger() {
     );
   }
   function reset() {
-    messages.length = 0;
+    messages = [];
     messageMap.clear();
     toolBuilders.clear();
   }