@assistant-ui/react 0.0.25 → 0.0.27

package/dist/index.mjs

@@ -1016,16 +1016,16 @@ __export(contentPart_exports, {
   InProgressIndicator: () => ContentPartInProgressIndicator
 });
 
-// src/runtime/vercel-ai/rsc/useVercelRSCRuntime.tsx
+// ../react-ai-sdk/src/rsc/useVercelRSCRuntime.tsx
 import { useEffect as useEffect7, useInsertionEffect, useState as useState3 } from "react";
 
-// src/runtime/vercel-ai/rsc/VercelRSCRuntime.tsx
+// ../react-ai-sdk/src/rsc/VercelRSCRuntime.tsx
 import { create as create4 } from "zustand";
 
-// src/runtime/vercel-ai/rsc/useVercelRSCSync.tsx
+// ../react-ai-sdk/src/rsc/useVercelRSCSync.tsx
 import { useEffect as useEffect6, useMemo as useMemo4 } from "react";
 
-// src/runtime/vercel-ai/utils/ThreadMessageConverter.ts
+// ../react-ai-sdk/src/utils/ThreadMessageConverter.ts
 var ThreadMessageConverter = class {
   cache = /* @__PURE__ */ new WeakMap();
   convertMessages(messages, converter, keyMapper = (key) => key) {
@@ -1039,13 +1039,13 @@ var ThreadMessageConverter = class {
   }
 };
 
-// src/runtime/vercel-ai/rsc/getVercelRSCMessage.tsx
+// ../react-ai-sdk/src/rsc/getVercelRSCMessage.tsx
 var symbolInnerRSCMessage = Symbol("innerVercelRSCMessage");
 var getVercelRSCMessage = (message) => {
   return message[symbolInnerRSCMessage];
 };
 
-// src/runtime/vercel-ai/rsc/useVercelRSCSync.tsx
+// ../react-ai-sdk/src/rsc/useVercelRSCSync.tsx
 var vercelToThreadMessage = (converter, rawMessage) => {
   const message = converter(rawMessage);
   return {
@@ -1071,7 +1071,7 @@ var useVercelRSCSync = (adapter, updateData) => {
   }, [updateData, converter, convertCallback, adapter.messages]);
 };
 
-// src/runtime/vercel-ai/rsc/VercelRSCRuntime.tsx
+// ../react-ai-sdk/src/rsc/VercelRSCRuntime.tsx
 var EMPTY_BRANCHES = Object.freeze([]);
 var VercelRSCRuntime = class {
   constructor(adapter) {
@@ -1143,7 +1143,7 @@ var VercelRSCRuntime = class {
   };
 };
 
-// src/runtime/vercel-ai/rsc/useVercelRSCRuntime.tsx
+// ../react-ai-sdk/src/rsc/useVercelRSCRuntime.tsx
 var useVercelRSCRuntime = (adapter) => {
   const [runtime] = useState3(() => new VercelRSCRuntime(adapter));
   useInsertionEffect(() => {
@@ -1155,179 +1155,19 @@ var useVercelRSCRuntime = (adapter) => {
   return runtime;
 };
 
-// src/runtime/vercel-ai/ui/use-chat/useVercelUseChatRuntime.tsx
+// ../react-ai-sdk/src/ui/use-chat/useVercelUseChatRuntime.tsx
 import { useEffect as useEffect10, useInsertionEffect as useInsertionEffect2, useState as useState4 } from "react";
 
-// src/runtime/vercel-ai/ui/VercelAIRuntime.tsx
+// ../react-ai-sdk/src/ui/VercelAIRuntime.tsx
 import { create as create5 } from "zustand";
 
-// src/runtime/utils/idUtils.tsx
-import { customAlphabet } from "nanoid/non-secure";
-var generateId = customAlphabet(
-  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
-  7
-);
-var optimisticPrefix = "__optimistic__";
-var generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;
-
-// src/runtime/utils/MessageRepository.tsx
-var findHead = (message) => {
-  if (message.next) return findHead(message.next);
-  return message;
-};
-var MessageRepository = class {
-  messages = /* @__PURE__ */ new Map();
-  // message_id -> item
-  head = null;
-  root = {
-    children: []
-  };
-  performOp(newParent, child, operation) {
-    const parentOrRoot = child.prev ?? this.root;
-    const newParentOrRoot = newParent ?? this.root;
-    if (operation === "relink" && parentOrRoot === newParentOrRoot) return;
-    if (operation !== "link") {
-      parentOrRoot.children = parentOrRoot.children.filter(
-        (m) => m !== child.current.id
-      );
-      if (child.prev?.next === child) {
-        const fallbackId = child.prev.children.at(-1);
-        const fallback = fallbackId ? this.messages.get(fallbackId) : null;
-        if (fallback === void 0) {
-          throw new Error(
-            "MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui."
-          );
-        }
-        child.prev.next = fallback;
-      }
-    }
-    if (operation !== "cut") {
-      newParentOrRoot.children = [
-        ...newParentOrRoot.children,
-        child.current.id
-      ];
-      if (newParent && (findHead(child) === this.head || newParent.next === null)) {
-        newParent.next = child;
-      }
-      child.prev = newParent;
-    }
-  }
-  getMessages() {
-    const messages = new Array(this.head?.level ?? 0);
-    for (let current = this.head; current; current = current.prev) {
-      messages[current.level] = current.current;
-    }
-    return messages;
-  }
-  addOrUpdateMessage(parentId, message) {
-    const existingItem = this.messages.get(message.id);
-    const prev = parentId ? this.messages.get(parentId) : null;
-    if (prev === void 0)
-      throw new Error(
-        "MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui."
-      );
-    if (existingItem) {
-      existingItem.current = message;
-      this.performOp(prev, existingItem, "relink");
-      return;
-    }
-    const newItem = {
-      prev,
-      current: message,
-      next: null,
-      children: [],
-      level: prev ? prev.level + 1 : 0
-    };
-    this.messages.set(message.id, newItem);
-    this.performOp(prev, newItem, "link");
-    if (this.head === prev) {
-      this.head = newItem;
-    }
-  }
-  appendOptimisticMessage(parentId, message) {
-    let optimisticId;
-    do {
-      optimisticId = generateOptimisticId();
-    } while (this.messages.has(optimisticId));
-    this.addOrUpdateMessage(parentId, {
-      ...message,
-      id: optimisticId,
-      createdAt: /* @__PURE__ */ new Date(),
-      ...message.role === "assistant" ? { status: "in_progress" } : void 0
-    });
-    return optimisticId;
-  }
-  deleteMessage(messageId, replacementId) {
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(deleteMessage): Optimistic message not found. This is likely an internal bug in assistant-ui."
-      );
-    const replacement = replacementId === void 0 ? message.prev : replacementId === null ? null : this.messages.get(replacementId);
-    if (replacement === void 0)
-      throw new Error(
-        "MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui."
-      );
-    for (const child of message.children) {
-      const childMessage = this.messages.get(child);
-      if (!childMessage)
-        throw new Error(
-          "MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui."
-        );
-      this.performOp(replacement, childMessage, "relink");
-    }
-    this.performOp(null, message, "cut");
-    this.messages.delete(messageId);
-    if (this.head === message) {
-      this.head = replacement ? findHead(replacement) : null;
-    }
-  }
-  getBranches(messageId) {
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui."
-      );
-    const { children } = message.prev ?? this.root;
-    return children;
-  }
-  switchToBranch(messageId) {
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui."
-      );
-    if (message.prev) {
-      message.prev.next = message;
-    }
-    this.head = findHead(message);
-  }
-  resetHead(messageId) {
-    if (messageId === null) {
-      this.head = null;
-      return;
-    }
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui."
-      );
-    this.head = message;
-    for (let current = message; current; current = current.prev) {
-      if (current.prev) {
-        current.prev.next = current;
-      }
-    }
-  }
-};
-
-// src/runtime/vercel-ai/ui/getVercelAIMessage.tsx
+// ../react-ai-sdk/src/ui/getVercelAIMessage.tsx
 var symbolInnerAIMessage = Symbol("innerVercelAIUIMessage");
 var getVercelAIMessage = (message) => {
   return message[symbolInnerAIMessage];
 };
 
-// src/runtime/vercel-ai/ui/utils/sliceMessagesUntil.tsx
+// ../react-ai-sdk/src/ui/utils/sliceMessagesUntil.tsx
 var sliceMessagesUntil = (messages, messageId) => {
   if (messageId == null) return [];
   let messageIdx = messages.findIndex((m) => m.id === messageId);
@@ -1341,7 +1181,7 @@ var sliceMessagesUntil = (messages, messageId) => {
   return messages.slice(0, messageIdx + 1);
 };
 
-// src/runtime/vercel-ai/ui/utils/useVercelAIComposerSync.tsx
+// ../react-ai-sdk/src/ui/utils/useVercelAIComposerSync.tsx
 import { useEffect as useEffect8 } from "react";
 var useVercelAIComposerSync = (vercel) => {
   const { useComposer } = useThreadContext();
@@ -1353,7 +1193,7 @@ var useVercelAIComposerSync = (vercel) => {
   }, [useComposer, vercel.input, vercel.setInput]);
 };
 
-// src/runtime/vercel-ai/ui/utils/useVercelAIThreadSync.tsx
+// ../react-ai-sdk/src/ui/utils/useVercelAIThreadSync.tsx
 import { useEffect as useEffect9, useMemo as useMemo5 } from "react";
 var getIsRunning = (vercel) => {
   if ("isLoading" in vercel) return vercel.isLoading;
@@ -1449,7 +1289,7 @@ var useVercelAIThreadSync = (vercel, updateData) => {
   }, [updateData, isRunning, vercel.messages, converter]);
 };
 
-// src/runtime/vercel-ai/ui/VercelAIRuntime.tsx
+// ../react-ai-sdk/src/ui/VercelAIRuntime.tsx
 var hasUpcomingMessage = (isRunning, messages) => {
   return isRunning && messages[messages.length - 1]?.role !== "assistant";
 };
@@ -1561,7 +1401,7 @@ var VercelAIRuntime = class {
   };
 };
 
-// src/runtime/vercel-ai/ui/use-chat/useVercelUseChatRuntime.tsx
+// ../react-ai-sdk/src/ui/use-chat/useVercelUseChatRuntime.tsx
 var useVercelUseChatRuntime = (chatHelpers) => {
   const [runtime] = useState4(() => new VercelAIRuntime(chatHelpers));
   useInsertionEffect2(() => {
@@ -1573,7 +1413,7 @@ var useVercelUseChatRuntime = (chatHelpers) => {
   return runtime;
 };
 
-// src/runtime/vercel-ai/ui/use-assistant/useVercelUseAssistantRuntime.tsx
+// ../react-ai-sdk/src/ui/use-assistant/useVercelUseAssistantRuntime.tsx
 import { useEffect as useEffect11, useInsertionEffect as useInsertionEffect3, useState as useState5 } from "react";
 var useVercelUseAssistantRuntime = (assistantHelpers) => {
   const [runtime] = useState5(() => new VercelAIRuntime(assistantHelpers));
@@ -1586,6 +1426,166 @@ var useVercelUseAssistantRuntime = (assistantHelpers) => {
   return runtime;
 };
 
+// src/runtime/utils/idUtils.tsx
+import { customAlphabet } from "nanoid/non-secure";
+var generateId = customAlphabet(
+  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
+  7
+);
+var optimisticPrefix = "__optimistic__";
+var generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;
+
+// src/runtime/utils/MessageRepository.tsx
+var findHead = (message) => {
+  if (message.next) return findHead(message.next);
+  return message;
+};
+var MessageRepository = class {
+  messages = /* @__PURE__ */ new Map();
+  // message_id -> item
+  head = null;
+  root = {
+    children: []
+  };
+  performOp(newParent, child, operation) {
+    const parentOrRoot = child.prev ?? this.root;
+    const newParentOrRoot = newParent ?? this.root;
+    if (operation === "relink" && parentOrRoot === newParentOrRoot) return;
+    if (operation !== "link") {
+      parentOrRoot.children = parentOrRoot.children.filter(
+        (m) => m !== child.current.id
+      );
+      if (child.prev?.next === child) {
+        const fallbackId = child.prev.children.at(-1);
+        const fallback = fallbackId ? this.messages.get(fallbackId) : null;
+        if (fallback === void 0) {
+          throw new Error(
+            "MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui."
+          );
+        }
+        child.prev.next = fallback;
+      }
+    }
+    if (operation !== "cut") {
+      newParentOrRoot.children = [
+        ...newParentOrRoot.children,
+        child.current.id
+      ];
+      if (newParent && (findHead(child) === this.head || newParent.next === null)) {
+        newParent.next = child;
+      }
+      child.prev = newParent;
+    }
+  }
+  getMessages() {
+    const messages = new Array(this.head?.level ?? 0);
+    for (let current = this.head; current; current = current.prev) {
+      messages[current.level] = current.current;
+    }
+    return messages;
+  }
+  addOrUpdateMessage(parentId, message) {
+    const existingItem = this.messages.get(message.id);
+    const prev = parentId ? this.messages.get(parentId) : null;
+    if (prev === void 0)
+      throw new Error(
+        "MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui."
+      );
+    if (existingItem) {
+      existingItem.current = message;
+      this.performOp(prev, existingItem, "relink");
+      return;
+    }
+    const newItem = {
+      prev,
+      current: message,
+      next: null,
+      children: [],
+      level: prev ? prev.level + 1 : 0
+    };
+    this.messages.set(message.id, newItem);
+    this.performOp(prev, newItem, "link");
+    if (this.head === prev) {
+      this.head = newItem;
+    }
+  }
+  appendOptimisticMessage(parentId, message) {
+    let optimisticId;
+    do {
+      optimisticId = generateOptimisticId();
+    } while (this.messages.has(optimisticId));
+    this.addOrUpdateMessage(parentId, {
+      ...message,
+      id: optimisticId,
+      createdAt: /* @__PURE__ */ new Date(),
+      ...message.role === "assistant" ? { status: "in_progress" } : void 0
+    });
+    return optimisticId;
+  }
+  deleteMessage(messageId, replacementId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(deleteMessage): Optimistic message not found. This is likely an internal bug in assistant-ui."
+      );
+    const replacement = replacementId === void 0 ? message.prev : replacementId === null ? null : this.messages.get(replacementId);
+    if (replacement === void 0)
+      throw new Error(
+        "MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui."
+      );
+    for (const child of message.children) {
+      const childMessage = this.messages.get(child);
+      if (!childMessage)
+        throw new Error(
+          "MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui."
+        );
+      this.performOp(replacement, childMessage, "relink");
+    }
+    this.performOp(null, message, "cut");
+    this.messages.delete(messageId);
+    if (this.head === message) {
+      this.head = replacement ? findHead(replacement) : null;
+    }
+  }
+  getBranches(messageId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui."
+      );
+    const { children } = message.prev ?? this.root;
+    return children;
+  }
+  switchToBranch(messageId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui."
+      );
+    if (message.prev) {
+      message.prev.next = message;
+    }
+    this.head = findHead(message);
+  }
+  resetHead(messageId) {
+    if (messageId === null) {
+      this.head = null;
+      return;
+    }
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui."
+      );
+    this.head = message;
+    for (let current = message; current; current = current.prev) {
+      if (current.prev) {
+        current.prev.next = current;
+      }
+    }
+  }
+};
+
 // src/context/providers/AssistantRuntimeProvider.tsx
 import { memo } from "react";
 
@@ -1787,15 +1787,17 @@ var LocalRuntime = class {
     this.abortController = new AbortController();
     this.notifySubscribers();
     try {
-      await this.adapter.run({
+      const updateHandler = ({ content }) => {
+        message.content = content;
+        this.repository.addOrUpdateMessage(parentId, { ...message });
+        this.notifySubscribers();
+      };
+      const result = await this.adapter.run({
         messages,
         abortSignal: this.abortController.signal,
-        onUpdate: ({ content }) => {
-          message.content = content;
-          this.repository.addOrUpdateMessage(parentId, { ...message });
-          this.notifySubscribers();
-        }
+        onUpdate: updateHandler
       });
+      updateHandler(result);
       message.status = "done";
       this.repository.addOrUpdateMessage(parentId, { ...message });
     } catch (e) {
@@ -1830,48 +1832,6 @@ var useLocalRuntime = (adapter) => {
   });
   return runtime;
 };
-
-// src/runtime/local/vercel/VercelModelAdapter.tsx
-import { streamText } from "ai";
-var VercelModelAdapter = class {
-  constructor(model) {
-    this.model = model;
-  }
-  async run({ messages, abortSignal, onUpdate }) {
-    const { fullStream } = await streamText({
-      model: this.model,
-      abortSignal,
-      messages: messages.map((m) => ({
-        role: m.role,
-        content: m.content.filter((c) => c.type !== "ui")
-      }))
-    });
-    const content = [];
-    for await (const aiPart of fullStream) {
-      switch (aiPart.type) {
-        case "text-delta": {
-          let part = content.at(-1);
-          if (!part || part.type !== "text") {
-            part = { type: "text", text: "" };
-            content.push(part);
-          }
-          part.text += aiPart.textDelta;
-          break;
-        }
-        case "tool-call": {
-          content.push({
-            type: "tool-call",
-            name: aiPart.toolName,
-            args: aiPart.args
-          });
-          break;
-        }
-      }
-      onUpdate({ content });
-    }
-    return { content };
-  }
-};
 export {
   actionBar_exports as ActionBarPrimitive,
   AssistantRuntimeProvider,
@@ -1884,7 +1844,7 @@ export {
   VercelRSCAssistantProvider,
   getVercelAIMessage,
   getVercelRSCMessage,
-  VercelModelAdapter as unstable_VercelModelAdapter,
+  MessageRepository as unstable_MessageRepository,
   useComposerContext as unstable_useComposerContext,
   useContentPartContext as unstable_useContentPartContext,
   useLocalRuntime as unstable_useLocalRuntime,