@assistant-ui/react 0.0.25 → 0.0.27

package/dist/index.js

@@ -41,7 +41,7 @@ __export(src_exports, {
   VercelRSCAssistantProvider: () => VercelRSCAssistantProvider,
   getVercelAIMessage: () => getVercelAIMessage,
   getVercelRSCMessage: () => getVercelRSCMessage,
-  unstable_VercelModelAdapter: () => VercelModelAdapter,
+  unstable_MessageRepository: () => MessageRepository,
   unstable_useComposerContext: () => useComposerContext,
   unstable_useContentPartContext: () => useContentPartContext,
   unstable_useLocalRuntime: () => useLocalRuntime,
@@ -1041,16 +1041,16 @@ __export(contentPart_exports, {
   InProgressIndicator: () => ContentPartInProgressIndicator
 });
 
-// src/runtime/vercel-ai/rsc/useVercelRSCRuntime.tsx
+// ../react-ai-sdk/src/rsc/useVercelRSCRuntime.tsx
 var import_react30 = require("react");
 
-// src/runtime/vercel-ai/rsc/VercelRSCRuntime.tsx
+// ../react-ai-sdk/src/rsc/VercelRSCRuntime.tsx
 var import_zustand4 = require("zustand");
 
-// src/runtime/vercel-ai/rsc/useVercelRSCSync.tsx
+// ../react-ai-sdk/src/rsc/useVercelRSCSync.tsx
 var import_react29 = require("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) {
@@ -1064,13 +1064,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 {
@@ -1096,7 +1096,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) {
@@ -1168,7 +1168,7 @@ var VercelRSCRuntime = class {
   };
 };
 
-// src/runtime/vercel-ai/rsc/useVercelRSCRuntime.tsx
+// ../react-ai-sdk/src/rsc/useVercelRSCRuntime.tsx
 var useVercelRSCRuntime = (adapter) => {
   const [runtime] = (0, import_react30.useState)(() => new VercelRSCRuntime(adapter));
   (0, import_react30.useInsertionEffect)(() => {
@@ -1180,179 +1180,19 @@ var useVercelRSCRuntime = (adapter) => {
   return runtime;
 };
 
-// src/runtime/vercel-ai/ui/use-chat/useVercelUseChatRuntime.tsx
-var import_react33 = require("react");
+// ../react-ai-sdk/src/ui/use-chat/useVercelUseChatRuntime.tsx
+var import_react35 = require("react");
 
-// src/runtime/vercel-ai/ui/VercelAIRuntime.tsx
+// ../react-ai-sdk/src/ui/VercelAIRuntime.tsx
 var import_zustand5 = require("zustand");
 
-// src/runtime/utils/idUtils.tsx
-var import_non_secure = require("nanoid/non-secure");
-var generateId = (0, import_non_secure.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);
@@ -1366,11 +1206,11 @@ var sliceMessagesUntil = (messages, messageId) => {
   return messages.slice(0, messageIdx + 1);
 };
 
-// src/runtime/vercel-ai/ui/utils/useVercelAIComposerSync.tsx
-var import_react31 = require("react");
+// ../react-ai-sdk/src/ui/utils/useVercelAIComposerSync.tsx
+var import_react32 = require("react");
 var useVercelAIComposerSync = (vercel) => {
   const { useComposer } = useThreadContext();
-  (0, import_react31.useEffect)(() => {
+  (0, import_react32.useEffect)(() => {
     useComposer.setState({
       value: vercel.input,
       setValue: vercel.setInput
@@ -1378,8 +1218,8 @@ var useVercelAIComposerSync = (vercel) => {
   }, [useComposer, vercel.input, vercel.setInput]);
 };
 
-// src/runtime/vercel-ai/ui/utils/useVercelAIThreadSync.tsx
-var import_react32 = require("react");
+// ../react-ai-sdk/src/ui/utils/useVercelAIThreadSync.tsx
+var import_react33 = require("react");
 var getIsRunning = (vercel) => {
   if ("isLoading" in vercel) return vercel.isLoading;
   return vercel.status === "in_progress";
@@ -1456,8 +1296,8 @@ var shallowArrayEqual = (a, b) => {
 };
 var useVercelAIThreadSync = (vercel, updateData) => {
   const isRunning = getIsRunning(vercel);
-  const converter = (0, import_react32.useMemo)(() => new ThreadMessageConverter(), []);
-  (0, import_react32.useEffect)(() => {
+  const converter = (0, import_react33.useMemo)(() => new ThreadMessageConverter(), []);
+  (0, import_react33.useEffect)(() => {
     const lastMessageId = vercel.messages.at(-1)?.id;
     const convertCallback = (messages2, cache) => {
       const status = lastMessageId === messages2[0].id && isRunning ? "in_progress" : "done";
@@ -1474,7 +1314,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";
 };
@@ -1586,36 +1426,196 @@ 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] = (0, import_react33.useState)(() => new VercelAIRuntime(chatHelpers));
-  (0, import_react33.useInsertionEffect)(() => {
+  const [runtime] = (0, import_react35.useState)(() => new VercelAIRuntime(chatHelpers));
+  (0, import_react35.useInsertionEffect)(() => {
     runtime.vercel = chatHelpers;
   });
-  (0, import_react33.useEffect)(() => {
+  (0, import_react35.useEffect)(() => {
     runtime.onVercelUpdated();
   });
   return runtime;
 };
 
-// src/runtime/vercel-ai/ui/use-assistant/useVercelUseAssistantRuntime.tsx
-var import_react34 = require("react");
+// ../react-ai-sdk/src/ui/use-assistant/useVercelUseAssistantRuntime.tsx
+var import_react36 = require("react");
 var useVercelUseAssistantRuntime = (assistantHelpers) => {
-  const [runtime] = (0, import_react34.useState)(() => new VercelAIRuntime(assistantHelpers));
-  (0, import_react34.useInsertionEffect)(() => {
+  const [runtime] = (0, import_react36.useState)(() => new VercelAIRuntime(assistantHelpers));
+  (0, import_react36.useInsertionEffect)(() => {
     runtime.vercel = assistantHelpers;
   });
-  (0, import_react34.useEffect)(() => {
+  (0, import_react36.useEffect)(() => {
     runtime.onVercelUpdated();
   });
   return runtime;
 };
 
+// src/runtime/utils/idUtils.tsx
+var import_non_secure = require("nanoid/non-secure");
+var generateId = (0, import_non_secure.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
-var import_react36 = require("react");
+var import_react38 = require("react");
 
 // src/context/providers/ThreadProvider.tsx
-var import_react35 = require("react");
+var import_react37 = require("react");
 
 // src/context/stores/Composer.ts
 var import_zustand6 = require("zustand");
@@ -1690,11 +1690,11 @@ var ThreadProvider = ({
   children,
   runtime
 }) => {
-  const runtimeRef = (0, import_react35.useRef)(runtime);
-  (0, import_react35.useInsertionEffect)(() => {
+  const runtimeRef = (0, import_react37.useRef)(runtime);
+  (0, import_react37.useInsertionEffect)(() => {
     runtimeRef.current = runtime;
   });
-  const [{ context, onRuntimeUpdate }] = (0, import_react35.useState)(() => {
+  const [{ context, onRuntimeUpdate }] = (0, import_react37.useState)(() => {
     const { useThread, onRuntimeUpdate: onRuntimeUpdate2 } = makeThreadStore(runtimeRef);
     const useViewport = makeThreadViewportStore();
     const useComposer = makeComposerStore(useThread);
@@ -1707,7 +1707,7 @@ var ThreadProvider = ({
       onRuntimeUpdate: onRuntimeUpdate2
     };
   });
-  (0, import_react35.useEffect)(() => {
+  (0, import_react37.useEffect)(() => {
     onRuntimeUpdate();
     return runtime.subscribe(onRuntimeUpdate);
   }, [onRuntimeUpdate, runtime]);
@@ -1723,7 +1723,7 @@ var import_jsx_runtime22 = require("react/jsx-runtime");
 var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
   return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(ThreadProvider, { runtime, children });
 };
-var AssistantRuntimeProvider = (0, import_react36.memo)(AssistantRuntimeProviderImpl);
+var AssistantRuntimeProvider = (0, import_react38.memo)(AssistantRuntimeProviderImpl);
 
 // src/runtime/vercel-deprecated/VercelAIAssistantProvider.tsx
 var import_jsx_runtime23 = require("react/jsx-runtime");
@@ -1762,7 +1762,7 @@ var VercelRSCAssistantProvider = ({
 };
 
 // src/runtime/local/useLocalRuntime.tsx
-var import_react37 = require("react");
+var import_react39 = require("react");
 
 // src/runtime/local/LocalRuntime.tsx
 var LocalRuntime = class {
@@ -1812,15 +1812,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) {
@@ -1849,54 +1851,12 @@ var LocalRuntime = class {
 
 // src/runtime/local/useLocalRuntime.tsx
 var useLocalRuntime = (adapter) => {
-  const [runtime] = (0, import_react37.useState)(() => new LocalRuntime(adapter));
-  (0, import_react37.useInsertionEffect)(() => {
+  const [runtime] = (0, import_react39.useState)(() => new LocalRuntime(adapter));
+  (0, import_react39.useInsertionEffect)(() => {
     runtime.adapter = adapter;
   });
   return runtime;
 };
-
-// src/runtime/local/vercel/VercelModelAdapter.tsx
-var import_ai = require("ai");
-var VercelModelAdapter = class {
-  constructor(model) {
-    this.model = model;
-  }
-  async run({ messages, abortSignal, onUpdate }) {
-    const { fullStream } = await (0, import_ai.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 };
-  }
-};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ActionBarPrimitive,
@@ -1910,7 +1870,7 @@ var VercelModelAdapter = class {
   VercelRSCAssistantProvider,
   getVercelAIMessage,
   getVercelRSCMessage,
-  unstable_VercelModelAdapter,
+  unstable_MessageRepository,
   unstable_useComposerContext,
   unstable_useContentPartContext,
   unstable_useLocalRuntime,