@assistant-ui/react 0.5.27 → 0.5.28

package/dist/index.mjs

@@ -1,13 +1,15 @@
 "use client";
 import {
+  PipeableTransformStream,
   mergeModelConfigs,
   runResultStream,
+  streamPartEncoderStream,
   toCoreMessage,
   toCoreMessages,
   toLanguageModelMessages,
   toLanguageModelTools,
   toolResultStream
-} from "./chunk-ZWRFAYHH.mjs";
+} from "./chunk-NLH52OUL.mjs";
 import {
   __export
 } from "./chunk-BJPOCE4O.mjs";
@@ -2524,6 +2526,52 @@ var fromLanguageModelTools = (tools) => {
   );
 };
 
+// src/runtimes/edge/streams/utils/chunkByLineStream.ts
+function chunkByLineStream() {
+  let buffer = "";
+  return new TransformStream({
+    transform(chunk, controller) {
+      buffer += chunk;
+      const lines = buffer.split("\n");
+      for (let i = 0; i < lines.length - 1; i++) {
+        controller.enqueue(lines[i]);
+      }
+      buffer = lines[lines.length - 1];
+    },
+    flush(controller) {
+      if (buffer) {
+        controller.enqueue(buffer);
+      }
+    }
+  });
+}
+
+// src/runtimes/edge/streams/utils/streamPartDecoderStream.ts
+var decodeStreamPart = (part) => {
+  const index = part.indexOf(":");
+  if (index === -1) throw new Error("Invalid stream part");
+  return {
+    type: part.slice(0, index),
+    value: JSON.parse(part.slice(index + 1))
+  };
+};
+function streamPartDecoderStream() {
+  const decodeStream = new TransformStream({
+    transform(chunk, controller) {
+      controller.enqueue(decodeStreamPart(chunk));
+    }
+  });
+  return new PipeableTransformStream((readable) => {
+    return readable.pipeThrough(new TextDecoderStream()).pipeThrough(chunkByLineStream()).pipeThrough(decodeStream);
+  });
+}
+
+// src/runtimes/edge/streams/utils/index.ts
+var streamUtils = {
+  streamPartEncoderStream,
+  streamPartDecoderStream
+};
+
 // src/runtimes/edge/useEdgeRuntime.ts
 import { useState as useState8 } from "react";
 
@@ -2532,9 +2580,8 @@ function assistantDecoderStream() {
   const toolCallNames = /* @__PURE__ */ new Map();
   let currentToolCall;
   return new TransformStream({
-    transform(chunk, controller) {
-      const [code, value] = parseStreamPart(chunk);
-      if (currentToolCall && code !== "2" /* ToolCallArgsTextDelta */ && code !== "E" /* Error */) {
+    transform({ type, value }, controller) {
+      if (currentToolCall && type !== "2" /* ToolCallArgsTextDelta */ && type !== "E" /* Error */) {
         controller.enqueue({
           type: "tool-call",
           toolCallType: "function",
@@ -2544,7 +2591,7 @@ function assistantDecoderStream() {
         });
         currentToolCall = void 0;
       }
-      switch (code) {
+      switch (type) {
         case "0" /* TextDelta */: {
           controller.enqueue({
             type: "text-delta",
@@ -2595,41 +2642,13 @@ function assistantDecoderStream() {
           break;
         }
         default: {
-          const unhandledType = code;
+          const unhandledType = type;
           throw new Error(`Unhandled chunk type: ${unhandledType}`);
         }
       }
     }
   });
 }
-var parseStreamPart = (part) => {
-  const index = part.indexOf(":");
-  if (index === -1) throw new Error("Invalid stream part");
-  return [
-    part.slice(0, index),
-    JSON.parse(part.slice(index + 1))
-  ];
-};
-
-// src/runtimes/edge/streams/chunkByLineStream.ts
-function chunkByLineStream() {
-  let buffer = "";
-  return new TransformStream({
-    transform(chunk, controller) {
-      buffer += chunk;
-      const lines = buffer.split("\n");
-      for (let i = 0; i < lines.length - 1; i++) {
-        controller.enqueue(lines[i]);
-      }
-      buffer = lines[lines.length - 1];
-    },
-    flush(controller) {
-      if (buffer) {
-        controller.enqueue(buffer);
-      }
-    }
-  });
-}
 
 // src/runtimes/edge/EdgeChatAdapter.ts
 function asAsyncIterable(source) {
@@ -2667,7 +2686,7 @@ var EdgeChatAdapter = class {
     if (result.status !== 200) {
       throw new Error(`Status ${result.status}: ${await result.text()}`);
     }
-    const stream = result.body.pipeThrough(new TextDecoderStream()).pipeThrough(chunkByLineStream()).pipeThrough(assistantDecoderStream()).pipeThrough(toolResultStream(config.tools)).pipeThrough(runResultStream());
+    const stream = result.body.pipeThrough(streamPartDecoderStream()).pipeThrough(assistantDecoderStream()).pipeThrough(toolResultStream(config.tools)).pipeThrough(runResultStream());
     let update;
     for await (update of asAsyncIterable(stream)) {
       yield update;
@@ -3901,6 +3920,7 @@ export {
   getExternalStoreMessage,
   makeAssistantTool,
   makeAssistantToolUI,
+  streamUtils,
   toCoreMessage,
   toCoreMessages,
   toLanguageModelMessages,