@assistant-ui/react 0.5.27 → 0.5.29

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-DXLSOE2J.mjs";
 import {
   __export
 } from "./chunk-BJPOCE4O.mjs";
@@ -164,13 +166,11 @@ var getThreadStateFromRuntime = (runtime) => {
   if (lastMessage?.role !== "assistant")
     return Object.freeze({
       isDisabled: runtime.isDisabled,
-      isRunning: false,
-      unstable_canAppendNew: !runtime.isDisabled
+      isRunning: false
     });
   return Object.freeze({
     isDisabled: runtime.isDisabled,
-    isRunning: lastMessage.status.type === "running",
-    unstable_canAppendNew: !runtime.isDisabled && lastMessage.status.type !== "running" && lastMessage.status.type !== "requires-action"
+    isRunning: lastMessage.status.type === "running"
   });
 };
 var makeThreadStore = (runtimeRef) => {
@@ -274,7 +274,7 @@ var ThreadProvider = ({
         const onThreadUpdate = () => {
           const oldState = context.useThread.getState();
           const state = getThreadStateFromRuntime(thread);
-          if (oldState.isDisabled !== state.isDisabled || oldState.isRunning !== state.isRunning || oldState.unstable_canAppendNew !== state.unstable_canAppendNew) {
+          if (oldState.isDisabled !== state.isDisabled || oldState.isRunning !== state.isRunning) {
             context.useThread.setState(
               getThreadStateFromRuntime(thread),
               true
@@ -680,7 +680,7 @@ var useComposerSend = () => {
   const { useComposer } = useComposerContext();
   const disabled = useCombinedStore(
     [useThread, useComposer],
-    (t, c) => !t.unstable_canAppendNew || !c.isEditing || c.value.length === 0
+    (t, c) => t.isRunning || !c.isEditing || c.value.length === 0
   );
   const callback = useCallback11(() => {
     const composerState = useComposer.getState();
@@ -1578,8 +1578,8 @@ var ComposerPrimitiveInput = forwardRef12(
     const handleKeyPress = (e) => {
       if (isDisabled) return;
       if (e.key === "Enter" && e.shiftKey === false) {
-        const { unstable_canAppendNew } = useThread.getState();
-        if (unstable_canAppendNew) {
+        const { isRunning } = useThread.getState();
+        if (!isRunning) {
           e.preventDefault();
           textareaRef.current?.closest("form")?.requestSubmit();
         }
@@ -2524,6 +2524,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 +2578,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 +2589,7 @@ function assistantDecoderStream() {
         });
         currentToolCall = void 0;
       }
-      switch (code) {
+      switch (type) {
         case "0" /* TextDelta */: {
           controller.enqueue({
             type: "text-delta",
@@ -2595,41 +2640,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 +2684,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, abortSignal)).pipeThrough(runResultStream());
     let update;
     for await (update of asAsyncIterable(stream)) {
       yield update;
@@ -3901,6 +3918,7 @@ export {
   getExternalStoreMessage,
   makeAssistantTool,
   makeAssistantToolUI,
+  streamUtils,
   toCoreMessage,
   toCoreMessages,
   toLanguageModelMessages,