@assistant-ui/react-ai-sdk 0.6.1 → 0.6.2

package/dist/index.js

@@ -49,11 +49,13 @@ var useVercelRSCRuntime = (adapter) => {
     onNew,
     onEdit: adapter.onEdit ?? adapter.edit,
     onReload: adapter.onReload ?? adapter.reload,
-    onCopy: null,
     convertMessage: (m) => vercelToThreadMessage(
       adapter.convertMessage ?? ((m2) => m2),
       m
-    )
+    ),
+    unstable_capabilities: {
+      copy: false
+    }
   };
   const runtime = (0, import_react.useExternalStoreRuntime)(eAdapter);
   return runtime;
@@ -255,7 +257,6 @@ var useVercelUseChatRuntime = (chatHelpers) => {
     onAddToolResult: ({ toolCallId, result }) => {
       chatHelpers.addToolResult({ toolCallId, result });
     },
-    // onCopy // TODO
     onNewThread: () => {
       chatHelpers.messages = [];
       chatHelpers.input = "";

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/use-assistant/useVercelUseAssistantRuntime.tsx","../src/ui/getVercelAIMessage.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    onCopy: null,\n    convertMessage: (m: T) =>\n      vercelToThreadMessage(\n        adapter.convertMessage ?? ((m) => m as VercelRSCMessage),\n        m,\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 = {\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\";\n\nexport const useVercelUseChatRuntime = (\n  chatHelpers: ReturnType<typeof useChat>,\n) => {\n  const messages = useCachedChunkedMessages(chatHelpers.messages);\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      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"Only text content is supported by VercelUseChatRuntime. Use the Edge runtime for image support.\",\n        );\n      await chatHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\n    },\n    onEdit: async (message) => {\n      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"Only text content is supported by VercelUseChatRuntime. Use the Edge runtime for image support.\",\n        );\n\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\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    // onCopy // TODO\n    onNewThread: () => {\n      chatHelpers.messages = [];\n      chatHelpers.input = \"\";\n      chatHelpers.setMessages([]);\n      chatHelpers.setInput(\"\");\n    },\n    convertMessage,\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 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\";\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      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"VercelUseAssistantRuntime only supports text content.\",\n        );\n\n      await assistantHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\n    },\n    onNewThread: () => {\n      assistantHelpers.messages = [];\n      assistantHelpers.input = \"\";\n      assistantHelpers.setMessages([]);\n      assistantHelpers.setInput(\"\");\n    },\n    convertMessage,\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 getVercelAIMessage = (message: ThreadMessage) => {\n  return getExternalStoreMessage(message) as Message[];\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;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,QAAQ;AAAA,IACR,gBAAgB,CAAC,MACf;AAAA,MACE,QAAQ,mBAAmB,CAACA,OAAMA;AAAA,MAClC;AAAA,IACF;AAAA,EACJ;AAEA,QAAM,cAAU,sCAAwB,QAAQ;AAChD,SAAO;AACT;;;AC9CA,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,MAAyB;AAAA,QAC7B,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;;;AFZO,IAAM,0BAA0B,CACrC,gBACG;AACH,QAAM,WAAW,yBAAyB,YAAY,QAAQ;AAC9D,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,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AACF,YAAM,YAAY,OAAO;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAEF,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;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;AAAA,IAEA,aAAa,MAAM;AACjB,kBAAY,WAAW,CAAC;AACxB,kBAAY,QAAQ;AACpB,kBAAY,YAAY,CAAC,CAAC;AAC1B,kBAAY,SAAS,EAAE;AAAA,IACzB;AAAA,IACA;AAAA,EACF,CAAC;AAED,eAAa,aAAa,OAAO;AAEjC,SAAO;AACT;;;AGhEA,IAAAC,gBAAwC;AAKjC,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,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAEF,YAAM,iBAAiB,OAAO;AAAA,QAC5B,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;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,EACF,CAAC;AAED,eAAa,kBAAkB,OAAO;AAEtC,SAAO;AACT;;;ACrCA,IAAAC,gBAGO;AAGA,IAAM,qBAAqB,CAAC,YAA2B;AAC5D,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/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/use-assistant/useVercelUseAssistantRuntime.tsx","../src/ui/getVercelAIMessage.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 = {\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\";\n\nexport const useVercelUseChatRuntime = (\n  chatHelpers: ReturnType<typeof useChat>,\n) => {\n  const messages = useCachedChunkedMessages(chatHelpers.messages);\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      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"Only text content is supported by VercelUseChatRuntime. Use the Edge runtime for image support.\",\n        );\n      await chatHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\n    },\n    onEdit: async (message) => {\n      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"Only text content is supported by VercelUseChatRuntime. Use the Edge runtime for image support.\",\n        );\n\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\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  });\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 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\";\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      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"VercelUseAssistantRuntime only supports text content.\",\n        );\n\n      await assistantHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\n    },\n    onNewThread: () => {\n      assistantHelpers.messages = [];\n      assistantHelpers.input = \"\";\n      assistantHelpers.setMessages([]);\n      assistantHelpers.setInput(\"\");\n    },\n    convertMessage,\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 getVercelAIMessage = (message: ThreadMessage) => {\n  return getExternalStoreMessage(message) as Message[];\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;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,MAAyB;AAAA,QAC7B,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;;;AFZO,IAAM,0BAA0B,CACrC,gBACG;AACH,QAAM,WAAW,yBAAyB,YAAY,QAAQ;AAC9D,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,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AACF,YAAM,YAAY,OAAO;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAEF,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;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,EACF,CAAC;AAED,eAAa,aAAa,OAAO;AAEjC,SAAO;AACT;;;AG/DA,IAAAC,gBAAwC;AAKjC,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,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAEF,YAAM,iBAAiB,OAAO;AAAA,QAC5B,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;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,EACF,CAAC;AAED,eAAa,kBAAkB,OAAO;AAEtC,SAAO;AACT;;;ACrCA,IAAAC,gBAGO;AAGA,IAAM,qBAAqB,CAAC,YAA2B;AAC5D,aAAO,uCAAwB,OAAO;AACxC;","names":["m","import_react","import_react","import_react","import_react","messages","import_react","import_react"]}

package/dist/index.mjs

@@ -21,11 +21,13 @@ var useVercelRSCRuntime = (adapter) => {
     onNew,
     onEdit: adapter.onEdit ?? adapter.edit,
     onReload: adapter.onReload ?? adapter.reload,
-    onCopy: null,
     convertMessage: (m) => vercelToThreadMessage(
       adapter.convertMessage ?? ((m2) => m2),
       m
-    )
+    ),
+    unstable_capabilities: {
+      copy: false
+    }
   };
   const runtime = useExternalStoreRuntime(eAdapter);
   return runtime;
@@ -231,7 +233,6 @@ var useVercelUseChatRuntime = (chatHelpers) => {
     onAddToolResult: ({ toolCallId, result }) => {
       chatHelpers.addToolResult({ toolCallId, result });
     },
-    // onCopy // TODO
     onNewThread: () => {
       chatHelpers.messages = [];
       chatHelpers.input = "";

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/use-assistant/useVercelUseAssistantRuntime.tsx","../src/ui/getVercelAIMessage.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    onCopy: null,\n    convertMessage: (m: T) =>\n      vercelToThreadMessage(\n        adapter.convertMessage ?? ((m) => m as VercelRSCMessage),\n        m,\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 = {\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\";\n\nexport const useVercelUseChatRuntime = (\n  chatHelpers: ReturnType<typeof useChat>,\n) => {\n  const messages = useCachedChunkedMessages(chatHelpers.messages);\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      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"Only text content is supported by VercelUseChatRuntime. Use the Edge runtime for image support.\",\n        );\n      await chatHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\n    },\n    onEdit: async (message) => {\n      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"Only text content is supported by VercelUseChatRuntime. Use the Edge runtime for image support.\",\n        );\n\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\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    // onCopy // TODO\n    onNewThread: () => {\n      chatHelpers.messages = [];\n      chatHelpers.input = \"\";\n      chatHelpers.setMessages([]);\n      chatHelpers.setInput(\"\");\n    },\n    convertMessage,\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 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\";\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      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"VercelUseAssistantRuntime only supports text content.\",\n        );\n\n      await assistantHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\n    },\n    onNewThread: () => {\n      assistantHelpers.messages = [];\n      assistantHelpers.input = \"\";\n      assistantHelpers.setMessages([]);\n      assistantHelpers.setInput(\"\");\n    },\n    convertMessage,\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 getVercelAIMessage = (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,QAAQ;AAAA,IACR,gBAAgB,CAAC,MACf;AAAA,MACE,QAAQ,mBAAmB,CAACA,OAAMA;AAAA,MAClC;AAAA,IACF;AAAA,EACJ;AAEA,QAAM,UAAU,wBAAwB,QAAQ;AAChD,SAAO;AACT;;;AC9CA;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,MAAyB;AAAA,QAC7B,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;;;AFZO,IAAM,0BAA0B,CACrC,gBACG;AACH,QAAM,WAAW,yBAAyB,YAAY,QAAQ;AAC9D,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,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AACF,YAAM,YAAY,OAAO;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAEF,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;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;AAAA,IAEA,aAAa,MAAM;AACjB,kBAAY,WAAW,CAAC;AACxB,kBAAY,QAAQ;AACpB,kBAAY,YAAY,CAAC,CAAC;AAC1B,kBAAY,SAAS,EAAE;AAAA,IACzB;AAAA,IACA;AAAA,EACF,CAAC;AAED,eAAa,aAAa,OAAO;AAEjC,SAAO;AACT;;;AGhEA,SAAS,2BAAAC,gCAA+B;AAKjC,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,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAEF,YAAM,iBAAiB,OAAO;AAAA,QAC5B,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;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,EACF,CAAC;AAED,eAAa,kBAAkB,OAAO;AAEtC,SAAO;AACT;;;ACrCA;AAAA,EACE,2BAAAC;AAAA,OAEK;AAGA,IAAM,qBAAqB,CAAC,YAA2B;AAC5D,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/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/use-assistant/useVercelUseAssistantRuntime.tsx","../src/ui/getVercelAIMessage.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 = {\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\";\n\nexport const useVercelUseChatRuntime = (\n  chatHelpers: ReturnType<typeof useChat>,\n) => {\n  const messages = useCachedChunkedMessages(chatHelpers.messages);\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      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"Only text content is supported by VercelUseChatRuntime. Use the Edge runtime for image support.\",\n        );\n      await chatHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\n    },\n    onEdit: async (message) => {\n      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"Only text content is supported by VercelUseChatRuntime. Use the Edge runtime for image support.\",\n        );\n\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\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  });\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 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\";\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      if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n        throw new Error(\n          \"VercelUseAssistantRuntime only supports text content.\",\n        );\n\n      await assistantHelpers.append({\n        role: message.role,\n        content: message.content[0].text,\n      });\n    },\n    onNewThread: () => {\n      assistantHelpers.messages = [];\n      assistantHelpers.input = \"\";\n      assistantHelpers.setMessages([]);\n      assistantHelpers.setInput(\"\");\n    },\n    convertMessage,\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 getVercelAIMessage = (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,MAAyB;AAAA,QAC7B,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;;;AFZO,IAAM,0BAA0B,CACrC,gBACG;AACH,QAAM,WAAW,yBAAyB,YAAY,QAAQ;AAC9D,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,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AACF,YAAM,YAAY,OAAO;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAEF,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,OAAO;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;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,EACF,CAAC;AAED,eAAa,aAAa,OAAO;AAEjC,SAAO;AACT;;;AG/DA,SAAS,2BAAAC,gCAA+B;AAKjC,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,UAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAEF,YAAM,iBAAiB,OAAO;AAAA,QAC5B,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,MAC9B,CAAC;AAAA,IACH;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,EACF,CAAC;AAED,eAAa,kBAAkB,OAAO;AAEtC,SAAO;AACT;;;ACrCA;AAAA,EACE,2BAAAC;AAAA,OAEK;AAGA,IAAM,qBAAqB,CAAC,YAA2B;AAC5D,SAAOA,yBAAwB,OAAO;AACxC;","names":["m","useExternalStoreRuntime","useExternalStoreRuntime","messages","useExternalStoreRuntime","useExternalStoreRuntime","getExternalStoreMessage"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ai-sdk",
-  "version": "0.6.1",
+  "version": "0.6.2",
   "license": "MIT",
   "exports": {
     ".": {
@@ -28,7 +28,7 @@
     "zustand": "^4.5.5"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.5.38",
+    "@assistant-ui/react": "^0.5.41",
     "@types/react": "*",
     "ai": "^3.2.x",
     "react": "^18"
@@ -39,7 +39,7 @@
     }
   },
   "devDependencies": {
-    "eslint-config-next": "14.2.6",
+    "eslint-config-next": "14.2.7",
     "tsup": "8.2.4",
     "@assistant-ui/tsconfig": "0.0.0"
   },