@assistant-ui/react-ai-sdk 0.6.4 → 0.6.6

package/dist/index.js

@@ -69,111 +69,77 @@ var getVercelRSCMessage = (message) => {
   return (0, import_react2.getExternalStoreMessage)(message);
 };
 
-// src/ui/utils/useCachedChunkedMessages.ts
-var import_react3 = require("react");
-var hasItems = (messages) => messages.length > 0;
-var chunkedMessages = (messages) => {
-  const chunks = [];
-  let currentChunk = [];
-  for (const message of messages) {
-    if (message.role === "assistant" || message.role === "data") {
-      currentChunk.push(message);
-    } else {
-      if (hasItems(currentChunk)) {
-        chunks.push(currentChunk);
-        currentChunk = [];
-      }
-      chunks.push([message]);
-    }
-  }
-  if (hasItems(currentChunk)) {
-    chunks.push(currentChunk);
-  }
-  return chunks;
-};
-var shallowArrayEqual = (a, b) => {
-  if (a.length !== b.length) return false;
-  for (let i = 0; i < a.length; i++) {
-    if (a[i] !== b[i]) return false;
-  }
-  return true;
-};
-var useCachedChunkedMessages = (messages) => {
-  const cache = (0, import_react3.useMemo)(() => /* @__PURE__ */ new WeakMap(), []);
-  return (0, import_react3.useMemo)(() => {
-    return chunkedMessages(messages).map((m) => {
-      const key = m[0];
-      if (!key) return m;
-      const cached = cache.get(key);
-      if (cached && shallowArrayEqual(cached, m)) return cached;
-      cache.set(key, m);
-      return m;
-    });
-  }, [messages, cache]);
-};
-
 // src/ui/utils/convertMessage.ts
-var convertMessage = (messages) => {
-  const firstMessage = messages[0];
-  if (!firstMessage) throw new Error("No messages found");
-  const common = {
-    id: firstMessage.id,
-    createdAt: firstMessage.createdAt ?? /* @__PURE__ */ new Date()
-  };
-  switch (firstMessage.role) {
+var convertMessage = (message) => {
+  switch (message.role) {
     case "user":
-      if (messages.length > 1) {
-        throw new Error(
-          "Multiple user messages found. This is likely an internal bug in assistant-ui."
-        );
-      }
       return {
-        ...common,
         role: "user",
-        content: [{ type: "text", text: firstMessage.content }]
+        id: message.id,
+        createdAt: message.createdAt,
+        content: message.content,
+        attachments: message.experimental_attachments?.map(
+          (attachment, idx) => ({
+            id: idx.toString(),
+            type: "file",
+            name: attachment.name ?? attachment.url,
+            content: []
+          })
+        )
       };
     case "system":
       return {
-        ...common,
         role: "system",
-        content: [{ type: "text", text: firstMessage.content }]
+        id: message.id,
+        createdAt: message.createdAt,
+        content: message.content
       };
-    case "data":
-    case "assistant": {
-      const res = {
-        ...common,
+    case "assistant":
+      return {
         role: "assistant",
-        content: messages.flatMap((message) => {
-          return [
-            ...message.content ? [{ type: "text", text: message.content }] : [],
-            ...message.toolInvocations?.map(
-              (t) => ({
-                type: "tool-call",
-                toolName: t.toolName,
-                toolCallId: t.toolCallId,
-                argsText: JSON.stringify(t.args),
-                args: t.args,
-                result: "result" in t ? t.result : void 0
-              })
-            ) ?? [],
-            ...typeof message.data === "object" && !Array.isArray(message.data) && message.data?.["type"] === "tool-call" ? [message.data] : []
-          ];
-        })
+        id: message.id,
+        createdAt: message.createdAt,
+        content: [
+          ...message.content ? [
+            {
+              type: "text",
+              text: message.content
+            }
+          ] : [],
+          ...message.toolInvocations?.map(
+            (t) => ({
+              type: "tool-call",
+              toolName: t.toolName,
+              toolCallId: t.toolCallId,
+              argsText: JSON.stringify(t.args),
+              args: t.args,
+              result: "result" in t ? t.result : void 0
+            })
+          ) ?? []
+        ]
       };
-      for (const message of messages) {
-        if (typeof message.data === "object" && !Array.isArray(message.data) && message.data?.["type"] === "tool-result") {
-          const toolCallId = message.data["toolCallId"];
-          const toolContent = res.content.find(
-            (c) => c.type === "tool-call" && c.toolCallId === toolCallId
-          );
-          if (!toolContent) throw new Error("Tool call not found");
-          toolContent.result = message.data["result"];
-        }
+    case "data": {
+      if (!message.data || !(typeof message.data === "object") || Array.isArray(message.data))
+        return [];
+      const data = message.data;
+      if (data.type === "tool-call") {
+        return {
+          role: "assistant",
+          id: message.id,
+          createdAt: message.createdAt,
+          content: [data]
+        };
+      } else if (data.type === "tool-result") {
+        return {
+          role: "tool",
+          toolCallId: data.toolCallId,
+          result: data.result
+        };
       }
-      return res;
+      return [];
     }
     default:
-      const _unsupported = firstMessage.role;
+      const _unsupported = message.role;
       throw new Error(
         `You have a message with an unsupported role. The role ${_unsupported} is not supported.`
       );
@@ -184,18 +150,18 @@ var convertMessage = (messages) => {
 var import_react6 = require("@assistant-ui/react");
 
 // src/ui/utils/useInputSync.tsx
-var import_react4 = require("react");
-var import_react5 = require("@assistant-ui/react");
+var import_react3 = require("react");
+var import_react4 = require("@assistant-ui/react");
 var useInputSync = (helpers, runtime) => {
-  const helpersRef = (0, import_react4.useRef)(helpers);
-  (0, import_react4.useEffect)(() => {
+  const helpersRef = (0, import_react3.useRef)(helpers);
+  (0, import_react3.useEffect)(() => {
     helpersRef.current = helpers;
     if (runtime.thread.composer.text !== helpers.input) {
       runtime.thread.composer.setText(helpers.input);
     }
   }, [helpers, runtime]);
-  (0, import_react4.useEffect)(() => {
-    return (0, import_react5.subscribeToMainThread)(runtime, () => {
+  (0, import_react3.useEffect)(() => {
+    return (0, import_react4.subscribeToMainThread)(runtime, () => {
       if (runtime.thread.composer.text !== helpersRef.current.input) {
         helpersRef.current.setInput(runtime.thread.composer.text);
       }
@@ -270,13 +236,23 @@ var vercelAttachmentAdapter = {
   }
 };
 
+// src/ui/getVercelAIMessages.tsx
+var import_react5 = require("@assistant-ui/react");
+var getVercelAIMessages = (message) => {
+  return (0, import_react5.getExternalStoreMessage)(message);
+};
+
 // src/ui/use-chat/useVercelUseChatRuntime.tsx
 var useVercelUseChatRuntime = (chatHelpers) => {
-  const messages = useCachedChunkedMessages(chatHelpers.messages);
+  const messages = (0, import_react6.useExternalMessageConverter)({
+    callback: convertMessage,
+    isRunning: chatHelpers.isLoading,
+    messages: chatHelpers.messages
+  });
   const runtime = (0, import_react6.useExternalStoreRuntime)({
     isRunning: chatHelpers.isLoading,
     messages,
-    setMessages: (messages2) => chatHelpers.setMessages(messages2.flat()),
+    setMessages: (messages2) => chatHelpers.setMessages(messages2.map(getVercelAIMessages).flat()),
     onCancel: async () => chatHelpers.stop(),
     onNew: async (message) => {
       await chatHelpers.append(await toCreateMessage(message));
@@ -303,7 +279,6 @@ var useVercelUseChatRuntime = (chatHelpers) => {
       chatHelpers.setMessages([]);
       chatHelpers.setInput("");
     },
-    convertMessage,
     adapters: {
       attachments: vercelAttachmentAdapter
     }
@@ -315,7 +290,11 @@ var useVercelUseChatRuntime = (chatHelpers) => {
 // src/ui/use-assistant/useVercelUseAssistantRuntime.tsx
 var import_react7 = require("@assistant-ui/react");
 var useVercelUseAssistantRuntime = (assistantHelpers) => {
-  const messages = useCachedChunkedMessages(assistantHelpers.messages);
+  const messages = (0, import_react7.useExternalMessageConverter)({
+    callback: convertMessage,
+    isRunning: assistantHelpers.status === "in_progress",
+    messages: assistantHelpers.messages
+  });
   const runtime = (0, import_react7.useExternalStoreRuntime)({
     isRunning: assistantHelpers.status === "in_progress",
     messages,
@@ -329,7 +308,6 @@ var useVercelUseAssistantRuntime = (assistantHelpers) => {
       assistantHelpers.setMessages([]);
       assistantHelpers.setInput("");
     },
-    convertMessage,
     adapters: {
       attachments: vercelAttachmentAdapter
     }
@@ -337,12 +315,6 @@ var useVercelUseAssistantRuntime = (assistantHelpers) => {
   useInputSync(assistantHelpers, runtime);
   return runtime;
 };
-
-// src/ui/getVercelAIMessages.tsx
-var import_react8 = require("@assistant-ui/react");
-var getVercelAIMessages = (message) => {
-  return (0, import_react8.getExternalStoreMessage)(message);
-};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   getVercelAIMessage,

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/rsc/useVercelRSCRuntime.tsx","../src/rsc/getVercelRSCMessage.tsx","../src/ui/utils/useCachedChunkedMessages.ts","../src/ui/utils/convertMessage.ts","../src/ui/use-chat/useVercelUseChatRuntime.tsx","../src/ui/utils/useInputSync.tsx","../src/ui/utils/sliceMessagesUntil.tsx","../src/ui/utils/toCreateMessage.ts","../src/ui/utils/vercelAttachmentAdapter.ts","../src/ui/use-assistant/useVercelUseAssistantRuntime.tsx","../src/ui/getVercelAIMessages.tsx"],"sourcesContent":["export * from \"./rsc\";\nexport * from \"./ui\";\n","\"use client\";\n\nimport type { VercelRSCAdapter } from \"./VercelRSCAdapter\";\nimport {\n  ExternalStoreAdapter,\n  ThreadMessageLike,\n  useExternalStoreRuntime,\n} from \"@assistant-ui/react\";\nimport { VercelRSCMessage } from \"./VercelRSCMessage\";\n\nconst vercelToThreadMessage = <T,>(\n  converter: (message: T) => VercelRSCMessage,\n  rawMessage: T,\n): ThreadMessageLike => {\n  const message = converter(rawMessage);\n\n  return {\n    id: message.id,\n    role: message.role,\n    content: [{ type: \"ui\", display: message.display }],\n    createdAt: message.createdAt,\n  };\n};\n\nexport const useVercelRSCRuntime = <T extends WeakKey>(\n  adapter: VercelRSCAdapter<T>,\n) => {\n  const onNew = adapter.onNew ?? adapter.append;\n  if (!onNew)\n    throw new Error(\"You must pass a onNew function to useVercelRSCRuntime\");\n  const eAdapter: ExternalStoreAdapter<any> = {\n    isRunning: adapter.isRunning,\n    messages: adapter.messages,\n    onNew,\n    onEdit: adapter.onEdit ?? adapter.edit,\n    onReload: adapter.onReload ?? adapter.reload,\n    convertMessage: (m: T) =>\n      vercelToThreadMessage(\n        adapter.convertMessage ?? ((m) => m as VercelRSCMessage),\n        m,\n      ),\n    unstable_capabilities: {\n      copy: false,\n    },\n  };\n\n  const runtime = useExternalStoreRuntime(eAdapter);\n  return runtime;\n};\n","import {\n  getExternalStoreMessage,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\n\nexport const symbolInnerRSCMessage = Symbol(\"innerVercelRSCMessage\");\n\nexport const getVercelRSCMessage = (message: ThreadMessage) => {\n  return getExternalStoreMessage(message);\n};\n","import { Message } from \"ai\";\nimport { useMemo } from \"react\";\n\ntype Chunk = [Message, ...Message[]];\nconst hasItems = (messages: Message[]): messages is Chunk =>\n  messages.length > 0;\n\nconst chunkedMessages = (messages: Message[]): Chunk[] => {\n  const chunks: Chunk[] = [];\n  let currentChunk: Message[] = [];\n\n  for (const message of messages) {\n    if (message.role === \"assistant\" || message.role === \"data\") {\n      currentChunk.push(message);\n    } else {\n      if (hasItems(currentChunk)) {\n        chunks.push(currentChunk);\n        currentChunk = [];\n      }\n      chunks.push([message]);\n    }\n  }\n\n  if (hasItems(currentChunk)) {\n    chunks.push(currentChunk);\n  }\n\n  return chunks;\n};\n\nconst shallowArrayEqual = (a: unknown[], b: unknown[]) => {\n  if (a.length !== b.length) return false;\n  for (let i = 0; i < a.length; i++) {\n    if (a[i] !== b[i]) return false;\n  }\n  return true;\n};\n\nexport const useCachedChunkedMessages = (messages: Message[]) => {\n  const cache = useMemo(() => new WeakMap<Message, Message[]>(), []);\n\n  return useMemo(() => {\n    return chunkedMessages(messages).map((m) => {\n      const key = m[0];\n      if (!key) return m;\n\n      const cached = cache.get(key);\n      if (cached && shallowArrayEqual(cached, m)) return cached;\n      cache.set(key, m);\n      return m;\n    });\n  }, [messages, cache]);\n};\n","import { Message } from \"ai\";\nimport { ThreadMessageLike } from \"@assistant-ui/react\";\nimport { ToolCallContentPart } from \"@assistant-ui/react\";\nimport { TextContentPart } from \"@assistant-ui/react\";\n\nexport const convertMessage = (messages: Message[]): ThreadMessageLike => {\n  const firstMessage = messages[0];\n  if (!firstMessage) throw new Error(\"No messages found\");\n\n  const common = {\n    id: firstMessage.id,\n    createdAt: firstMessage.createdAt ?? new Date(),\n  };\n\n  switch (firstMessage.role) {\n    case \"user\":\n      if (messages.length > 1) {\n        throw new Error(\n          \"Multiple user messages found. This is likely an internal bug in assistant-ui.\",\n        );\n      }\n\n      return {\n        ...common,\n        role: \"user\",\n        content: [{ type: \"text\", text: firstMessage.content }],\n      };\n\n    case \"system\":\n      return {\n        ...common,\n        role: \"system\",\n        content: [{ type: \"text\", text: firstMessage.content }],\n      };\n\n    case \"data\":\n    case \"assistant\": {\n      const res: ThreadMessageLike & { content: any[] } = {\n        ...common,\n        role: \"assistant\",\n        content: messages.flatMap((message) => {\n          return [\n            ...(message.content\n              ? [{ type: \"text\", text: message.content } as TextContentPart]\n              : []),\n            ...(message.toolInvocations?.map(\n              (t) =>\n                ({\n                  type: \"tool-call\",\n                  toolName: t.toolName,\n                  toolCallId: t.toolCallId,\n                  argsText: JSON.stringify(t.args),\n                  args: t.args,\n                  result: \"result\" in t ? t.result : undefined,\n                }) satisfies ToolCallContentPart,\n            ) ?? []),\n            ...(typeof message.data === \"object\" &&\n            !Array.isArray(message.data) &&\n            message.data?.[\"type\"] === \"tool-call\"\n              ? [message.data as ToolCallContentPart]\n              : []),\n          ];\n        }),\n      };\n\n      for (const message of messages) {\n        if (\n          typeof message.data === \"object\" &&\n          !Array.isArray(message.data) &&\n          message.data?.[\"type\"] === \"tool-result\"\n        ) {\n          const toolCallId = message.data[\"toolCallId\"];\n          const toolContent = res.content.find(\n            (c) => c.type === \"tool-call\" && c.toolCallId === toolCallId,\n          ) as ToolCallContentPart | undefined;\n          if (!toolContent) throw new Error(\"Tool call not found\");\n          toolContent.result = message.data[\"result\"];\n        }\n      }\n\n      return res;\n    }\n\n    default:\n      const _unsupported: \"function\" | \"tool\" = firstMessage.role;\n      throw new Error(\n        `You have a message with an unsupported role. The role ${_unsupported} is not supported.`,\n      );\n  }\n};\n","import type { useChat } from \"ai/react\";\nimport { useCachedChunkedMessages } from \"../utils/useCachedChunkedMessages\";\nimport { convertMessage } from \"../utils/convertMessage\";\nimport { useExternalStoreRuntime } from \"@assistant-ui/react\";\nimport { useInputSync } from \"../utils/useInputSync\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\n\nexport const useVercelUseChatRuntime = (\n  chatHelpers: ReturnType<typeof useChat>,\n) => {\n  const messages = useCachedChunkedMessages(chatHelpers.messages);\n\n  const runtime = useExternalStoreRuntime({\n    isRunning: chatHelpers.isLoading,\n    messages,\n    setMessages: (messages) => chatHelpers.setMessages(messages.flat()),\n    onCancel: async () => chatHelpers.stop(),\n    onNew: async (message) => {\n      await chatHelpers.append(await toCreateMessage(message));\n    },\n    onEdit: async (message) => {\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.append(await toCreateMessage(message));\n    },\n    onReload: async (parentId: string | null) => {\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.reload();\n    },\n    onAddToolResult: ({ toolCallId, result }) => {\n      chatHelpers.addToolResult({ toolCallId, result });\n    },\n    onNewThread: () => {\n      chatHelpers.messages = [];\n      chatHelpers.input = \"\";\n      chatHelpers.setMessages([]);\n      chatHelpers.setInput(\"\");\n    },\n    convertMessage,\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n    },\n  });\n\n  useInputSync(chatHelpers, runtime);\n\n  return runtime;\n};\n","import { useRef, useEffect } from \"react\";\nimport {\n  ExternalStoreRuntime,\n  subscribeToMainThread,\n} from \"@assistant-ui/react\";\nimport { useAssistant, useChat } from \"ai/react\";\n\ntype VercelHelpers =\n  | ReturnType<typeof useChat>\n  | ReturnType<typeof useAssistant>;\n\nexport const useInputSync = (\n  helpers: VercelHelpers,\n  runtime: ExternalStoreRuntime,\n) => {\n  // sync input from vercel to assistant-ui\n  const helpersRef = useRef(helpers);\n  useEffect(() => {\n    helpersRef.current = helpers;\n    if (runtime.thread.composer.text !== helpers.input) {\n      runtime.thread.composer.setText(helpers.input);\n    }\n  }, [helpers, runtime]);\n\n  // sync input from assistant-ui to vercel\n  useEffect(() => {\n    return subscribeToMainThread(runtime, () => {\n      if (runtime.thread.composer.text !== helpersRef.current.input) {\n        helpersRef.current.setInput(runtime.thread.composer.text);\n      }\n    });\n  }, [runtime]);\n};\n","import type { Message } from \"ai\";\n\nexport const sliceMessagesUntil = (\n  messages: Message[],\n  messageId: string | null,\n) => {\n  if (messageId == null) return [];\n\n  let messageIdx = messages.findIndex((m) => m.id === messageId);\n  if (messageIdx === -1)\n    throw new Error(\n      \"useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui.\",\n    );\n\n  while (messages[messageIdx + 1]?.role === \"assistant\") {\n    messageIdx++;\n  }\n\n  return messages.slice(0, messageIdx + 1);\n};\n","import { AppendMessage } from \"@assistant-ui/react\";\nimport { CreateMessage } from \"ai\";\n\nexport const toCreateMessage = async (\n  message: AppendMessage,\n): Promise<CreateMessage> => {\n  const content = message.content\n    .filter((part) => part.type === \"text\")\n    .map((t) => t.text)\n    .join(\"\\n\\n\");\n\n  const images = message.content\n    .filter((part) => part.type === \"image\")\n    .map((part) => ({ url: part.image }));\n\n  return {\n    role: message.role,\n    content,\n    experimental_attachments: [\n      ...images,\n      ...(await Promise.all(\n        (message.attachments ?? []).map(async (m) => {\n          if (m.file == null)\n            throw new Error(\"Attachment did not contain a file\");\n          return {\n            contentType: m.file.type,\n            name: m.file.name,\n            url: await getFileDataURL(m.file),\n          };\n        }),\n      )),\n    ],\n  };\n};\n\nconst getFileDataURL = (file: File) =>\n  new Promise<string>((resolve, reject) => {\n    const reader = new FileReader();\n\n    reader.onload = () => resolve(reader.result as string);\n    reader.onerror = (error) => reject(error);\n\n    reader.readAsDataURL(file);\n  });\n","import { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { generateId } from \"ai\";\n\nexport const vercelAttachmentAdapter: AttachmentAdapter = {\n  accept:\n    \"image/*, text/plain, text/html, text/markdown, text/csv, text/xml, text/json, text/css\",\n  async add({ file }) {\n    return {\n      id: generateId(),\n      type: \"file\",\n      name: file.name,\n      file,\n      content: [],\n    };\n  },\n  async send(attachment) {\n    // noop\n    return {\n      ...attachment,\n      content: [],\n    };\n  },\n  async remove() {\n    // noop\n  },\n};\n","import type { useAssistant } from \"ai/react\";\nimport { useExternalStoreRuntime } from \"@assistant-ui/react\";\nimport { useCachedChunkedMessages } from \"../utils/useCachedChunkedMessages\";\nimport { convertMessage } from \"../utils/convertMessage\";\nimport { useInputSync } from \"../utils/useInputSync\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\n\nexport const useVercelUseAssistantRuntime = (\n  assistantHelpers: ReturnType<typeof useAssistant>,\n) => {\n  const messages = useCachedChunkedMessages(assistantHelpers.messages);\n  const runtime = useExternalStoreRuntime({\n    isRunning: assistantHelpers.status === \"in_progress\",\n    messages,\n    onCancel: async () => assistantHelpers.stop(),\n    onNew: async (message) => {\n      await assistantHelpers.append(await toCreateMessage(message));\n    },\n    onNewThread: () => {\n      assistantHelpers.messages = [];\n      assistantHelpers.input = \"\";\n      assistantHelpers.setMessages([]);\n      assistantHelpers.setInput(\"\");\n    },\n    convertMessage,\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n    },\n  });\n\n  useInputSync(assistantHelpers, runtime);\n\n  return runtime;\n};\n","import {\n  getExternalStoreMessage,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\nimport type { Message } from \"ai\";\n\nexport const getVercelAIMessages = (message: ThreadMessage) => {\n  return getExternalStoreMessage(message) as Message[];\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACGA,mBAIO;AAGP,IAAM,wBAAwB,CAC5B,WACA,eACsB;AACtB,QAAM,UAAU,UAAU,UAAU;AAEpC,SAAO;AAAA,IACL,IAAI,QAAQ;AAAA,IACZ,MAAM,QAAQ;AAAA,IACd,SAAS,CAAC,EAAE,MAAM,MAAM,SAAS,QAAQ,QAAQ,CAAC;AAAA,IAClD,WAAW,QAAQ;AAAA,EACrB;AACF;AAEO,IAAM,sBAAsB,CACjC,YACG;AACH,QAAM,QAAQ,QAAQ,SAAS,QAAQ;AACvC,MAAI,CAAC;AACH,UAAM,IAAI,MAAM,uDAAuD;AACzE,QAAM,WAAsC;AAAA,IAC1C,WAAW,QAAQ;AAAA,IACnB,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA,QAAQ,QAAQ,UAAU,QAAQ;AAAA,IAClC,UAAU,QAAQ,YAAY,QAAQ;AAAA,IACtC,gBAAgB,CAAC,MACf;AAAA,MACE,QAAQ,mBAAmB,CAACA,OAAMA;AAAA,MAClC;AAAA,IACF;AAAA,IACF,uBAAuB;AAAA,MACrB,MAAM;AAAA,IACR;AAAA,EACF;AAEA,QAAM,cAAU,sCAAwB,QAAQ;AAChD,SAAO;AACT;;;AChDA,IAAAC,gBAGO;AAEA,IAAM,wBAAwB,OAAO,uBAAuB;AAE5D,IAAM,sBAAsB,CAAC,YAA2B;AAC7D,aAAO,uCAAwB,OAAO;AACxC;;;ACRA,IAAAC,gBAAwB;AAGxB,IAAM,WAAW,CAAC,aAChB,SAAS,SAAS;AAEpB,IAAM,kBAAkB,CAAC,aAAiC;AACxD,QAAM,SAAkB,CAAC;AACzB,MAAI,eAA0B,CAAC;AAE/B,aAAW,WAAW,UAAU;AAC9B,QAAI,QAAQ,SAAS,eAAe,QAAQ,SAAS,QAAQ;AAC3D,mBAAa,KAAK,OAAO;AAAA,IAC3B,OAAO;AACL,UAAI,SAAS,YAAY,GAAG;AAC1B,eAAO,KAAK,YAAY;AACxB,uBAAe,CAAC;AAAA,MAClB;AACA,aAAO,KAAK,CAAC,OAAO,CAAC;AAAA,IACvB;AAAA,EACF;AAEA,MAAI,SAAS,YAAY,GAAG;AAC1B,WAAO,KAAK,YAAY;AAAA,EAC1B;AAEA,SAAO;AACT;AAEA,IAAM,oBAAoB,CAAC,GAAc,MAAiB;AACxD,MAAI,EAAE,WAAW,EAAE,OAAQ,QAAO;AAClC,WAAS,IAAI,GAAG,IAAI,EAAE,QAAQ,KAAK;AACjC,QAAI,EAAE,CAAC,MAAM,EAAE,CAAC,EAAG,QAAO;AAAA,EAC5B;AACA,SAAO;AACT;AAEO,IAAM,2BAA2B,CAAC,aAAwB;AAC/D,QAAM,YAAQ,uBAAQ,MAAM,oBAAI,QAA4B,GAAG,CAAC,CAAC;AAEjE,aAAO,uBAAQ,MAAM;AACnB,WAAO,gBAAgB,QAAQ,EAAE,IAAI,CAAC,MAAM;AAC1C,YAAM,MAAM,EAAE,CAAC;AACf,UAAI,CAAC,IAAK,QAAO;AAEjB,YAAM,SAAS,MAAM,IAAI,GAAG;AAC5B,UAAI,UAAU,kBAAkB,QAAQ,CAAC,EAAG,QAAO;AACnD,YAAM,IAAI,KAAK,CAAC;AAChB,aAAO;AAAA,IACT,CAAC;AAAA,EACH,GAAG,CAAC,UAAU,KAAK,CAAC;AACtB;;;AC/CO,IAAM,iBAAiB,CAAC,aAA2C;AACxE,QAAM,eAAe,SAAS,CAAC;AAC/B,MAAI,CAAC,aAAc,OAAM,IAAI,MAAM,mBAAmB;AAEtD,QAAM,SAAS;AAAA,IACb,IAAI,aAAa;AAAA,IACjB,WAAW,aAAa,aAAa,oBAAI,KAAK;AAAA,EAChD;AAEA,UAAQ,aAAa,MAAM;AAAA,IACzB,KAAK;AACH,UAAI,SAAS,SAAS,GAAG;AACvB,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AAEA,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAM;AAAA,QACN,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,aAAa,QAAQ,CAAC;AAAA,MACxD;AAAA,IAEF,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAM;AAAA,QACN,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,aAAa,QAAQ,CAAC;AAAA,MACxD;AAAA,IAEF,KAAK;AAAA,IACL,KAAK,aAAa;AAChB,YAAM,MAA8C;AAAA,QAClD,GAAG;AAAA,QACH,MAAM;AAAA,QACN,SAAS,SAAS,QAAQ,CAAC,YAAY;AACrC,iBAAO;AAAA,YACL,GAAI,QAAQ,UACR,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,QAAQ,CAAoB,IAC3D,CAAC;AAAA,YACL,GAAI,QAAQ,iBAAiB;AAAA,cAC3B,CAAC,OACE;AAAA,gBACC,MAAM;AAAA,gBACN,UAAU,EAAE;AAAA,gBACZ,YAAY,EAAE;AAAA,gBACd,UAAU,KAAK,UAAU,EAAE,IAAI;AAAA,gBAC/B,MAAM,EAAE;AAAA,gBACR,QAAQ,YAAY,IAAI,EAAE,SAAS;AAAA,cACrC;AAAA,YACJ,KAAK,CAAC;AAAA,YACN,GAAI,OAAO,QAAQ,SAAS,YAC5B,CAAC,MAAM,QAAQ,QAAQ,IAAI,KAC3B,QAAQ,OAAO,MAAM,MAAM,cACvB,CAAC,QAAQ,IAA2B,IACpC,CAAC;AAAA,UACP;AAAA,QACF,CAAC;AAAA,MACH;AAEA,iBAAW,WAAW,UAAU;AAC9B,YACE,OAAO,QAAQ,SAAS,YACxB,CAAC,MAAM,QAAQ,QAAQ,IAAI,KAC3B,QAAQ,OAAO,MAAM,MAAM,eAC3B;AACA,gBAAM,aAAa,QAAQ,KAAK,YAAY;AAC5C,gBAAM,cAAc,IAAI,QAAQ;AAAA,YAC9B,CAAC,MAAM,EAAE,SAAS,eAAe,EAAE,eAAe;AAAA,UACpD;AACA,cAAI,CAAC,YAAa,OAAM,IAAI,MAAM,qBAAqB;AACvD,sBAAY,SAAS,QAAQ,KAAK,QAAQ;AAAA,QAC5C;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IAEA;AACE,YAAM,eAAoC,aAAa;AACvD,YAAM,IAAI;AAAA,QACR,yDAAyD,YAAY;AAAA,MACvE;AAAA,EACJ;AACF;;;ACtFA,IAAAC,gBAAwC;;;ACHxC,IAAAC,gBAAkC;AAClC,IAAAA,gBAGO;AAOA,IAAM,eAAe,CAC1B,SACA,YACG;AAEH,QAAM,iBAAa,sBAAO,OAAO;AACjC,+BAAU,MAAM;AACd,eAAW,UAAU;AACrB,QAAI,QAAQ,OAAO,SAAS,SAAS,QAAQ,OAAO;AAClD,cAAQ,OAAO,SAAS,QAAQ,QAAQ,KAAK;AAAA,IAC/C;AAAA,EACF,GAAG,CAAC,SAAS,OAAO,CAAC;AAGrB,+BAAU,MAAM;AACd,eAAO,qCAAsB,SAAS,MAAM;AAC1C,UAAI,QAAQ,OAAO,SAAS,SAAS,WAAW,QAAQ,OAAO;AAC7D,mBAAW,QAAQ,SAAS,QAAQ,OAAO,SAAS,IAAI;AAAA,MAC1D;AAAA,IACF,CAAC;AAAA,EACH,GAAG,CAAC,OAAO,CAAC;AACd;;;AC9BO,IAAM,qBAAqB,CAChC,UACA,cACG;AACH,MAAI,aAAa,KAAM,QAAO,CAAC;AAE/B,MAAI,aAAa,SAAS,UAAU,CAAC,MAAM,EAAE,OAAO,SAAS;AAC7D,MAAI,eAAe;AACjB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAEF,SAAO,SAAS,aAAa,CAAC,GAAG,SAAS,aAAa;AACrD;AAAA,EACF;AAEA,SAAO,SAAS,MAAM,GAAG,aAAa,CAAC;AACzC;;;AChBO,IAAM,kBAAkB,OAC7B,YAC2B;AAC3B,QAAM,UAAU,QAAQ,QACrB,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,MAAM,EAAE,IAAI,EACjB,KAAK,MAAM;AAEd,QAAM,SAAS,QAAQ,QACpB,OAAO,CAAC,SAAS,KAAK,SAAS,OAAO,EACtC,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,MAAM,EAAE;AAEtC,SAAO;AAAA,IACL,MAAM,QAAQ;AAAA,IACd;AAAA,IACA,0BAA0B;AAAA,MACxB,GAAG;AAAA,MACH,GAAI,MAAM,QAAQ;AAAA,SACf,QAAQ,eAAe,CAAC,GAAG,IAAI,OAAO,MAAM;AAC3C,cAAI,EAAE,QAAQ;AACZ,kBAAM,IAAI,MAAM,mCAAmC;AACrD,iBAAO;AAAA,YACL,aAAa,EAAE,KAAK;AAAA,YACpB,MAAM,EAAE,KAAK;AAAA,YACb,KAAK,MAAM,eAAe,EAAE,IAAI;AAAA,UAClC;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,iBAAiB,CAAC,SACtB,IAAI,QAAgB,CAAC,SAAS,WAAW;AACvC,QAAM,SAAS,IAAI,WAAW;AAE9B,SAAO,SAAS,MAAM,QAAQ,OAAO,MAAgB;AACrD,SAAO,UAAU,CAAC,UAAU,OAAO,KAAK;AAExC,SAAO,cAAc,IAAI;AAC3B,CAAC;;;AC1CH,gBAA2B;AAEpB,IAAM,0BAA6C;AAAA,EACxD,QACE;AAAA,EACF,MAAM,IAAI,EAAE,KAAK,GAAG;AAClB,WAAO;AAAA,MACL,QAAI,sBAAW;AAAA,MACf,MAAM;AAAA,MACN,MAAM,KAAK;AAAA,MACX;AAAA,MACA,SAAS,CAAC;AAAA,IACZ;AAAA,EACF;AAAA,EACA,MAAM,KAAK,YAAY;AAErB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,SAAS,CAAC;AAAA,IACZ;AAAA,EACF;AAAA,EACA,MAAM,SAAS;AAAA,EAEf;AACF;;;AJhBO,IAAM,0BAA0B,CACrC,gBACG;AACH,QAAM,WAAW,yBAAyB,YAAY,QAAQ;AAE9D,QAAM,cAAU,uCAAwB;AAAA,IACtC,WAAW,YAAY;AAAA,IACvB;AAAA,IACA,aAAa,CAACC,cAAa,YAAY,YAAYA,UAAS,KAAK,CAAC;AAAA,IAClE,UAAU,YAAY,YAAY,KAAK;AAAA,IACvC,OAAO,OAAO,YAAY;AACxB,YAAM,YAAY,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IACzD;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IACzD;AAAA,IACA,UAAU,OAAO,aAA4B;AAC3C,YAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;AACrE,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO;AAAA,IAC3B;AAAA,IACA,iBAAiB,CAAC,EAAE,YAAY,OAAO,MAAM;AAC3C,kBAAY,cAAc,EAAE,YAAY,OAAO,CAAC;AAAA,IAClD;AAAA,IACA,aAAa,MAAM;AACjB,kBAAY,WAAW,CAAC;AACxB,kBAAY,QAAQ;AACpB,kBAAY,YAAY,CAAC,CAAC;AAC1B,kBAAY,SAAS,EAAE;AAAA,IACzB;AAAA,IACA;AAAA,IACA,UAAU;AAAA,MACR,aAAa;AAAA,IACf;AAAA,EACF,CAAC;AAED,eAAa,aAAa,OAAO;AAEjC,SAAO;AACT;;;AKtDA,IAAAC,gBAAwC;AAOjC,IAAM,+BAA+B,CAC1C,qBACG;AACH,QAAM,WAAW,yBAAyB,iBAAiB,QAAQ;AACnE,QAAM,cAAU,uCAAwB;AAAA,IACtC,WAAW,iBAAiB,WAAW;AAAA,IACvC;AAAA,IACA,UAAU,YAAY,iBAAiB,KAAK;AAAA,IAC5C,OAAO,OAAO,YAAY;AACxB,YAAM,iBAAiB,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IAC9D;AAAA,IACA,aAAa,MAAM;AACjB,uBAAiB,WAAW,CAAC;AAC7B,uBAAiB,QAAQ;AACzB,uBAAiB,YAAY,CAAC,CAAC;AAC/B,uBAAiB,SAAS,EAAE;AAAA,IAC9B;AAAA,IACA;AAAA,IACA,UAAU;AAAA,MACR,aAAa;AAAA,IACf;AAAA,EACF,CAAC;AAED,eAAa,kBAAkB,OAAO;AAEtC,SAAO;AACT;;;AClCA,IAAAC,gBAGO;AAGA,IAAM,sBAAsB,CAAC,YAA2B;AAC7D,aAAO,uCAAwB,OAAO;AACxC;","names":["m","import_react","import_react","import_react","import_react","messages","import_react","import_react"]}
+{"version":3,"sources":["../src/index.ts","../src/rsc/useVercelRSCRuntime.tsx","../src/rsc/getVercelRSCMessage.tsx","../src/ui/utils/convertMessage.ts","../src/ui/use-chat/useVercelUseChatRuntime.tsx","../src/ui/utils/useInputSync.tsx","../src/ui/utils/sliceMessagesUntil.tsx","../src/ui/utils/toCreateMessage.ts","../src/ui/utils/vercelAttachmentAdapter.ts","../src/ui/getVercelAIMessages.tsx","../src/ui/use-assistant/useVercelUseAssistantRuntime.tsx"],"sourcesContent":["export * from \"./rsc\";\nexport * from \"./ui\";\n","\"use client\";\n\nimport type { VercelRSCAdapter } from \"./VercelRSCAdapter\";\nimport {\n  ExternalStoreAdapter,\n  ThreadMessageLike,\n  useExternalStoreRuntime,\n} from \"@assistant-ui/react\";\nimport { VercelRSCMessage } from \"./VercelRSCMessage\";\n\nconst vercelToThreadMessage = <T,>(\n  converter: (message: T) => VercelRSCMessage,\n  rawMessage: T,\n): ThreadMessageLike => {\n  const message = converter(rawMessage);\n\n  return {\n    id: message.id,\n    role: message.role,\n    content: [{ type: \"ui\", display: message.display }],\n    createdAt: message.createdAt,\n  };\n};\n\nexport const useVercelRSCRuntime = <T extends WeakKey>(\n  adapter: VercelRSCAdapter<T>,\n) => {\n  const onNew = adapter.onNew ?? adapter.append;\n  if (!onNew)\n    throw new Error(\"You must pass a onNew function to useVercelRSCRuntime\");\n  const eAdapter: ExternalStoreAdapter<any> = {\n    isRunning: adapter.isRunning,\n    messages: adapter.messages,\n    onNew,\n    onEdit: adapter.onEdit ?? adapter.edit,\n    onReload: adapter.onReload ?? adapter.reload,\n    convertMessage: (m: T) =>\n      vercelToThreadMessage(\n        adapter.convertMessage ?? ((m) => m as VercelRSCMessage),\n        m,\n      ),\n    unstable_capabilities: {\n      copy: false,\n    },\n  };\n\n  const runtime = useExternalStoreRuntime(eAdapter);\n  return runtime;\n};\n","import {\n  getExternalStoreMessage,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\n\nexport const symbolInnerRSCMessage = Symbol(\"innerVercelRSCMessage\");\n\nexport const getVercelRSCMessage = (message: ThreadMessage) => {\n  return getExternalStoreMessage(message);\n};\n","import { Message } from \"ai\";\nimport { useExternalMessageConverter } from \"@assistant-ui/react\";\nimport { ToolCallContentPart } from \"@assistant-ui/react\";\nimport { TextContentPart } from \"@assistant-ui/react\";\nimport { MessageAttachment } from \"../../../../react/src/context/stores/Attachment\";\n\nexport const convertMessage: useExternalMessageConverter.Callback<Message> = (\n  message,\n) => {\n  switch (message.role) {\n    case \"user\":\n      return {\n        role: \"user\",\n        id: message.id,\n        createdAt: message.createdAt,\n        content: message.content,\n\n        attachments: message.experimental_attachments?.map(\n          (attachment, idx) =>\n            ({\n              id: idx.toString(),\n              type: \"file\",\n              name: attachment.name ?? attachment.url,\n              content: [],\n            }) satisfies MessageAttachment,\n        ),\n      };\n\n    case \"system\":\n      return {\n        role: \"system\",\n        id: message.id,\n        createdAt: message.createdAt,\n        content: message.content,\n      };\n\n    case \"assistant\":\n      return {\n        role: \"assistant\",\n        id: message.id,\n        createdAt: message.createdAt,\n        content: [\n          ...(message.content\n            ? [\n                {\n                  type: \"text\",\n                  text: message.content,\n                } satisfies TextContentPart,\n              ]\n            : []),\n          ...(message.toolInvocations?.map(\n            (t) =>\n              ({\n                type: \"tool-call\",\n                toolName: t.toolName,\n                toolCallId: t.toolCallId,\n                argsText: JSON.stringify(t.args),\n                args: t.args,\n                result: \"result\" in t ? t.result : undefined,\n              }) satisfies ToolCallContentPart,\n          ) ?? []),\n        ],\n      };\n\n    case \"data\": {\n      type MaybeSupportedDataMessage =\n        | { type?: \"unsafe_other\" }\n        | ToolCallContentPart\n        | {\n            type: \"tool-result\";\n            toolCallId: string;\n            result: any;\n          };\n\n      if (\n        !message.data ||\n        !(typeof message.data === \"object\") ||\n        Array.isArray(message.data)\n      )\n        return [];\n\n      const data = message.data as MaybeSupportedDataMessage;\n\n      if (data.type === \"tool-call\") {\n        return {\n          role: \"assistant\",\n          id: message.id,\n          createdAt: message.createdAt,\n          content: [data],\n        };\n      } else if (data.type === \"tool-result\") {\n        return {\n          role: \"tool\",\n          toolCallId: data.toolCallId,\n          result: data.result,\n        };\n      }\n      return [];\n    }\n\n    default:\n      // TODO handle tool and function messages\n      const _unsupported: \"function\" | \"tool\" = message.role;\n      throw new Error(\n        `You have a message with an unsupported role. The role ${_unsupported} is not supported.`,\n      );\n  }\n};\n","import type { useChat } from \"ai/react\";\nimport { convertMessage } from \"../utils/convertMessage\";\nimport {\n  useExternalMessageConverter,\n  useExternalStoreRuntime,\n} from \"@assistant-ui/react\";\nimport { useInputSync } from \"../utils/useInputSync\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../getVercelAIMessages\";\n\nexport const useVercelUseChatRuntime = (\n  chatHelpers: ReturnType<typeof useChat>,\n) => {\n  const messages = useExternalMessageConverter({\n    callback: convertMessage,\n    isRunning: chatHelpers.isLoading,\n    messages: chatHelpers.messages,\n  });\n\n  const runtime = useExternalStoreRuntime({\n    isRunning: chatHelpers.isLoading,\n    messages,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(messages.map(getVercelAIMessages).flat()),\n    onCancel: async () => chatHelpers.stop(),\n    onNew: async (message) => {\n      await chatHelpers.append(await toCreateMessage(message));\n    },\n    onEdit: async (message) => {\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.append(await toCreateMessage(message));\n    },\n    onReload: async (parentId: string | null) => {\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.reload();\n    },\n    onAddToolResult: ({ toolCallId, result }) => {\n      chatHelpers.addToolResult({ toolCallId, result });\n    },\n    onNewThread: () => {\n      chatHelpers.messages = [];\n      chatHelpers.input = \"\";\n      chatHelpers.setMessages([]);\n      chatHelpers.setInput(\"\");\n    },\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n    },\n  });\n\n  useInputSync(chatHelpers, runtime);\n\n  return runtime;\n};\n","import { useRef, useEffect } from \"react\";\nimport {\n  ExternalStoreRuntime,\n  subscribeToMainThread,\n} from \"@assistant-ui/react\";\nimport { useAssistant, useChat } from \"ai/react\";\n\ntype VercelHelpers =\n  | ReturnType<typeof useChat>\n  | ReturnType<typeof useAssistant>;\n\nexport const useInputSync = (\n  helpers: VercelHelpers,\n  runtime: ExternalStoreRuntime,\n) => {\n  // sync input from vercel to assistant-ui\n  const helpersRef = useRef(helpers);\n  useEffect(() => {\n    helpersRef.current = helpers;\n    if (runtime.thread.composer.text !== helpers.input) {\n      runtime.thread.composer.setText(helpers.input);\n    }\n  }, [helpers, runtime]);\n\n  // sync input from assistant-ui to vercel\n  useEffect(() => {\n    return subscribeToMainThread(runtime, () => {\n      if (runtime.thread.composer.text !== helpersRef.current.input) {\n        helpersRef.current.setInput(runtime.thread.composer.text);\n      }\n    });\n  }, [runtime]);\n};\n","import type { Message } from \"ai\";\n\nexport const sliceMessagesUntil = (\n  messages: Message[],\n  messageId: string | null,\n) => {\n  if (messageId == null) return [];\n\n  let messageIdx = messages.findIndex((m) => m.id === messageId);\n  if (messageIdx === -1)\n    throw new Error(\n      \"useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui.\",\n    );\n\n  while (messages[messageIdx + 1]?.role === \"assistant\") {\n    messageIdx++;\n  }\n\n  return messages.slice(0, messageIdx + 1);\n};\n","import { AppendMessage } from \"@assistant-ui/react\";\nimport { CreateMessage } from \"ai\";\n\nexport const toCreateMessage = async (\n  message: AppendMessage,\n): Promise<CreateMessage> => {\n  const content = message.content\n    .filter((part) => part.type === \"text\")\n    .map((t) => t.text)\n    .join(\"\\n\\n\");\n\n  const images = message.content\n    .filter((part) => part.type === \"image\")\n    .map((part) => ({ url: part.image }));\n\n  return {\n    role: message.role,\n    content,\n    experimental_attachments: [\n      ...images,\n      ...(await Promise.all(\n        (message.attachments ?? []).map(async (m) => {\n          if (m.file == null)\n            throw new Error(\"Attachment did not contain a file\");\n          return {\n            contentType: m.file.type,\n            name: m.file.name,\n            url: await getFileDataURL(m.file),\n          };\n        }),\n      )),\n    ],\n  };\n};\n\nconst getFileDataURL = (file: File) =>\n  new Promise<string>((resolve, reject) => {\n    const reader = new FileReader();\n\n    reader.onload = () => resolve(reader.result as string);\n    reader.onerror = (error) => reject(error);\n\n    reader.readAsDataURL(file);\n  });\n","import { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { generateId } from \"ai\";\n\nexport const vercelAttachmentAdapter: AttachmentAdapter = {\n  accept:\n    \"image/*, text/plain, text/html, text/markdown, text/csv, text/xml, text/json, text/css\",\n  async add({ file }) {\n    return {\n      id: generateId(),\n      type: \"file\",\n      name: file.name,\n      file,\n      content: [],\n    };\n  },\n  async send(attachment) {\n    // noop\n    return {\n      ...attachment,\n      content: [],\n    };\n  },\n  async remove() {\n    // noop\n  },\n};\n","import {\n  getExternalStoreMessage,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\nimport type { Message } from \"ai\";\n\nexport const getVercelAIMessages = (message: ThreadMessage) => {\n  return getExternalStoreMessage(message) as Message[];\n};\n","import type { useAssistant } from \"ai/react\";\nimport {\n  useExternalMessageConverter,\n  useExternalStoreRuntime,\n} from \"@assistant-ui/react\";\nimport { convertMessage } from \"../utils/convertMessage\";\nimport { useInputSync } from \"../utils/useInputSync\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\n\nexport const useVercelUseAssistantRuntime = (\n  assistantHelpers: ReturnType<typeof useAssistant>,\n) => {\n  const messages = useExternalMessageConverter({\n    callback: convertMessage,\n    isRunning: assistantHelpers.status === \"in_progress\",\n    messages: assistantHelpers.messages,\n  });\n  const runtime = useExternalStoreRuntime({\n    isRunning: assistantHelpers.status === \"in_progress\",\n    messages,\n    onCancel: async () => assistantHelpers.stop(),\n    onNew: async (message) => {\n      await assistantHelpers.append(await toCreateMessage(message));\n    },\n    onNewThread: () => {\n      assistantHelpers.messages = [];\n      assistantHelpers.input = \"\";\n      assistantHelpers.setMessages([]);\n      assistantHelpers.setInput(\"\");\n    },\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n    },\n  });\n\n  useInputSync(assistantHelpers, runtime);\n\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACGA,mBAIO;AAGP,IAAM,wBAAwB,CAC5B,WACA,eACsB;AACtB,QAAM,UAAU,UAAU,UAAU;AAEpC,SAAO;AAAA,IACL,IAAI,QAAQ;AAAA,IACZ,MAAM,QAAQ;AAAA,IACd,SAAS,CAAC,EAAE,MAAM,MAAM,SAAS,QAAQ,QAAQ,CAAC;AAAA,IAClD,WAAW,QAAQ;AAAA,EACrB;AACF;AAEO,IAAM,sBAAsB,CACjC,YACG;AACH,QAAM,QAAQ,QAAQ,SAAS,QAAQ;AACvC,MAAI,CAAC;AACH,UAAM,IAAI,MAAM,uDAAuD;AACzE,QAAM,WAAsC;AAAA,IAC1C,WAAW,QAAQ;AAAA,IACnB,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA,QAAQ,QAAQ,UAAU,QAAQ;AAAA,IAClC,UAAU,QAAQ,YAAY,QAAQ;AAAA,IACtC,gBAAgB,CAAC,MACf;AAAA,MACE,QAAQ,mBAAmB,CAACA,OAAMA;AAAA,MAClC;AAAA,IACF;AAAA,IACF,uBAAuB;AAAA,MACrB,MAAM;AAAA,IACR;AAAA,EACF;AAEA,QAAM,cAAU,sCAAwB,QAAQ;AAChD,SAAO;AACT;;;AChDA,IAAAC,gBAGO;AAEA,IAAM,wBAAwB,OAAO,uBAAuB;AAE5D,IAAM,sBAAsB,CAAC,YAA2B;AAC7D,aAAO,uCAAwB,OAAO;AACxC;;;ACHO,IAAM,iBAAgE,CAC3E,YACG;AACH,UAAQ,QAAQ,MAAM;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,WAAW,QAAQ;AAAA,QACnB,SAAS,QAAQ;AAAA,QAEjB,aAAa,QAAQ,0BAA0B;AAAA,UAC7C,CAAC,YAAY,SACV;AAAA,YACC,IAAI,IAAI,SAAS;AAAA,YACjB,MAAM;AAAA,YACN,MAAM,WAAW,QAAQ,WAAW;AAAA,YACpC,SAAS,CAAC;AAAA,UACZ;AAAA,QACJ;AAAA,MACF;AAAA,IAEF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,WAAW,QAAQ;AAAA,QACnB,SAAS,QAAQ;AAAA,MACnB;AAAA,IAEF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,WAAW,QAAQ;AAAA,QACnB,SAAS;AAAA,UACP,GAAI,QAAQ,UACR;AAAA,YACE;AAAA,cACE,MAAM;AAAA,cACN,MAAM,QAAQ;AAAA,YAChB;AAAA,UACF,IACA,CAAC;AAAA,UACL,GAAI,QAAQ,iBAAiB;AAAA,YAC3B,CAAC,OACE;AAAA,cACC,MAAM;AAAA,cACN,UAAU,EAAE;AAAA,cACZ,YAAY,EAAE;AAAA,cACd,UAAU,KAAK,UAAU,EAAE,IAAI;AAAA,cAC/B,MAAM,EAAE;AAAA,cACR,QAAQ,YAAY,IAAI,EAAE,SAAS;AAAA,YACrC;AAAA,UACJ,KAAK,CAAC;AAAA,QACR;AAAA,MACF;AAAA,IAEF,KAAK,QAAQ;AAUX,UACE,CAAC,QAAQ,QACT,EAAE,OAAO,QAAQ,SAAS,aAC1B,MAAM,QAAQ,QAAQ,IAAI;AAE1B,eAAO,CAAC;AAEV,YAAM,OAAO,QAAQ;AAErB,UAAI,KAAK,SAAS,aAAa;AAC7B,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ,WAAW,QAAQ;AAAA,UACnB,SAAS,CAAC,IAAI;AAAA,QAChB;AAAA,MACF,WAAW,KAAK,SAAS,eAAe;AACtC,eAAO;AAAA,UACL,MAAM;AAAA,UACN,YAAY,KAAK;AAAA,UACjB,QAAQ,KAAK;AAAA,QACf;AAAA,MACF;AACA,aAAO,CAAC;AAAA,IACV;AAAA,IAEA;AAEE,YAAM,eAAoC,QAAQ;AAClD,YAAM,IAAI;AAAA,QACR,yDAAyD,YAAY;AAAA,MACvE;AAAA,EACJ;AACF;;;ACzGA,IAAAC,gBAGO;;;ACLP,IAAAC,gBAAkC;AAClC,IAAAA,gBAGO;AAOA,IAAM,eAAe,CAC1B,SACA,YACG;AAEH,QAAM,iBAAa,sBAAO,OAAO;AACjC,+BAAU,MAAM;AACd,eAAW,UAAU;AACrB,QAAI,QAAQ,OAAO,SAAS,SAAS,QAAQ,OAAO;AAClD,cAAQ,OAAO,SAAS,QAAQ,QAAQ,KAAK;AAAA,IAC/C;AAAA,EACF,GAAG,CAAC,SAAS,OAAO,CAAC;AAGrB,+BAAU,MAAM;AACd,eAAO,qCAAsB,SAAS,MAAM;AAC1C,UAAI,QAAQ,OAAO,SAAS,SAAS,WAAW,QAAQ,OAAO;AAC7D,mBAAW,QAAQ,SAAS,QAAQ,OAAO,SAAS,IAAI;AAAA,MAC1D;AAAA,IACF,CAAC;AAAA,EACH,GAAG,CAAC,OAAO,CAAC;AACd;;;AC9BO,IAAM,qBAAqB,CAChC,UACA,cACG;AACH,MAAI,aAAa,KAAM,QAAO,CAAC;AAE/B,MAAI,aAAa,SAAS,UAAU,CAAC,MAAM,EAAE,OAAO,SAAS;AAC7D,MAAI,eAAe;AACjB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAEF,SAAO,SAAS,aAAa,CAAC,GAAG,SAAS,aAAa;AACrD;AAAA,EACF;AAEA,SAAO,SAAS,MAAM,GAAG,aAAa,CAAC;AACzC;;;AChBO,IAAM,kBAAkB,OAC7B,YAC2B;AAC3B,QAAM,UAAU,QAAQ,QACrB,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,MAAM,EAAE,IAAI,EACjB,KAAK,MAAM;AAEd,QAAM,SAAS,QAAQ,QACpB,OAAO,CAAC,SAAS,KAAK,SAAS,OAAO,EACtC,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,MAAM,EAAE;AAEtC,SAAO;AAAA,IACL,MAAM,QAAQ;AAAA,IACd;AAAA,IACA,0BAA0B;AAAA,MACxB,GAAG;AAAA,MACH,GAAI,MAAM,QAAQ;AAAA,SACf,QAAQ,eAAe,CAAC,GAAG,IAAI,OAAO,MAAM;AAC3C,cAAI,EAAE,QAAQ;AACZ,kBAAM,IAAI,MAAM,mCAAmC;AACrD,iBAAO;AAAA,YACL,aAAa,EAAE,KAAK;AAAA,YACpB,MAAM,EAAE,KAAK;AAAA,YACb,KAAK,MAAM,eAAe,EAAE,IAAI;AAAA,UAClC;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,iBAAiB,CAAC,SACtB,IAAI,QAAgB,CAAC,SAAS,WAAW;AACvC,QAAM,SAAS,IAAI,WAAW;AAE9B,SAAO,SAAS,MAAM,QAAQ,OAAO,MAAgB;AACrD,SAAO,UAAU,CAAC,UAAU,OAAO,KAAK;AAExC,SAAO,cAAc,IAAI;AAC3B,CAAC;;;AC1CH,gBAA2B;AAEpB,IAAM,0BAA6C;AAAA,EACxD,QACE;AAAA,EACF,MAAM,IAAI,EAAE,KAAK,GAAG;AAClB,WAAO;AAAA,MACL,QAAI,sBAAW;AAAA,MACf,MAAM;AAAA,MACN,MAAM,KAAK;AAAA,MACX;AAAA,MACA,SAAS,CAAC;AAAA,IACZ;AAAA,EACF;AAAA,EACA,MAAM,KAAK,YAAY;AAErB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,SAAS,CAAC;AAAA,IACZ;AAAA,EACF;AAAA,EACA,MAAM,SAAS;AAAA,EAEf;AACF;;;ACzBA,IAAAC,gBAGO;AAGA,IAAM,sBAAsB,CAAC,YAA2B;AAC7D,aAAO,uCAAwB,OAAO;AACxC;;;ALIO,IAAM,0BAA0B,CACrC,gBACG;AACH,QAAM,eAAW,2CAA4B;AAAA,IAC3C,UAAU;AAAA,IACV,WAAW,YAAY;AAAA,IACvB,UAAU,YAAY;AAAA,EACxB,CAAC;AAED,QAAM,cAAU,uCAAwB;AAAA,IACtC,WAAW,YAAY;AAAA,IACvB;AAAA,IACA,aAAa,CAACC,cACZ,YAAY,YAAYA,UAAS,IAAI,mBAAmB,EAAE,KAAK,CAAC;AAAA,IAClE,UAAU,YAAY,YAAY,KAAK;AAAA,IACvC,OAAO,OAAO,YAAY;AACxB,YAAM,YAAY,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IACzD;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IACzD;AAAA,IACA,UAAU,OAAO,aAA4B;AAC3C,YAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;AACrE,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO;AAAA,IAC3B;AAAA,IACA,iBAAiB,CAAC,EAAE,YAAY,OAAO,MAAM;AAC3C,kBAAY,cAAc,EAAE,YAAY,OAAO,CAAC;AAAA,IAClD;AAAA,IACA,aAAa,MAAM;AACjB,kBAAY,WAAW,CAAC;AACxB,kBAAY,QAAQ;AACpB,kBAAY,YAAY,CAAC,CAAC;AAC1B,kBAAY,SAAS,EAAE;AAAA,IACzB;AAAA,IACA,UAAU;AAAA,MACR,aAAa;AAAA,IACf;AAAA,EACF,CAAC;AAED,eAAa,aAAa,OAAO;AAEjC,SAAO;AACT;;;AM7DA,IAAAC,gBAGO;AAMA,IAAM,+BAA+B,CAC1C,qBACG;AACH,QAAM,eAAW,2CAA4B;AAAA,IAC3C,UAAU;AAAA,IACV,WAAW,iBAAiB,WAAW;AAAA,IACvC,UAAU,iBAAiB;AAAA,EAC7B,CAAC;AACD,QAAM,cAAU,uCAAwB;AAAA,IACtC,WAAW,iBAAiB,WAAW;AAAA,IACvC;AAAA,IACA,UAAU,YAAY,iBAAiB,KAAK;AAAA,IAC5C,OAAO,OAAO,YAAY;AACxB,YAAM,iBAAiB,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IAC9D;AAAA,IACA,aAAa,MAAM;AACjB,uBAAiB,WAAW,CAAC;AAC7B,uBAAiB,QAAQ;AACzB,uBAAiB,YAAY,CAAC,CAAC;AAC/B,uBAAiB,SAAS,EAAE;AAAA,IAC9B;AAAA,IACA,UAAU;AAAA,MACR,aAAa;AAAA,IACf;AAAA,EACF,CAAC;AAED,eAAa,kBAAkB,OAAO;AAEtC,SAAO;AACT;","names":["m","import_react","import_react","import_react","import_react","messages","import_react"]}

package/dist/index.mjs

@@ -42,111 +42,77 @@ var getVercelRSCMessage = (message) => {
   return getExternalStoreMessage(message);
 };
 
-// src/ui/utils/useCachedChunkedMessages.ts
-import { useMemo } from "react";
-var hasItems = (messages) => messages.length > 0;
-var chunkedMessages = (messages) => {
-  const chunks = [];
-  let currentChunk = [];
-  for (const message of messages) {
-    if (message.role === "assistant" || message.role === "data") {
-      currentChunk.push(message);
-    } else {
-      if (hasItems(currentChunk)) {
-        chunks.push(currentChunk);
-        currentChunk = [];
-      }
-      chunks.push([message]);
-    }
-  }
-  if (hasItems(currentChunk)) {
-    chunks.push(currentChunk);
-  }
-  return chunks;
-};
-var shallowArrayEqual = (a, b) => {
-  if (a.length !== b.length) return false;
-  for (let i = 0; i < a.length; i++) {
-    if (a[i] !== b[i]) return false;
-  }
-  return true;
-};
-var useCachedChunkedMessages = (messages) => {
-  const cache = useMemo(() => /* @__PURE__ */ new WeakMap(), []);
-  return useMemo(() => {
-    return chunkedMessages(messages).map((m) => {
-      const key = m[0];
-      if (!key) return m;
-      const cached = cache.get(key);
-      if (cached && shallowArrayEqual(cached, m)) return cached;
-      cache.set(key, m);
-      return m;
-    });
-  }, [messages, cache]);
-};
-
 // src/ui/utils/convertMessage.ts
-var convertMessage = (messages) => {
-  const firstMessage = messages[0];
-  if (!firstMessage) throw new Error("No messages found");
-  const common = {
-    id: firstMessage.id,
-    createdAt: firstMessage.createdAt ?? /* @__PURE__ */ new Date()
-  };
-  switch (firstMessage.role) {
+var convertMessage = (message) => {
+  switch (message.role) {
     case "user":
-      if (messages.length > 1) {
-        throw new Error(
-          "Multiple user messages found. This is likely an internal bug in assistant-ui."
-        );
-      }
       return {
-        ...common,
         role: "user",
-        content: [{ type: "text", text: firstMessage.content }]
+        id: message.id,
+        createdAt: message.createdAt,
+        content: message.content,
+        attachments: message.experimental_attachments?.map(
+          (attachment, idx) => ({
+            id: idx.toString(),
+            type: "file",
+            name: attachment.name ?? attachment.url,
+            content: []
+          })
+        )
       };
     case "system":
       return {
-        ...common,
         role: "system",
-        content: [{ type: "text", text: firstMessage.content }]
+        id: message.id,
+        createdAt: message.createdAt,
+        content: message.content
       };
-    case "data":
-    case "assistant": {
-      const res = {
-        ...common,
+    case "assistant":
+      return {
         role: "assistant",
-        content: messages.flatMap((message) => {
-          return [
-            ...message.content ? [{ type: "text", text: message.content }] : [],
-            ...message.toolInvocations?.map(
-              (t) => ({
-                type: "tool-call",
-                toolName: t.toolName,
-                toolCallId: t.toolCallId,
-                argsText: JSON.stringify(t.args),
-                args: t.args,
-                result: "result" in t ? t.result : void 0
-              })
-            ) ?? [],
-            ...typeof message.data === "object" && !Array.isArray(message.data) && message.data?.["type"] === "tool-call" ? [message.data] : []
-          ];
-        })
+        id: message.id,
+        createdAt: message.createdAt,
+        content: [
+          ...message.content ? [
+            {
+              type: "text",
+              text: message.content
+            }
+          ] : [],
+          ...message.toolInvocations?.map(
+            (t) => ({
+              type: "tool-call",
+              toolName: t.toolName,
+              toolCallId: t.toolCallId,
+              argsText: JSON.stringify(t.args),
+              args: t.args,
+              result: "result" in t ? t.result : void 0
+            })
+          ) ?? []
+        ]
       };
-      for (const message of messages) {
-        if (typeof message.data === "object" && !Array.isArray(message.data) && message.data?.["type"] === "tool-result") {
-          const toolCallId = message.data["toolCallId"];
-          const toolContent = res.content.find(
-            (c) => c.type === "tool-call" && c.toolCallId === toolCallId
-          );
-          if (!toolContent) throw new Error("Tool call not found");
-          toolContent.result = message.data["result"];
-        }
+    case "data": {
+      if (!message.data || !(typeof message.data === "object") || Array.isArray(message.data))
+        return [];
+      const data = message.data;
+      if (data.type === "tool-call") {
+        return {
+          role: "assistant",
+          id: message.id,
+          createdAt: message.createdAt,
+          content: [data]
+        };
+      } else if (data.type === "tool-result") {
+        return {
+          role: "tool",
+          toolCallId: data.toolCallId,
+          result: data.result
+        };
       }
-      return res;
+      return [];
     }
     default:
-      const _unsupported = firstMessage.role;
+      const _unsupported = message.role;
       throw new Error(
         `You have a message with an unsupported role. The role ${_unsupported} is not supported.`
       );
@@ -154,7 +120,10 @@ var convertMessage = (messages) => {
 };
 
 // src/ui/use-chat/useVercelUseChatRuntime.tsx
-import { useExternalStoreRuntime as useExternalStoreRuntime2 } from "@assistant-ui/react";
+import {
+  useExternalMessageConverter,
+  useExternalStoreRuntime as useExternalStoreRuntime2
+} from "@assistant-ui/react";
 
 // src/ui/utils/useInputSync.tsx
 import { useRef, useEffect } from "react";
@@ -245,13 +214,25 @@ var vercelAttachmentAdapter = {
   }
 };
 
+// src/ui/getVercelAIMessages.tsx
+import {
+  getExternalStoreMessage as getExternalStoreMessage2
+} from "@assistant-ui/react";
+var getVercelAIMessages = (message) => {
+  return getExternalStoreMessage2(message);
+};
+
 // src/ui/use-chat/useVercelUseChatRuntime.tsx
 var useVercelUseChatRuntime = (chatHelpers) => {
-  const messages = useCachedChunkedMessages(chatHelpers.messages);
+  const messages = useExternalMessageConverter({
+    callback: convertMessage,
+    isRunning: chatHelpers.isLoading,
+    messages: chatHelpers.messages
+  });
   const runtime = useExternalStoreRuntime2({
     isRunning: chatHelpers.isLoading,
     messages,
-    setMessages: (messages2) => chatHelpers.setMessages(messages2.flat()),
+    setMessages: (messages2) => chatHelpers.setMessages(messages2.map(getVercelAIMessages).flat()),
     onCancel: async () => chatHelpers.stop(),
     onNew: async (message) => {
       await chatHelpers.append(await toCreateMessage(message));
@@ -278,7 +259,6 @@ var useVercelUseChatRuntime = (chatHelpers) => {
       chatHelpers.setMessages([]);
       chatHelpers.setInput("");
     },
-    convertMessage,
     adapters: {
       attachments: vercelAttachmentAdapter
     }
@@ -288,9 +268,16 @@ var useVercelUseChatRuntime = (chatHelpers) => {
 };
 
 // src/ui/use-assistant/useVercelUseAssistantRuntime.tsx
-import { useExternalStoreRuntime as useExternalStoreRuntime3 } from "@assistant-ui/react";
+import {
+  useExternalMessageConverter as useExternalMessageConverter2,
+  useExternalStoreRuntime as useExternalStoreRuntime3
+} from "@assistant-ui/react";
 var useVercelUseAssistantRuntime = (assistantHelpers) => {
-  const messages = useCachedChunkedMessages(assistantHelpers.messages);
+  const messages = useExternalMessageConverter2({
+    callback: convertMessage,
+    isRunning: assistantHelpers.status === "in_progress",
+    messages: assistantHelpers.messages
+  });
   const runtime = useExternalStoreRuntime3({
     isRunning: assistantHelpers.status === "in_progress",
     messages,
@@ -304,7 +291,6 @@ var useVercelUseAssistantRuntime = (assistantHelpers) => {
       assistantHelpers.setMessages([]);
       assistantHelpers.setInput("");
     },
-    convertMessage,
     adapters: {
       attachments: vercelAttachmentAdapter
     }
@@ -312,14 +298,6 @@ var useVercelUseAssistantRuntime = (assistantHelpers) => {
   useInputSync(assistantHelpers, runtime);
   return runtime;
 };
-
-// src/ui/getVercelAIMessages.tsx
-import {
-  getExternalStoreMessage as getExternalStoreMessage2
-} from "@assistant-ui/react";
-var getVercelAIMessages = (message) => {
-  return getExternalStoreMessage2(message);
-};
 export {
   getVercelAIMessages as getVercelAIMessage,
   getVercelAIMessages,

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/rsc/useVercelRSCRuntime.tsx","../src/rsc/getVercelRSCMessage.tsx","../src/ui/utils/useCachedChunkedMessages.ts","../src/ui/utils/convertMessage.ts","../src/ui/use-chat/useVercelUseChatRuntime.tsx","../src/ui/utils/useInputSync.tsx","../src/ui/utils/sliceMessagesUntil.tsx","../src/ui/utils/toCreateMessage.ts","../src/ui/utils/vercelAttachmentAdapter.ts","../src/ui/use-assistant/useVercelUseAssistantRuntime.tsx","../src/ui/getVercelAIMessages.tsx"],"sourcesContent":["\"use client\";\n\nimport type { VercelRSCAdapter } from \"./VercelRSCAdapter\";\nimport {\n  ExternalStoreAdapter,\n  ThreadMessageLike,\n  useExternalStoreRuntime,\n} from \"@assistant-ui/react\";\nimport { VercelRSCMessage } from \"./VercelRSCMessage\";\n\nconst vercelToThreadMessage = <T,>(\n  converter: (message: T) => VercelRSCMessage,\n  rawMessage: T,\n): ThreadMessageLike => {\n  const message = converter(rawMessage);\n\n  return {\n    id: message.id,\n    role: message.role,\n    content: [{ type: \"ui\", display: message.display }],\n    createdAt: message.createdAt,\n  };\n};\n\nexport const useVercelRSCRuntime = <T extends WeakKey>(\n  adapter: VercelRSCAdapter<T>,\n) => {\n  const onNew = adapter.onNew ?? adapter.append;\n  if (!onNew)\n    throw new Error(\"You must pass a onNew function to useVercelRSCRuntime\");\n  const eAdapter: ExternalStoreAdapter<any> = {\n    isRunning: adapter.isRunning,\n    messages: adapter.messages,\n    onNew,\n    onEdit: adapter.onEdit ?? adapter.edit,\n    onReload: adapter.onReload ?? adapter.reload,\n    convertMessage: (m: T) =>\n      vercelToThreadMessage(\n        adapter.convertMessage ?? ((m) => m as VercelRSCMessage),\n        m,\n      ),\n    unstable_capabilities: {\n      copy: false,\n    },\n  };\n\n  const runtime = useExternalStoreRuntime(eAdapter);\n  return runtime;\n};\n","import {\n  getExternalStoreMessage,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\n\nexport const symbolInnerRSCMessage = Symbol(\"innerVercelRSCMessage\");\n\nexport const getVercelRSCMessage = (message: ThreadMessage) => {\n  return getExternalStoreMessage(message);\n};\n","import { Message } from \"ai\";\nimport { useMemo } from \"react\";\n\ntype Chunk = [Message, ...Message[]];\nconst hasItems = (messages: Message[]): messages is Chunk =>\n  messages.length > 0;\n\nconst chunkedMessages = (messages: Message[]): Chunk[] => {\n  const chunks: Chunk[] = [];\n  let currentChunk: Message[] = [];\n\n  for (const message of messages) {\n    if (message.role === \"assistant\" || message.role === \"data\") {\n      currentChunk.push(message);\n    } else {\n      if (hasItems(currentChunk)) {\n        chunks.push(currentChunk);\n        currentChunk = [];\n      }\n      chunks.push([message]);\n    }\n  }\n\n  if (hasItems(currentChunk)) {\n    chunks.push(currentChunk);\n  }\n\n  return chunks;\n};\n\nconst shallowArrayEqual = (a: unknown[], b: unknown[]) => {\n  if (a.length !== b.length) return false;\n  for (let i = 0; i < a.length; i++) {\n    if (a[i] !== b[i]) return false;\n  }\n  return true;\n};\n\nexport const useCachedChunkedMessages = (messages: Message[]) => {\n  const cache = useMemo(() => new WeakMap<Message, Message[]>(), []);\n\n  return useMemo(() => {\n    return chunkedMessages(messages).map((m) => {\n      const key = m[0];\n      if (!key) return m;\n\n      const cached = cache.get(key);\n      if (cached && shallowArrayEqual(cached, m)) return cached;\n      cache.set(key, m);\n      return m;\n    });\n  }, [messages, cache]);\n};\n","import { Message } from \"ai\";\nimport { ThreadMessageLike } from \"@assistant-ui/react\";\nimport { ToolCallContentPart } from \"@assistant-ui/react\";\nimport { TextContentPart } from \"@assistant-ui/react\";\n\nexport const convertMessage = (messages: Message[]): ThreadMessageLike => {\n  const firstMessage = messages[0];\n  if (!firstMessage) throw new Error(\"No messages found\");\n\n  const common = {\n    id: firstMessage.id,\n    createdAt: firstMessage.createdAt ?? new Date(),\n  };\n\n  switch (firstMessage.role) {\n    case \"user\":\n      if (messages.length > 1) {\n        throw new Error(\n          \"Multiple user messages found. This is likely an internal bug in assistant-ui.\",\n        );\n      }\n\n      return {\n        ...common,\n        role: \"user\",\n        content: [{ type: \"text\", text: firstMessage.content }],\n      };\n\n    case \"system\":\n      return {\n        ...common,\n        role: \"system\",\n        content: [{ type: \"text\", text: firstMessage.content }],\n      };\n\n    case \"data\":\n    case \"assistant\": {\n      const res: ThreadMessageLike & { content: any[] } = {\n        ...common,\n        role: \"assistant\",\n        content: messages.flatMap((message) => {\n          return [\n            ...(message.content\n              ? [{ type: \"text\", text: message.content } as TextContentPart]\n              : []),\n            ...(message.toolInvocations?.map(\n              (t) =>\n                ({\n                  type: \"tool-call\",\n                  toolName: t.toolName,\n                  toolCallId: t.toolCallId,\n                  argsText: JSON.stringify(t.args),\n                  args: t.args,\n                  result: \"result\" in t ? t.result : undefined,\n                }) satisfies ToolCallContentPart,\n            ) ?? []),\n            ...(typeof message.data === \"object\" &&\n            !Array.isArray(message.data) &&\n            message.data?.[\"type\"] === \"tool-call\"\n              ? [message.data as ToolCallContentPart]\n              : []),\n          ];\n        }),\n      };\n\n      for (const message of messages) {\n        if (\n          typeof message.data === \"object\" &&\n          !Array.isArray(message.data) &&\n          message.data?.[\"type\"] === \"tool-result\"\n        ) {\n          const toolCallId = message.data[\"toolCallId\"];\n          const toolContent = res.content.find(\n            (c) => c.type === \"tool-call\" && c.toolCallId === toolCallId,\n          ) as ToolCallContentPart | undefined;\n          if (!toolContent) throw new Error(\"Tool call not found\");\n          toolContent.result = message.data[\"result\"];\n        }\n      }\n\n      return res;\n    }\n\n    default:\n      const _unsupported: \"function\" | \"tool\" = firstMessage.role;\n      throw new Error(\n        `You have a message with an unsupported role. The role ${_unsupported} is not supported.`,\n      );\n  }\n};\n","import type { useChat } from \"ai/react\";\nimport { useCachedChunkedMessages } from \"../utils/useCachedChunkedMessages\";\nimport { convertMessage } from \"../utils/convertMessage\";\nimport { useExternalStoreRuntime } from \"@assistant-ui/react\";\nimport { useInputSync } from \"../utils/useInputSync\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\n\nexport const useVercelUseChatRuntime = (\n  chatHelpers: ReturnType<typeof useChat>,\n) => {\n  const messages = useCachedChunkedMessages(chatHelpers.messages);\n\n  const runtime = useExternalStoreRuntime({\n    isRunning: chatHelpers.isLoading,\n    messages,\n    setMessages: (messages) => chatHelpers.setMessages(messages.flat()),\n    onCancel: async () => chatHelpers.stop(),\n    onNew: async (message) => {\n      await chatHelpers.append(await toCreateMessage(message));\n    },\n    onEdit: async (message) => {\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.append(await toCreateMessage(message));\n    },\n    onReload: async (parentId: string | null) => {\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.reload();\n    },\n    onAddToolResult: ({ toolCallId, result }) => {\n      chatHelpers.addToolResult({ toolCallId, result });\n    },\n    onNewThread: () => {\n      chatHelpers.messages = [];\n      chatHelpers.input = \"\";\n      chatHelpers.setMessages([]);\n      chatHelpers.setInput(\"\");\n    },\n    convertMessage,\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n    },\n  });\n\n  useInputSync(chatHelpers, runtime);\n\n  return runtime;\n};\n","import { useRef, useEffect } from \"react\";\nimport {\n  ExternalStoreRuntime,\n  subscribeToMainThread,\n} from \"@assistant-ui/react\";\nimport { useAssistant, useChat } from \"ai/react\";\n\ntype VercelHelpers =\n  | ReturnType<typeof useChat>\n  | ReturnType<typeof useAssistant>;\n\nexport const useInputSync = (\n  helpers: VercelHelpers,\n  runtime: ExternalStoreRuntime,\n) => {\n  // sync input from vercel to assistant-ui\n  const helpersRef = useRef(helpers);\n  useEffect(() => {\n    helpersRef.current = helpers;\n    if (runtime.thread.composer.text !== helpers.input) {\n      runtime.thread.composer.setText(helpers.input);\n    }\n  }, [helpers, runtime]);\n\n  // sync input from assistant-ui to vercel\n  useEffect(() => {\n    return subscribeToMainThread(runtime, () => {\n      if (runtime.thread.composer.text !== helpersRef.current.input) {\n        helpersRef.current.setInput(runtime.thread.composer.text);\n      }\n    });\n  }, [runtime]);\n};\n","import type { Message } from \"ai\";\n\nexport const sliceMessagesUntil = (\n  messages: Message[],\n  messageId: string | null,\n) => {\n  if (messageId == null) return [];\n\n  let messageIdx = messages.findIndex((m) => m.id === messageId);\n  if (messageIdx === -1)\n    throw new Error(\n      \"useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui.\",\n    );\n\n  while (messages[messageIdx + 1]?.role === \"assistant\") {\n    messageIdx++;\n  }\n\n  return messages.slice(0, messageIdx + 1);\n};\n","import { AppendMessage } from \"@assistant-ui/react\";\nimport { CreateMessage } from \"ai\";\n\nexport const toCreateMessage = async (\n  message: AppendMessage,\n): Promise<CreateMessage> => {\n  const content = message.content\n    .filter((part) => part.type === \"text\")\n    .map((t) => t.text)\n    .join(\"\\n\\n\");\n\n  const images = message.content\n    .filter((part) => part.type === \"image\")\n    .map((part) => ({ url: part.image }));\n\n  return {\n    role: message.role,\n    content,\n    experimental_attachments: [\n      ...images,\n      ...(await Promise.all(\n        (message.attachments ?? []).map(async (m) => {\n          if (m.file == null)\n            throw new Error(\"Attachment did not contain a file\");\n          return {\n            contentType: m.file.type,\n            name: m.file.name,\n            url: await getFileDataURL(m.file),\n          };\n        }),\n      )),\n    ],\n  };\n};\n\nconst getFileDataURL = (file: File) =>\n  new Promise<string>((resolve, reject) => {\n    const reader = new FileReader();\n\n    reader.onload = () => resolve(reader.result as string);\n    reader.onerror = (error) => reject(error);\n\n    reader.readAsDataURL(file);\n  });\n","import { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { generateId } from \"ai\";\n\nexport const vercelAttachmentAdapter: AttachmentAdapter = {\n  accept:\n    \"image/*, text/plain, text/html, text/markdown, text/csv, text/xml, text/json, text/css\",\n  async add({ file }) {\n    return {\n      id: generateId(),\n      type: \"file\",\n      name: file.name,\n      file,\n      content: [],\n    };\n  },\n  async send(attachment) {\n    // noop\n    return {\n      ...attachment,\n      content: [],\n    };\n  },\n  async remove() {\n    // noop\n  },\n};\n","import type { useAssistant } from \"ai/react\";\nimport { useExternalStoreRuntime } from \"@assistant-ui/react\";\nimport { useCachedChunkedMessages } from \"../utils/useCachedChunkedMessages\";\nimport { convertMessage } from \"../utils/convertMessage\";\nimport { useInputSync } from \"../utils/useInputSync\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\n\nexport const useVercelUseAssistantRuntime = (\n  assistantHelpers: ReturnType<typeof useAssistant>,\n) => {\n  const messages = useCachedChunkedMessages(assistantHelpers.messages);\n  const runtime = useExternalStoreRuntime({\n    isRunning: assistantHelpers.status === \"in_progress\",\n    messages,\n    onCancel: async () => assistantHelpers.stop(),\n    onNew: async (message) => {\n      await assistantHelpers.append(await toCreateMessage(message));\n    },\n    onNewThread: () => {\n      assistantHelpers.messages = [];\n      assistantHelpers.input = \"\";\n      assistantHelpers.setMessages([]);\n      assistantHelpers.setInput(\"\");\n    },\n    convertMessage,\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n    },\n  });\n\n  useInputSync(assistantHelpers, runtime);\n\n  return runtime;\n};\n","import {\n  getExternalStoreMessage,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\nimport type { Message } from \"ai\";\n\nexport const getVercelAIMessages = (message: ThreadMessage) => {\n  return getExternalStoreMessage(message) as Message[];\n};\n"],"mappings":";AAGA;AAAA,EAGE;AAAA,OACK;AAGP,IAAM,wBAAwB,CAC5B,WACA,eACsB;AACtB,QAAM,UAAU,UAAU,UAAU;AAEpC,SAAO;AAAA,IACL,IAAI,QAAQ;AAAA,IACZ,MAAM,QAAQ;AAAA,IACd,SAAS,CAAC,EAAE,MAAM,MAAM,SAAS,QAAQ,QAAQ,CAAC;AAAA,IAClD,WAAW,QAAQ;AAAA,EACrB;AACF;AAEO,IAAM,sBAAsB,CACjC,YACG;AACH,QAAM,QAAQ,QAAQ,SAAS,QAAQ;AACvC,MAAI,CAAC;AACH,UAAM,IAAI,MAAM,uDAAuD;AACzE,QAAM,WAAsC;AAAA,IAC1C,WAAW,QAAQ;AAAA,IACnB,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA,QAAQ,QAAQ,UAAU,QAAQ;AAAA,IAClC,UAAU,QAAQ,YAAY,QAAQ;AAAA,IACtC,gBAAgB,CAAC,MACf;AAAA,MACE,QAAQ,mBAAmB,CAACA,OAAMA;AAAA,MAClC;AAAA,IACF;AAAA,IACF,uBAAuB;AAAA,MACrB,MAAM;AAAA,IACR;AAAA,EACF;AAEA,QAAM,UAAU,wBAAwB,QAAQ;AAChD,SAAO;AACT;;;AChDA;AAAA,EACE;AAAA,OAEK;AAEA,IAAM,wBAAwB,OAAO,uBAAuB;AAE5D,IAAM,sBAAsB,CAAC,YAA2B;AAC7D,SAAO,wBAAwB,OAAO;AACxC;;;ACRA,SAAS,eAAe;AAGxB,IAAM,WAAW,CAAC,aAChB,SAAS,SAAS;AAEpB,IAAM,kBAAkB,CAAC,aAAiC;AACxD,QAAM,SAAkB,CAAC;AACzB,MAAI,eAA0B,CAAC;AAE/B,aAAW,WAAW,UAAU;AAC9B,QAAI,QAAQ,SAAS,eAAe,QAAQ,SAAS,QAAQ;AAC3D,mBAAa,KAAK,OAAO;AAAA,IAC3B,OAAO;AACL,UAAI,SAAS,YAAY,GAAG;AAC1B,eAAO,KAAK,YAAY;AACxB,uBAAe,CAAC;AAAA,MAClB;AACA,aAAO,KAAK,CAAC,OAAO,CAAC;AAAA,IACvB;AAAA,EACF;AAEA,MAAI,SAAS,YAAY,GAAG;AAC1B,WAAO,KAAK,YAAY;AAAA,EAC1B;AAEA,SAAO;AACT;AAEA,IAAM,oBAAoB,CAAC,GAAc,MAAiB;AACxD,MAAI,EAAE,WAAW,EAAE,OAAQ,QAAO;AAClC,WAAS,IAAI,GAAG,IAAI,EAAE,QAAQ,KAAK;AACjC,QAAI,EAAE,CAAC,MAAM,EAAE,CAAC,EAAG,QAAO;AAAA,EAC5B;AACA,SAAO;AACT;AAEO,IAAM,2BAA2B,CAAC,aAAwB;AAC/D,QAAM,QAAQ,QAAQ,MAAM,oBAAI,QAA4B,GAAG,CAAC,CAAC;AAEjE,SAAO,QAAQ,MAAM;AACnB,WAAO,gBAAgB,QAAQ,EAAE,IAAI,CAAC,MAAM;AAC1C,YAAM,MAAM,EAAE,CAAC;AACf,UAAI,CAAC,IAAK,QAAO;AAEjB,YAAM,SAAS,MAAM,IAAI,GAAG;AAC5B,UAAI,UAAU,kBAAkB,QAAQ,CAAC,EAAG,QAAO;AACnD,YAAM,IAAI,KAAK,CAAC;AAChB,aAAO;AAAA,IACT,CAAC;AAAA,EACH,GAAG,CAAC,UAAU,KAAK,CAAC;AACtB;;;AC/CO,IAAM,iBAAiB,CAAC,aAA2C;AACxE,QAAM,eAAe,SAAS,CAAC;AAC/B,MAAI,CAAC,aAAc,OAAM,IAAI,MAAM,mBAAmB;AAEtD,QAAM,SAAS;AAAA,IACb,IAAI,aAAa;AAAA,IACjB,WAAW,aAAa,aAAa,oBAAI,KAAK;AAAA,EAChD;AAEA,UAAQ,aAAa,MAAM;AAAA,IACzB,KAAK;AACH,UAAI,SAAS,SAAS,GAAG;AACvB,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AAEA,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAM;AAAA,QACN,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,aAAa,QAAQ,CAAC;AAAA,MACxD;AAAA,IAEF,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAM;AAAA,QACN,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,aAAa,QAAQ,CAAC;AAAA,MACxD;AAAA,IAEF,KAAK;AAAA,IACL,KAAK,aAAa;AAChB,YAAM,MAA8C;AAAA,QAClD,GAAG;AAAA,QACH,MAAM;AAAA,QACN,SAAS,SAAS,QAAQ,CAAC,YAAY;AACrC,iBAAO;AAAA,YACL,GAAI,QAAQ,UACR,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,QAAQ,CAAoB,IAC3D,CAAC;AAAA,YACL,GAAI,QAAQ,iBAAiB;AAAA,cAC3B,CAAC,OACE;AAAA,gBACC,MAAM;AAAA,gBACN,UAAU,EAAE;AAAA,gBACZ,YAAY,EAAE;AAAA,gBACd,UAAU,KAAK,UAAU,EAAE,IAAI;AAAA,gBAC/B,MAAM,EAAE;AAAA,gBACR,QAAQ,YAAY,IAAI,EAAE,SAAS;AAAA,cACrC;AAAA,YACJ,KAAK,CAAC;AAAA,YACN,GAAI,OAAO,QAAQ,SAAS,YAC5B,CAAC,MAAM,QAAQ,QAAQ,IAAI,KAC3B,QAAQ,OAAO,MAAM,MAAM,cACvB,CAAC,QAAQ,IAA2B,IACpC,CAAC;AAAA,UACP;AAAA,QACF,CAAC;AAAA,MACH;AAEA,iBAAW,WAAW,UAAU;AAC9B,YACE,OAAO,QAAQ,SAAS,YACxB,CAAC,MAAM,QAAQ,QAAQ,IAAI,KAC3B,QAAQ,OAAO,MAAM,MAAM,eAC3B;AACA,gBAAM,aAAa,QAAQ,KAAK,YAAY;AAC5C,gBAAM,cAAc,IAAI,QAAQ;AAAA,YAC9B,CAAC,MAAM,EAAE,SAAS,eAAe,EAAE,eAAe;AAAA,UACpD;AACA,cAAI,CAAC,YAAa,OAAM,IAAI,MAAM,qBAAqB;AACvD,sBAAY,SAAS,QAAQ,KAAK,QAAQ;AAAA,QAC5C;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IAEA;AACE,YAAM,eAAoC,aAAa;AACvD,YAAM,IAAI;AAAA,QACR,yDAAyD,YAAY;AAAA,MACvE;AAAA,EACJ;AACF;;;ACtFA,SAAS,2BAAAC,gCAA+B;;;ACHxC,SAAS,QAAQ,iBAAiB;AAClC;AAAA,EAEE;AAAA,OACK;AAOA,IAAM,eAAe,CAC1B,SACA,YACG;AAEH,QAAM,aAAa,OAAO,OAAO;AACjC,YAAU,MAAM;AACd,eAAW,UAAU;AACrB,QAAI,QAAQ,OAAO,SAAS,SAAS,QAAQ,OAAO;AAClD,cAAQ,OAAO,SAAS,QAAQ,QAAQ,KAAK;AAAA,IAC/C;AAAA,EACF,GAAG,CAAC,SAAS,OAAO,CAAC;AAGrB,YAAU,MAAM;AACd,WAAO,sBAAsB,SAAS,MAAM;AAC1C,UAAI,QAAQ,OAAO,SAAS,SAAS,WAAW,QAAQ,OAAO;AAC7D,mBAAW,QAAQ,SAAS,QAAQ,OAAO,SAAS,IAAI;AAAA,MAC1D;AAAA,IACF,CAAC;AAAA,EACH,GAAG,CAAC,OAAO,CAAC;AACd;;;AC9BO,IAAM,qBAAqB,CAChC,UACA,cACG;AACH,MAAI,aAAa,KAAM,QAAO,CAAC;AAE/B,MAAI,aAAa,SAAS,UAAU,CAAC,MAAM,EAAE,OAAO,SAAS;AAC7D,MAAI,eAAe;AACjB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAEF,SAAO,SAAS,aAAa,CAAC,GAAG,SAAS,aAAa;AACrD;AAAA,EACF;AAEA,SAAO,SAAS,MAAM,GAAG,aAAa,CAAC;AACzC;;;AChBO,IAAM,kBAAkB,OAC7B,YAC2B;AAC3B,QAAM,UAAU,QAAQ,QACrB,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,MAAM,EAAE,IAAI,EACjB,KAAK,MAAM;AAEd,QAAM,SAAS,QAAQ,QACpB,OAAO,CAAC,SAAS,KAAK,SAAS,OAAO,EACtC,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,MAAM,EAAE;AAEtC,SAAO;AAAA,IACL,MAAM,QAAQ;AAAA,IACd;AAAA,IACA,0BAA0B;AAAA,MACxB,GAAG;AAAA,MACH,GAAI,MAAM,QAAQ;AAAA,SACf,QAAQ,eAAe,CAAC,GAAG,IAAI,OAAO,MAAM;AAC3C,cAAI,EAAE,QAAQ;AACZ,kBAAM,IAAI,MAAM,mCAAmC;AACrD,iBAAO;AAAA,YACL,aAAa,EAAE,KAAK;AAAA,YACpB,MAAM,EAAE,KAAK;AAAA,YACb,KAAK,MAAM,eAAe,EAAE,IAAI;AAAA,UAClC;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,iBAAiB,CAAC,SACtB,IAAI,QAAgB,CAAC,SAAS,WAAW;AACvC,QAAM,SAAS,IAAI,WAAW;AAE9B,SAAO,SAAS,MAAM,QAAQ,OAAO,MAAgB;AACrD,SAAO,UAAU,CAAC,UAAU,OAAO,KAAK;AAExC,SAAO,cAAc,IAAI;AAC3B,CAAC;;;AC1CH,SAAS,kBAAkB;AAEpB,IAAM,0BAA6C;AAAA,EACxD,QACE;AAAA,EACF,MAAM,IAAI,EAAE,KAAK,GAAG;AAClB,WAAO;AAAA,MACL,IAAI,WAAW;AAAA,MACf,MAAM;AAAA,MACN,MAAM,KAAK;AAAA,MACX;AAAA,MACA,SAAS,CAAC;AAAA,IACZ;AAAA,EACF;AAAA,EACA,MAAM,KAAK,YAAY;AAErB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,SAAS,CAAC;AAAA,IACZ;AAAA,EACF;AAAA,EACA,MAAM,SAAS;AAAA,EAEf;AACF;;;AJhBO,IAAM,0BAA0B,CACrC,gBACG;AACH,QAAM,WAAW,yBAAyB,YAAY,QAAQ;AAE9D,QAAM,UAAUC,yBAAwB;AAAA,IACtC,WAAW,YAAY;AAAA,IACvB;AAAA,IACA,aAAa,CAACC,cAAa,YAAY,YAAYA,UAAS,KAAK,CAAC;AAAA,IAClE,UAAU,YAAY,YAAY,KAAK;AAAA,IACvC,OAAO,OAAO,YAAY;AACxB,YAAM,YAAY,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IACzD;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IACzD;AAAA,IACA,UAAU,OAAO,aAA4B;AAC3C,YAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;AACrE,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO;AAAA,IAC3B;AAAA,IACA,iBAAiB,CAAC,EAAE,YAAY,OAAO,MAAM;AAC3C,kBAAY,cAAc,EAAE,YAAY,OAAO,CAAC;AAAA,IAClD;AAAA,IACA,aAAa,MAAM;AACjB,kBAAY,WAAW,CAAC;AACxB,kBAAY,QAAQ;AACpB,kBAAY,YAAY,CAAC,CAAC;AAC1B,kBAAY,SAAS,EAAE;AAAA,IACzB;AAAA,IACA;AAAA,IACA,UAAU;AAAA,MACR,aAAa;AAAA,IACf;AAAA,EACF,CAAC;AAED,eAAa,aAAa,OAAO;AAEjC,SAAO;AACT;;;AKtDA,SAAS,2BAAAC,gCAA+B;AAOjC,IAAM,+BAA+B,CAC1C,qBACG;AACH,QAAM,WAAW,yBAAyB,iBAAiB,QAAQ;AACnE,QAAM,UAAUC,yBAAwB;AAAA,IACtC,WAAW,iBAAiB,WAAW;AAAA,IACvC;AAAA,IACA,UAAU,YAAY,iBAAiB,KAAK;AAAA,IAC5C,OAAO,OAAO,YAAY;AACxB,YAAM,iBAAiB,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IAC9D;AAAA,IACA,aAAa,MAAM;AACjB,uBAAiB,WAAW,CAAC;AAC7B,uBAAiB,QAAQ;AACzB,uBAAiB,YAAY,CAAC,CAAC;AAC/B,uBAAiB,SAAS,EAAE;AAAA,IAC9B;AAAA,IACA;AAAA,IACA,UAAU;AAAA,MACR,aAAa;AAAA,IACf;AAAA,EACF,CAAC;AAED,eAAa,kBAAkB,OAAO;AAEtC,SAAO;AACT;;;AClCA;AAAA,EACE,2BAAAC;AAAA,OAEK;AAGA,IAAM,sBAAsB,CAAC,YAA2B;AAC7D,SAAOA,yBAAwB,OAAO;AACxC;","names":["m","useExternalStoreRuntime","useExternalStoreRuntime","messages","useExternalStoreRuntime","useExternalStoreRuntime","getExternalStoreMessage"]}
+{"version":3,"sources":["../src/rsc/useVercelRSCRuntime.tsx","../src/rsc/getVercelRSCMessage.tsx","../src/ui/utils/convertMessage.ts","../src/ui/use-chat/useVercelUseChatRuntime.tsx","../src/ui/utils/useInputSync.tsx","../src/ui/utils/sliceMessagesUntil.tsx","../src/ui/utils/toCreateMessage.ts","../src/ui/utils/vercelAttachmentAdapter.ts","../src/ui/getVercelAIMessages.tsx","../src/ui/use-assistant/useVercelUseAssistantRuntime.tsx"],"sourcesContent":["\"use client\";\n\nimport type { VercelRSCAdapter } from \"./VercelRSCAdapter\";\nimport {\n  ExternalStoreAdapter,\n  ThreadMessageLike,\n  useExternalStoreRuntime,\n} from \"@assistant-ui/react\";\nimport { VercelRSCMessage } from \"./VercelRSCMessage\";\n\nconst vercelToThreadMessage = <T,>(\n  converter: (message: T) => VercelRSCMessage,\n  rawMessage: T,\n): ThreadMessageLike => {\n  const message = converter(rawMessage);\n\n  return {\n    id: message.id,\n    role: message.role,\n    content: [{ type: \"ui\", display: message.display }],\n    createdAt: message.createdAt,\n  };\n};\n\nexport const useVercelRSCRuntime = <T extends WeakKey>(\n  adapter: VercelRSCAdapter<T>,\n) => {\n  const onNew = adapter.onNew ?? adapter.append;\n  if (!onNew)\n    throw new Error(\"You must pass a onNew function to useVercelRSCRuntime\");\n  const eAdapter: ExternalStoreAdapter<any> = {\n    isRunning: adapter.isRunning,\n    messages: adapter.messages,\n    onNew,\n    onEdit: adapter.onEdit ?? adapter.edit,\n    onReload: adapter.onReload ?? adapter.reload,\n    convertMessage: (m: T) =>\n      vercelToThreadMessage(\n        adapter.convertMessage ?? ((m) => m as VercelRSCMessage),\n        m,\n      ),\n    unstable_capabilities: {\n      copy: false,\n    },\n  };\n\n  const runtime = useExternalStoreRuntime(eAdapter);\n  return runtime;\n};\n","import {\n  getExternalStoreMessage,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\n\nexport const symbolInnerRSCMessage = Symbol(\"innerVercelRSCMessage\");\n\nexport const getVercelRSCMessage = (message: ThreadMessage) => {\n  return getExternalStoreMessage(message);\n};\n","import { Message } from \"ai\";\nimport { useExternalMessageConverter } from \"@assistant-ui/react\";\nimport { ToolCallContentPart } from \"@assistant-ui/react\";\nimport { TextContentPart } from \"@assistant-ui/react\";\nimport { MessageAttachment } from \"../../../../react/src/context/stores/Attachment\";\n\nexport const convertMessage: useExternalMessageConverter.Callback<Message> = (\n  message,\n) => {\n  switch (message.role) {\n    case \"user\":\n      return {\n        role: \"user\",\n        id: message.id,\n        createdAt: message.createdAt,\n        content: message.content,\n\n        attachments: message.experimental_attachments?.map(\n          (attachment, idx) =>\n            ({\n              id: idx.toString(),\n              type: \"file\",\n              name: attachment.name ?? attachment.url,\n              content: [],\n            }) satisfies MessageAttachment,\n        ),\n      };\n\n    case \"system\":\n      return {\n        role: \"system\",\n        id: message.id,\n        createdAt: message.createdAt,\n        content: message.content,\n      };\n\n    case \"assistant\":\n      return {\n        role: \"assistant\",\n        id: message.id,\n        createdAt: message.createdAt,\n        content: [\n          ...(message.content\n            ? [\n                {\n                  type: \"text\",\n                  text: message.content,\n                } satisfies TextContentPart,\n              ]\n            : []),\n          ...(message.toolInvocations?.map(\n            (t) =>\n              ({\n                type: \"tool-call\",\n                toolName: t.toolName,\n                toolCallId: t.toolCallId,\n                argsText: JSON.stringify(t.args),\n                args: t.args,\n                result: \"result\" in t ? t.result : undefined,\n              }) satisfies ToolCallContentPart,\n          ) ?? []),\n        ],\n      };\n\n    case \"data\": {\n      type MaybeSupportedDataMessage =\n        | { type?: \"unsafe_other\" }\n        | ToolCallContentPart\n        | {\n            type: \"tool-result\";\n            toolCallId: string;\n            result: any;\n          };\n\n      if (\n        !message.data ||\n        !(typeof message.data === \"object\") ||\n        Array.isArray(message.data)\n      )\n        return [];\n\n      const data = message.data as MaybeSupportedDataMessage;\n\n      if (data.type === \"tool-call\") {\n        return {\n          role: \"assistant\",\n          id: message.id,\n          createdAt: message.createdAt,\n          content: [data],\n        };\n      } else if (data.type === \"tool-result\") {\n        return {\n          role: \"tool\",\n          toolCallId: data.toolCallId,\n          result: data.result,\n        };\n      }\n      return [];\n    }\n\n    default:\n      // TODO handle tool and function messages\n      const _unsupported: \"function\" | \"tool\" = message.role;\n      throw new Error(\n        `You have a message with an unsupported role. The role ${_unsupported} is not supported.`,\n      );\n  }\n};\n","import type { useChat } from \"ai/react\";\nimport { convertMessage } from \"../utils/convertMessage\";\nimport {\n  useExternalMessageConverter,\n  useExternalStoreRuntime,\n} from \"@assistant-ui/react\";\nimport { useInputSync } from \"../utils/useInputSync\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../getVercelAIMessages\";\n\nexport const useVercelUseChatRuntime = (\n  chatHelpers: ReturnType<typeof useChat>,\n) => {\n  const messages = useExternalMessageConverter({\n    callback: convertMessage,\n    isRunning: chatHelpers.isLoading,\n    messages: chatHelpers.messages,\n  });\n\n  const runtime = useExternalStoreRuntime({\n    isRunning: chatHelpers.isLoading,\n    messages,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(messages.map(getVercelAIMessages).flat()),\n    onCancel: async () => chatHelpers.stop(),\n    onNew: async (message) => {\n      await chatHelpers.append(await toCreateMessage(message));\n    },\n    onEdit: async (message) => {\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.append(await toCreateMessage(message));\n    },\n    onReload: async (parentId: string | null) => {\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.reload();\n    },\n    onAddToolResult: ({ toolCallId, result }) => {\n      chatHelpers.addToolResult({ toolCallId, result });\n    },\n    onNewThread: () => {\n      chatHelpers.messages = [];\n      chatHelpers.input = \"\";\n      chatHelpers.setMessages([]);\n      chatHelpers.setInput(\"\");\n    },\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n    },\n  });\n\n  useInputSync(chatHelpers, runtime);\n\n  return runtime;\n};\n","import { useRef, useEffect } from \"react\";\nimport {\n  ExternalStoreRuntime,\n  subscribeToMainThread,\n} from \"@assistant-ui/react\";\nimport { useAssistant, useChat } from \"ai/react\";\n\ntype VercelHelpers =\n  | ReturnType<typeof useChat>\n  | ReturnType<typeof useAssistant>;\n\nexport const useInputSync = (\n  helpers: VercelHelpers,\n  runtime: ExternalStoreRuntime,\n) => {\n  // sync input from vercel to assistant-ui\n  const helpersRef = useRef(helpers);\n  useEffect(() => {\n    helpersRef.current = helpers;\n    if (runtime.thread.composer.text !== helpers.input) {\n      runtime.thread.composer.setText(helpers.input);\n    }\n  }, [helpers, runtime]);\n\n  // sync input from assistant-ui to vercel\n  useEffect(() => {\n    return subscribeToMainThread(runtime, () => {\n      if (runtime.thread.composer.text !== helpersRef.current.input) {\n        helpersRef.current.setInput(runtime.thread.composer.text);\n      }\n    });\n  }, [runtime]);\n};\n","import type { Message } from \"ai\";\n\nexport const sliceMessagesUntil = (\n  messages: Message[],\n  messageId: string | null,\n) => {\n  if (messageId == null) return [];\n\n  let messageIdx = messages.findIndex((m) => m.id === messageId);\n  if (messageIdx === -1)\n    throw new Error(\n      \"useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui.\",\n    );\n\n  while (messages[messageIdx + 1]?.role === \"assistant\") {\n    messageIdx++;\n  }\n\n  return messages.slice(0, messageIdx + 1);\n};\n","import { AppendMessage } from \"@assistant-ui/react\";\nimport { CreateMessage } from \"ai\";\n\nexport const toCreateMessage = async (\n  message: AppendMessage,\n): Promise<CreateMessage> => {\n  const content = message.content\n    .filter((part) => part.type === \"text\")\n    .map((t) => t.text)\n    .join(\"\\n\\n\");\n\n  const images = message.content\n    .filter((part) => part.type === \"image\")\n    .map((part) => ({ url: part.image }));\n\n  return {\n    role: message.role,\n    content,\n    experimental_attachments: [\n      ...images,\n      ...(await Promise.all(\n        (message.attachments ?? []).map(async (m) => {\n          if (m.file == null)\n            throw new Error(\"Attachment did not contain a file\");\n          return {\n            contentType: m.file.type,\n            name: m.file.name,\n            url: await getFileDataURL(m.file),\n          };\n        }),\n      )),\n    ],\n  };\n};\n\nconst getFileDataURL = (file: File) =>\n  new Promise<string>((resolve, reject) => {\n    const reader = new FileReader();\n\n    reader.onload = () => resolve(reader.result as string);\n    reader.onerror = (error) => reject(error);\n\n    reader.readAsDataURL(file);\n  });\n","import { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { generateId } from \"ai\";\n\nexport const vercelAttachmentAdapter: AttachmentAdapter = {\n  accept:\n    \"image/*, text/plain, text/html, text/markdown, text/csv, text/xml, text/json, text/css\",\n  async add({ file }) {\n    return {\n      id: generateId(),\n      type: \"file\",\n      name: file.name,\n      file,\n      content: [],\n    };\n  },\n  async send(attachment) {\n    // noop\n    return {\n      ...attachment,\n      content: [],\n    };\n  },\n  async remove() {\n    // noop\n  },\n};\n","import {\n  getExternalStoreMessage,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\nimport type { Message } from \"ai\";\n\nexport const getVercelAIMessages = (message: ThreadMessage) => {\n  return getExternalStoreMessage(message) as Message[];\n};\n","import type { useAssistant } from \"ai/react\";\nimport {\n  useExternalMessageConverter,\n  useExternalStoreRuntime,\n} from \"@assistant-ui/react\";\nimport { convertMessage } from \"../utils/convertMessage\";\nimport { useInputSync } from \"../utils/useInputSync\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\n\nexport const useVercelUseAssistantRuntime = (\n  assistantHelpers: ReturnType<typeof useAssistant>,\n) => {\n  const messages = useExternalMessageConverter({\n    callback: convertMessage,\n    isRunning: assistantHelpers.status === \"in_progress\",\n    messages: assistantHelpers.messages,\n  });\n  const runtime = useExternalStoreRuntime({\n    isRunning: assistantHelpers.status === \"in_progress\",\n    messages,\n    onCancel: async () => assistantHelpers.stop(),\n    onNew: async (message) => {\n      await assistantHelpers.append(await toCreateMessage(message));\n    },\n    onNewThread: () => {\n      assistantHelpers.messages = [];\n      assistantHelpers.input = \"\";\n      assistantHelpers.setMessages([]);\n      assistantHelpers.setInput(\"\");\n    },\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n    },\n  });\n\n  useInputSync(assistantHelpers, runtime);\n\n  return runtime;\n};\n"],"mappings":";AAGA;AAAA,EAGE;AAAA,OACK;AAGP,IAAM,wBAAwB,CAC5B,WACA,eACsB;AACtB,QAAM,UAAU,UAAU,UAAU;AAEpC,SAAO;AAAA,IACL,IAAI,QAAQ;AAAA,IACZ,MAAM,QAAQ;AAAA,IACd,SAAS,CAAC,EAAE,MAAM,MAAM,SAAS,QAAQ,QAAQ,CAAC;AAAA,IAClD,WAAW,QAAQ;AAAA,EACrB;AACF;AAEO,IAAM,sBAAsB,CACjC,YACG;AACH,QAAM,QAAQ,QAAQ,SAAS,QAAQ;AACvC,MAAI,CAAC;AACH,UAAM,IAAI,MAAM,uDAAuD;AACzE,QAAM,WAAsC;AAAA,IAC1C,WAAW,QAAQ;AAAA,IACnB,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA,QAAQ,QAAQ,UAAU,QAAQ;AAAA,IAClC,UAAU,QAAQ,YAAY,QAAQ;AAAA,IACtC,gBAAgB,CAAC,MACf;AAAA,MACE,QAAQ,mBAAmB,CAACA,OAAMA;AAAA,MAClC;AAAA,IACF;AAAA,IACF,uBAAuB;AAAA,MACrB,MAAM;AAAA,IACR;AAAA,EACF;AAEA,QAAM,UAAU,wBAAwB,QAAQ;AAChD,SAAO;AACT;;;AChDA;AAAA,EACE;AAAA,OAEK;AAEA,IAAM,wBAAwB,OAAO,uBAAuB;AAE5D,IAAM,sBAAsB,CAAC,YAA2B;AAC7D,SAAO,wBAAwB,OAAO;AACxC;;;ACHO,IAAM,iBAAgE,CAC3E,YACG;AACH,UAAQ,QAAQ,MAAM;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,WAAW,QAAQ;AAAA,QACnB,SAAS,QAAQ;AAAA,QAEjB,aAAa,QAAQ,0BAA0B;AAAA,UAC7C,CAAC,YAAY,SACV;AAAA,YACC,IAAI,IAAI,SAAS;AAAA,YACjB,MAAM;AAAA,YACN,MAAM,WAAW,QAAQ,WAAW;AAAA,YACpC,SAAS,CAAC;AAAA,UACZ;AAAA,QACJ;AAAA,MACF;AAAA,IAEF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,WAAW,QAAQ;AAAA,QACnB,SAAS,QAAQ;AAAA,MACnB;AAAA,IAEF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,WAAW,QAAQ;AAAA,QACnB,SAAS;AAAA,UACP,GAAI,QAAQ,UACR;AAAA,YACE;AAAA,cACE,MAAM;AAAA,cACN,MAAM,QAAQ;AAAA,YAChB;AAAA,UACF,IACA,CAAC;AAAA,UACL,GAAI,QAAQ,iBAAiB;AAAA,YAC3B,CAAC,OACE;AAAA,cACC,MAAM;AAAA,cACN,UAAU,EAAE;AAAA,cACZ,YAAY,EAAE;AAAA,cACd,UAAU,KAAK,UAAU,EAAE,IAAI;AAAA,cAC/B,MAAM,EAAE;AAAA,cACR,QAAQ,YAAY,IAAI,EAAE,SAAS;AAAA,YACrC;AAAA,UACJ,KAAK,CAAC;AAAA,QACR;AAAA,MACF;AAAA,IAEF,KAAK,QAAQ;AAUX,UACE,CAAC,QAAQ,QACT,EAAE,OAAO,QAAQ,SAAS,aAC1B,MAAM,QAAQ,QAAQ,IAAI;AAE1B,eAAO,CAAC;AAEV,YAAM,OAAO,QAAQ;AAErB,UAAI,KAAK,SAAS,aAAa;AAC7B,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ,WAAW,QAAQ;AAAA,UACnB,SAAS,CAAC,IAAI;AAAA,QAChB;AAAA,MACF,WAAW,KAAK,SAAS,eAAe;AACtC,eAAO;AAAA,UACL,MAAM;AAAA,UACN,YAAY,KAAK;AAAA,UACjB,QAAQ,KAAK;AAAA,QACf;AAAA,MACF;AACA,aAAO,CAAC;AAAA,IACV;AAAA,IAEA;AAEE,YAAM,eAAoC,QAAQ;AAClD,YAAM,IAAI;AAAA,QACR,yDAAyD,YAAY;AAAA,MACvE;AAAA,EACJ;AACF;;;ACzGA;AAAA,EACE;AAAA,EACA,2BAAAC;AAAA,OACK;;;ACLP,SAAS,QAAQ,iBAAiB;AAClC;AAAA,EAEE;AAAA,OACK;AAOA,IAAM,eAAe,CAC1B,SACA,YACG;AAEH,QAAM,aAAa,OAAO,OAAO;AACjC,YAAU,MAAM;AACd,eAAW,UAAU;AACrB,QAAI,QAAQ,OAAO,SAAS,SAAS,QAAQ,OAAO;AAClD,cAAQ,OAAO,SAAS,QAAQ,QAAQ,KAAK;AAAA,IAC/C;AAAA,EACF,GAAG,CAAC,SAAS,OAAO,CAAC;AAGrB,YAAU,MAAM;AACd,WAAO,sBAAsB,SAAS,MAAM;AAC1C,UAAI,QAAQ,OAAO,SAAS,SAAS,WAAW,QAAQ,OAAO;AAC7D,mBAAW,QAAQ,SAAS,QAAQ,OAAO,SAAS,IAAI;AAAA,MAC1D;AAAA,IACF,CAAC;AAAA,EACH,GAAG,CAAC,OAAO,CAAC;AACd;;;AC9BO,IAAM,qBAAqB,CAChC,UACA,cACG;AACH,MAAI,aAAa,KAAM,QAAO,CAAC;AAE/B,MAAI,aAAa,SAAS,UAAU,CAAC,MAAM,EAAE,OAAO,SAAS;AAC7D,MAAI,eAAe;AACjB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAEF,SAAO,SAAS,aAAa,CAAC,GAAG,SAAS,aAAa;AACrD;AAAA,EACF;AAEA,SAAO,SAAS,MAAM,GAAG,aAAa,CAAC;AACzC;;;AChBO,IAAM,kBAAkB,OAC7B,YAC2B;AAC3B,QAAM,UAAU,QAAQ,QACrB,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,MAAM,EAAE,IAAI,EACjB,KAAK,MAAM;AAEd,QAAM,SAAS,QAAQ,QACpB,OAAO,CAAC,SAAS,KAAK,SAAS,OAAO,EACtC,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,MAAM,EAAE;AAEtC,SAAO;AAAA,IACL,MAAM,QAAQ;AAAA,IACd;AAAA,IACA,0BAA0B;AAAA,MACxB,GAAG;AAAA,MACH,GAAI,MAAM,QAAQ;AAAA,SACf,QAAQ,eAAe,CAAC,GAAG,IAAI,OAAO,MAAM;AAC3C,cAAI,EAAE,QAAQ;AACZ,kBAAM,IAAI,MAAM,mCAAmC;AACrD,iBAAO;AAAA,YACL,aAAa,EAAE,KAAK;AAAA,YACpB,MAAM,EAAE,KAAK;AAAA,YACb,KAAK,MAAM,eAAe,EAAE,IAAI;AAAA,UAClC;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,iBAAiB,CAAC,SACtB,IAAI,QAAgB,CAAC,SAAS,WAAW;AACvC,QAAM,SAAS,IAAI,WAAW;AAE9B,SAAO,SAAS,MAAM,QAAQ,OAAO,MAAgB;AACrD,SAAO,UAAU,CAAC,UAAU,OAAO,KAAK;AAExC,SAAO,cAAc,IAAI;AAC3B,CAAC;;;AC1CH,SAAS,kBAAkB;AAEpB,IAAM,0BAA6C;AAAA,EACxD,QACE;AAAA,EACF,MAAM,IAAI,EAAE,KAAK,GAAG;AAClB,WAAO;AAAA,MACL,IAAI,WAAW;AAAA,MACf,MAAM;AAAA,MACN,MAAM,KAAK;AAAA,MACX;AAAA,MACA,SAAS,CAAC;AAAA,IACZ;AAAA,EACF;AAAA,EACA,MAAM,KAAK,YAAY;AAErB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,SAAS,CAAC;AAAA,IACZ;AAAA,EACF;AAAA,EACA,MAAM,SAAS;AAAA,EAEf;AACF;;;ACzBA;AAAA,EACE,2BAAAC;AAAA,OAEK;AAGA,IAAM,sBAAsB,CAAC,YAA2B;AAC7D,SAAOA,yBAAwB,OAAO;AACxC;;;ALIO,IAAM,0BAA0B,CACrC,gBACG;AACH,QAAM,WAAW,4BAA4B;AAAA,IAC3C,UAAU;AAAA,IACV,WAAW,YAAY;AAAA,IACvB,UAAU,YAAY;AAAA,EACxB,CAAC;AAED,QAAM,UAAUC,yBAAwB;AAAA,IACtC,WAAW,YAAY;AAAA,IACvB;AAAA,IACA,aAAa,CAACC,cACZ,YAAY,YAAYA,UAAS,IAAI,mBAAmB,EAAE,KAAK,CAAC;AAAA,IAClE,UAAU,YAAY,YAAY,KAAK;AAAA,IACvC,OAAO,OAAO,YAAY;AACxB,YAAM,YAAY,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IACzD;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IACzD;AAAA,IACA,UAAU,OAAO,aAA4B;AAC3C,YAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;AACrE,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO;AAAA,IAC3B;AAAA,IACA,iBAAiB,CAAC,EAAE,YAAY,OAAO,MAAM;AAC3C,kBAAY,cAAc,EAAE,YAAY,OAAO,CAAC;AAAA,IAClD;AAAA,IACA,aAAa,MAAM;AACjB,kBAAY,WAAW,CAAC;AACxB,kBAAY,QAAQ;AACpB,kBAAY,YAAY,CAAC,CAAC;AAC1B,kBAAY,SAAS,EAAE;AAAA,IACzB;AAAA,IACA,UAAU;AAAA,MACR,aAAa;AAAA,IACf;AAAA,EACF,CAAC;AAED,eAAa,aAAa,OAAO;AAEjC,SAAO;AACT;;;AM7DA;AAAA,EACE,+BAAAC;AAAA,EACA,2BAAAC;AAAA,OACK;AAMA,IAAM,+BAA+B,CAC1C,qBACG;AACH,QAAM,WAAWC,6BAA4B;AAAA,IAC3C,UAAU;AAAA,IACV,WAAW,iBAAiB,WAAW;AAAA,IACvC,UAAU,iBAAiB;AAAA,EAC7B,CAAC;AACD,QAAM,UAAUC,yBAAwB;AAAA,IACtC,WAAW,iBAAiB,WAAW;AAAA,IACvC;AAAA,IACA,UAAU,YAAY,iBAAiB,KAAK;AAAA,IAC5C,OAAO,OAAO,YAAY;AACxB,YAAM,iBAAiB,OAAO,MAAM,gBAAgB,OAAO,CAAC;AAAA,IAC9D;AAAA,IACA,aAAa,MAAM;AACjB,uBAAiB,WAAW,CAAC;AAC7B,uBAAiB,QAAQ;AACzB,uBAAiB,YAAY,CAAC,CAAC;AAC/B,uBAAiB,SAAS,EAAE;AAAA,IAC9B;AAAA,IACA,UAAU;AAAA,MACR,aAAa;AAAA,IACf;AAAA,EACF,CAAC;AAED,eAAa,kBAAkB,OAAO;AAEtC,SAAO;AACT;","names":["m","useExternalStoreRuntime","getExternalStoreMessage","useExternalStoreRuntime","messages","useExternalMessageConverter","useExternalStoreRuntime","useExternalMessageConverter","useExternalStoreRuntime"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ai-sdk",
-  "version": "0.6.4",
+  "version": "0.6.6",
   "license": "MIT",
   "exports": {
     ".": {
@@ -28,7 +28,7 @@
     "zustand": "^4.5.5"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.5.50",
+    "@assistant-ui/react": "^0.5.54",
     "@types/react": "*",
     "ai": "^3.2.x",
     "react": "^18"
@@ -39,7 +39,7 @@
     }
   },
   "devDependencies": {
-    "eslint-config-next": "14.2.9",
+    "eslint-config-next": "14.2.11",
     "tsup": "8.2.4",
     "@assistant-ui/tsconfig": "0.0.0"
   },