@axiom-lattice/client-sdk 1.0.46 → 2.0.0

package/dist/index.mjs

@@ -2220,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
@@ -2370,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