@thorprovider/medusa-extended 1.11.2 → 1.11.3

package/dist/index.mjs

@@ -449,6 +449,70 @@ var DropshipperClient = class {
       body
     );
   }
+  /**
+   * Stream the AI assistant's reply.
+   *
+   * Unlike `chatAssistant`, this surfaces the model's progress: the backend
+   * resolves the intent first and then streams the answer token by token
+   * (catalog searches) or returns the full answer in one shot (greetings/other).
+   * The response is normalized into a single `ReadableStream<ChatStreamEvent>`
+   * so callers can show a live "thinking" label and render text incrementally.
+   *
+   * `POST /admin/thor/dropshipper/ai/chat`
+   */
+  async chatAssistantStream(body) {
+    const url = `${this.baseUrl}/admin/thor/dropshipper/ai/chat`;
+    this.logger.debug("[dropshipper] POST /admin/thor/dropshipper/ai/chat (stream)");
+    const init = {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${this.token}`,
+        "Content-Type": "application/json",
+        Accept: "text/event-stream"
+      },
+      body: JSON.stringify(body)
+    };
+    let response;
+    try {
+      response = await fetchBackendWithResilience(url, init);
+    } catch (networkError) {
+      throw new ProviderAPIError3(
+        `[dropshipper] Network error on POST /admin/thor/dropshipper/ai/chat: ${networkError.message}`,
+        "NETWORK_ERROR",
+        networkError
+      );
+    }
+    if (!response.ok) {
+      let message = `HTTP ${response.status}`;
+      try {
+        const json = await response.json();
+        message = json?.message ?? message;
+      } catch {
+      }
+      throw new ProviderAPIError3(
+        `[dropshipper] POST /admin/thor/dropshipper/ai/chat failed: ${message}`,
+        `DROPSHIPPER_API_${response.status}`,
+        response.status
+      );
+    }
+    const contentType = response.headers.get("content-type") ?? "";
+    if (contentType.includes("text/event-stream") && response.body) {
+      return mapUiMessageStreamToEvents(response.body);
+    }
+    const data = await response.json();
+    return new ReadableStream({
+      start(controller) {
+        controller.enqueue({ type: "start", intent: data.intent });
+        controller.enqueue({ type: "text-delta", delta: data.answer });
+        controller.enqueue({
+          type: "finish",
+          intent: data.intent,
+          sources: data.sources ?? []
+        });
+        controller.close();
+      }
+    });
+  }
   // -------------------------------------------------------------------------
   // Notifications
   // -------------------------------------------------------------------------
@@ -1607,6 +1671,81 @@ var DropshipperClient = class {
     );
   }
 };
+async function* parseSse(stream) {
+  const reader = stream.getReader();
+  const decoder = new TextDecoder();
+  let buffer = "";
+  try {
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) break;
+      buffer += decoder.decode(value, { stream: true });
+      const lines = buffer.split(/\r?\n/);
+      buffer = lines.pop() ?? "";
+      for (const line of lines) {
+        const trimmed = line.trim();
+        if (!trimmed.startsWith("data:")) continue;
+        const payload = trimmed.slice(5).trim();
+        if (!payload || payload === "[DONE]") continue;
+        try {
+          yield JSON.parse(payload);
+        } catch {
+        }
+      }
+    }
+    const tail = buffer.trim();
+    if (tail.startsWith("data:")) {
+      const payload = tail.slice(5).trim();
+      if (payload && payload !== "[DONE]") {
+        try {
+          yield JSON.parse(payload);
+        } catch {
+        }
+      }
+    }
+  } finally {
+    reader.releaseLock();
+  }
+}
+function mapUiMessageStreamToEvents(stream) {
+  return new ReadableStream({
+    async start(controller) {
+      for await (const raw of parseSse(stream)) {
+        const chunk = raw;
+        switch (chunk.type) {
+          case "start":
+            controller.enqueue({
+              type: "start",
+              intent: chunk.messageMetadata?.intent
+            });
+            break;
+          case "text-delta":
+            if (chunk.delta !== void 0) {
+              controller.enqueue({ type: "text-delta", delta: chunk.delta });
+            }
+            break;
+          case "finish":
+            controller.enqueue({
+              type: "finish",
+              intent: chunk.messageMetadata?.intent ?? "other",
+              sources: chunk.messageMetadata?.sources ?? []
+            });
+            break;
+          case "error":
+            controller.enqueue({
+              type: "error",
+              errorText: chunk.errorText ?? "Error inesperado del asistente."
+            });
+            break;
+        }
+      }
+      controller.close();
+    },
+    cancel() {
+      stream.cancel?.();
+    }
+  });
+}
 export {
   DropshipperClient,
   MedusaAdminClient