@assistant-ui/react-ai-sdk 1.0.4 → 1.0.6

package/dist/ui/use-chat/useAISDKRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAISDKRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.tsx"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAEL,oBAAoB,EACpB,oBAAoB,EACpB,gBAAgB,EAGjB,MAAM,qBAAqB,CAAC;AAa7B,MAAM,MAAM,mBAAmB,GAAG;IAChC,QAAQ,CAAC,EACL,CAAC,WAAW,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC,GAAG;QAC/C,OAAO,CAAC,EAAE,oBAAoB,GAAG,SAAS,CAAC;KAC5C,CAAC,GACF,SAAS,CAAC;CACf,CAAC;AAEF,eAAO,MAAM,eAAe,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EACtE,aAAa,UAAU,CAAC,OAAO,OAAO,CAAC,UAAU,CAAC,CAAC,EACnD,UAAS,mBAAwB,qBA+ElC,CAAC"}
+{"version":3,"file":"useAISDKRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.tsx"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAEL,oBAAoB,EACpB,oBAAoB,EACpB,gBAAgB,EAIjB,MAAM,qBAAqB,CAAC;AAa7B,MAAM,MAAM,mBAAmB,GAAG;IAChC,QAAQ,CAAC,EACL,CAAC,WAAW,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC,GAAG;QAC/C,OAAO,CAAC,EAAE,oBAAoB,GAAG,SAAS,CAAC;KAC5C,CAAC,GACF,SAAS,CAAC;CACf,CAAC;AAEF,eAAO,MAAM,eAAe,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EACtE,aAAa,UAAU,CAAC,OAAO,OAAO,CAAC,UAAU,CAAC,CAAC,EACnD,UAAS,mBAAwB,qBAiFlC,CAAC"}

package/dist/ui/use-chat/useAISDKRuntime.js

@@ -2,7 +2,8 @@
 
 // src/ui/use-chat/useAISDKRuntime.tsx
 import {
-  useExternalStoreRuntime
+  useExternalStoreRuntime,
+  useRuntimeAdapters
 } from "@assistant-ui/react";
 import { sliceMessagesUntil } from "../utils/sliceMessagesUntil.js";
 import { toCreateMessage } from "../utils/toCreateMessage.js";
@@ -15,6 +16,7 @@ import {
 import { useExternalHistory } from "./useExternalHistory.js";
 import { useMemo } from "react";
 var useAISDKRuntime = (chatHelpers, adapter = {}) => {
+  const contextAdapters = useRuntimeAdapters();
   const messages = AISDKMessageConverter.useThreadMessages({
     isRunning: chatHelpers.status === "submitted" || chatHelpers.status == "streaming",
     messages: chatHelpers.messages
@@ -44,7 +46,7 @@ var useAISDKRuntime = (chatHelpers, adapter = {}) => {
     ),
     onCancel: async () => chatHelpers.stop(),
     onNew: async (message) => {
-      const createMessage = await toCreateMessage(message);
+      const createMessage = toCreateMessage(message);
       await chatHelpers.sendMessage(createMessage, {
         metadata: message.runConfig
       });
@@ -55,7 +57,7 @@ var useAISDKRuntime = (chatHelpers, adapter = {}) => {
         message.parentId
       );
       chatHelpers.setMessages(newMessages);
-      const createMessage = await toCreateMessage(message);
+      const createMessage = toCreateMessage(message);
       await chatHelpers.sendMessage(createMessage, {
         metadata: message.runConfig
       });
@@ -74,6 +76,7 @@ var useAISDKRuntime = (chatHelpers, adapter = {}) => {
     },
     adapters: {
       attachments: vercelAttachmentAdapter,
+      ...contextAdapters,
       ...adapter.adapters
     },
     isLoading

package/dist/ui/use-chat/useAISDKRuntime.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/ui/use-chat/useAISDKRuntime.tsx"],"sourcesContent":["\"use client\";\n\nimport type { UIMessage, useChat } from \"@ai-sdk/react\";\nimport {\n  useExternalStoreRuntime,\n  ExternalStoreAdapter,\n  ThreadHistoryAdapter,\n  AssistantRuntime,\n  ThreadMessage,\n  MessageFormatAdapter,\n} from \"@assistant-ui/react\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../getVercelAIMessages\";\nimport { AISDKMessageConverter } from \"../utils/convertMessage\";\nimport {\n  AISDKStorageFormat,\n  aiSDKV5FormatAdapter,\n} from \"../adapters/aiSDKFormatAdapter\";\nimport { useExternalHistory } from \"./useExternalHistory\";\nimport { useMemo } from \"react\";\n\nexport type AISDKRuntimeAdapter = {\n  adapters?:\n    | (NonNullable<ExternalStoreAdapter[\"adapters\"]> & {\n        history?: ThreadHistoryAdapter | undefined;\n      })\n    | undefined;\n};\n\nexport const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>,\n  adapter: AISDKRuntimeAdapter = {},\n) => {\n  const messages = AISDKMessageConverter.useThreadMessages({\n    isRunning:\n      chatHelpers.status === \"submitted\" || chatHelpers.status == \"streaming\",\n    messages: chatHelpers.messages,\n  });\n\n  const isLoading = useExternalHistory(\n    useMemo(\n      () => ({\n        get current(): AssistantRuntime {\n          return runtime;\n        },\n      }),\n      // eslint-disable-next-line react-hooks/exhaustive-deps\n      [],\n    ),\n    adapter.adapters?.history,\n    AISDKMessageConverter.toThreadMessages as (\n      messages: UI_MESSAGE[],\n    ) => ThreadMessage[],\n    aiSDKV5FormatAdapter as MessageFormatAdapter<\n      UI_MESSAGE,\n      AISDKStorageFormat\n    >,\n    (messages) => {\n      chatHelpers.setMessages(messages);\n    },\n  );\n\n  const runtime = useExternalStoreRuntime({\n    isRunning:\n      chatHelpers.status === \"submitted\" || chatHelpers.status === \"streaming\",\n    messages,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(\n        messages.map(getVercelAIMessages<UI_MESSAGE>).flat(),\n      ),\n    onCancel: async () => chatHelpers.stop(),\n    onNew: async (message) => {\n      const createMessage = await toCreateMessage<UI_MESSAGE>(message);\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onEdit: async (message) => {\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      const createMessage = await toCreateMessage<UI_MESSAGE>(message);\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onReload: async (parentId: string | null, config) => {\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.regenerate({ metadata: config.runConfig });\n    },\n    onAddToolResult: ({ toolCallId, result }) => {\n      chatHelpers.addToolResult({\n        tool: toolCallId,\n        toolCallId,\n        output: result,\n      });\n    },\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n      ...adapter.adapters,\n    },\n    isLoading,\n  });\n\n  return runtime;\n};\n"],"mappings":";;;AAGA;AAAA,EACE;AAAA,OAMK;AACP,SAAS,0BAA0B;AACnC,SAAS,uBAAuB;AAChC,SAAS,+BAA+B;AACxC,SAAS,2BAA2B;AACpC,SAAS,6BAA6B;AACtC;AAAA,EAEE;AAAA,OACK;AACP,SAAS,0BAA0B;AACnC,SAAS,eAAe;AAUjB,IAAM,kBAAkB,CAC7B,aACA,UAA+B,CAAC,MAC7B;AACH,QAAM,WAAW,sBAAsB,kBAAkB;AAAA,IACvD,WACE,YAAY,WAAW,eAAe,YAAY,UAAU;AAAA,IAC9D,UAAU,YAAY;AAAA,EACxB,CAAC;AAED,QAAM,YAAY;AAAA,IAChB;AAAA,MACE,OAAO;AAAA,QACL,IAAI,UAA4B;AAC9B,iBAAO;AAAA,QACT;AAAA,MACF;AAAA;AAAA,MAEA,CAAC;AAAA,IACH;AAAA,IACA,QAAQ,UAAU;AAAA,IAClB,sBAAsB;AAAA,IAGtB;AAAA,IAIA,CAACA,cAAa;AACZ,kBAAY,YAAYA,SAAQ;AAAA,IAClC;AAAA,EACF;AAEA,QAAM,UAAU,wBAAwB;AAAA,IACtC,WACE,YAAY,WAAW,eAAe,YAAY,WAAW;AAAA,IAC/D;AAAA,IACA,aAAa,CAACA,cACZ,YAAY;AAAA,MACVA,UAAS,IAAI,mBAA+B,EAAE,KAAK;AAAA,IACrD;AAAA,IACF,UAAU,YAAY,YAAY,KAAK;AAAA,IACvC,OAAO,OAAO,YAAY;AACxB,YAAM,gBAAgB,MAAM,gBAA4B,OAAO;AAC/D,YAAM,YAAY,YAAY,eAAe;AAAA,QAC3C,UAAU,QAAQ;AAAA,MACpB,CAAC;AAAA,IACH;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,gBAAgB,MAAM,gBAA4B,OAAO;AAC/D,YAAM,YAAY,YAAY,eAAe;AAAA,QAC3C,UAAU,QAAQ;AAAA,MACpB,CAAC;AAAA,IACH;AAAA,IACA,UAAU,OAAO,UAAyB,WAAW;AACnD,YAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;AACrE,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,WAAW,EAAE,UAAU,OAAO,UAAU,CAAC;AAAA,IAC7D;AAAA,IACA,iBAAiB,CAAC,EAAE,YAAY,OAAO,MAAM;AAC3C,kBAAY,cAAc;AAAA,QACxB,MAAM;AAAA,QACN;AAAA,QACA,QAAQ;AAAA,MACV,CAAC;AAAA,IACH;AAAA,IACA,UAAU;AAAA,MACR,aAAa;AAAA,MACb,GAAG,QAAQ;AAAA,IACb;AAAA,IACA;AAAA,EACF,CAAC;AAED,SAAO;AACT;","names":["messages"]}
+{"version":3,"sources":["../../../src/ui/use-chat/useAISDKRuntime.tsx"],"sourcesContent":["\"use client\";\n\nimport type { UIMessage, useChat } from \"@ai-sdk/react\";\nimport {\n  useExternalStoreRuntime,\n  ExternalStoreAdapter,\n  ThreadHistoryAdapter,\n  AssistantRuntime,\n  ThreadMessage,\n  MessageFormatAdapter,\n  useRuntimeAdapters,\n} from \"@assistant-ui/react\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../getVercelAIMessages\";\nimport { AISDKMessageConverter } from \"../utils/convertMessage\";\nimport {\n  AISDKStorageFormat,\n  aiSDKV5FormatAdapter,\n} from \"../adapters/aiSDKFormatAdapter\";\nimport { useExternalHistory } from \"./useExternalHistory\";\nimport { useMemo } from \"react\";\n\nexport type AISDKRuntimeAdapter = {\n  adapters?:\n    | (NonNullable<ExternalStoreAdapter[\"adapters\"]> & {\n        history?: ThreadHistoryAdapter | undefined;\n      })\n    | undefined;\n};\n\nexport const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>,\n  adapter: AISDKRuntimeAdapter = {},\n) => {\n  const contextAdapters = useRuntimeAdapters();\n  const messages = AISDKMessageConverter.useThreadMessages({\n    isRunning:\n      chatHelpers.status === \"submitted\" || chatHelpers.status == \"streaming\",\n    messages: chatHelpers.messages,\n  });\n\n  const isLoading = useExternalHistory(\n    useMemo(\n      () => ({\n        get current(): AssistantRuntime {\n          return runtime;\n        },\n      }),\n      // eslint-disable-next-line react-hooks/exhaustive-deps\n      [],\n    ),\n    adapter.adapters?.history,\n    AISDKMessageConverter.toThreadMessages as (\n      messages: UI_MESSAGE[],\n    ) => ThreadMessage[],\n    aiSDKV5FormatAdapter as MessageFormatAdapter<\n      UI_MESSAGE,\n      AISDKStorageFormat\n    >,\n    (messages) => {\n      chatHelpers.setMessages(messages);\n    },\n  );\n\n  const runtime = useExternalStoreRuntime({\n    isRunning:\n      chatHelpers.status === \"submitted\" || chatHelpers.status === \"streaming\",\n    messages,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(\n        messages.map(getVercelAIMessages<UI_MESSAGE>).flat(),\n      ),\n    onCancel: async () => chatHelpers.stop(),\n    onNew: async (message) => {\n      const createMessage = toCreateMessage<UI_MESSAGE>(message);\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onEdit: async (message) => {\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      const createMessage = toCreateMessage<UI_MESSAGE>(message);\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onReload: async (parentId: string | null, config) => {\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.regenerate({ metadata: config.runConfig });\n    },\n    onAddToolResult: ({ toolCallId, result }) => {\n      chatHelpers.addToolResult({\n        tool: toolCallId,\n        toolCallId,\n        output: result,\n      });\n    },\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n      ...contextAdapters,\n      ...adapter.adapters,\n    },\n    isLoading,\n  });\n\n  return runtime;\n};\n"],"mappings":";;;AAGA;AAAA,EACE;AAAA,EAMA;AAAA,OACK;AACP,SAAS,0BAA0B;AACnC,SAAS,uBAAuB;AAChC,SAAS,+BAA+B;AACxC,SAAS,2BAA2B;AACpC,SAAS,6BAA6B;AACtC;AAAA,EAEE;AAAA,OACK;AACP,SAAS,0BAA0B;AACnC,SAAS,eAAe;AAUjB,IAAM,kBAAkB,CAC7B,aACA,UAA+B,CAAC,MAC7B;AACH,QAAM,kBAAkB,mBAAmB;AAC3C,QAAM,WAAW,sBAAsB,kBAAkB;AAAA,IACvD,WACE,YAAY,WAAW,eAAe,YAAY,UAAU;AAAA,IAC9D,UAAU,YAAY;AAAA,EACxB,CAAC;AAED,QAAM,YAAY;AAAA,IAChB;AAAA,MACE,OAAO;AAAA,QACL,IAAI,UAA4B;AAC9B,iBAAO;AAAA,QACT;AAAA,MACF;AAAA;AAAA,MAEA,CAAC;AAAA,IACH;AAAA,IACA,QAAQ,UAAU;AAAA,IAClB,sBAAsB;AAAA,IAGtB;AAAA,IAIA,CAACA,cAAa;AACZ,kBAAY,YAAYA,SAAQ;AAAA,IAClC;AAAA,EACF;AAEA,QAAM,UAAU,wBAAwB;AAAA,IACtC,WACE,YAAY,WAAW,eAAe,YAAY,WAAW;AAAA,IAC/D;AAAA,IACA,aAAa,CAACA,cACZ,YAAY;AAAA,MACVA,UAAS,IAAI,mBAA+B,EAAE,KAAK;AAAA,IACrD;AAAA,IACF,UAAU,YAAY,YAAY,KAAK;AAAA,IACvC,OAAO,OAAO,YAAY;AACxB,YAAM,gBAAgB,gBAA4B,OAAO;AACzD,YAAM,YAAY,YAAY,eAAe;AAAA,QAC3C,UAAU,QAAQ;AAAA,MACpB,CAAC;AAAA,IACH;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,gBAAgB,gBAA4B,OAAO;AACzD,YAAM,YAAY,YAAY,eAAe;AAAA,QAC3C,UAAU,QAAQ;AAAA,MACpB,CAAC;AAAA,IACH;AAAA,IACA,UAAU,OAAO,UAAyB,WAAW;AACnD,YAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;AACrE,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,WAAW,EAAE,UAAU,OAAO,UAAU,CAAC;AAAA,IAC7D;AAAA,IACA,iBAAiB,CAAC,EAAE,YAAY,OAAO,MAAM;AAC3C,kBAAY,cAAc;AAAA,QACxB,MAAM;AAAA,QACN;AAAA,QACA,QAAQ;AAAA,MACV,CAAC;AAAA,IACH;AAAA,IACA,UAAU;AAAA,MACR,aAAa;AAAA,MACb,GAAG;AAAA,MACH,GAAG,QAAQ;AAAA,IACb;AAAA,IACA;AAAA,EACF,CAAC;AAED,SAAO;AACT;","names":["messages"]}

package/dist/ui/use-chat/useChatRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useChatRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useChatRuntime.tsx"],"names":[],"mappings":"AAEA,OAAO,EAAW,KAAK,SAAS,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EACL,gBAAgB,EAIjB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAmB,KAAK,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAC9E,OAAO,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;AAG9B,MAAM,MAAM,qBAAqB,CAAC,UAAU,SAAS,SAAS,GAAG,SAAS,IACxE,QAAQ,CAAC,UAAU,CAAC,GAAG;IACrB,KAAK,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACnC,QAAQ,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,GAAG,SAAS,CAAC;CACxD,CAAC;AAEJ,eAAO,MAAM,oBAAoB,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EAC3E,UAAU,qBAAqB,CAAC,UAAU,CAAC,KAC1C,gBAsDF,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EAAE,wBAGtE,qBAAqB,CAAC,UAAU,CAAM,KAAG,gBAQ3C,CAAC"}
+{"version":3,"file":"useChatRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useChatRuntime.tsx"],"names":[],"mappings":"AAEA,OAAO,EAAW,KAAK,SAAS,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EACL,gBAAgB,EAGjB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAmB,KAAK,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAC9E,OAAO,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;AAG9B,MAAM,MAAM,qBAAqB,CAAC,UAAU,SAAS,SAAS,GAAG,SAAS,IACxE,QAAQ,CAAC,UAAU,CAAC,GAAG;IACrB,KAAK,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACnC,QAAQ,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,GAAG,SAAS,CAAC;CACxD,CAAC;AAEJ,eAAO,MAAM,oBAAoB,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EAC3E,UAAU,qBAAqB,CAAC,UAAU,CAAC,KAC1C,gBAgDF,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EAAE,wBAGtE,qBAAqB,CAAC,UAAU,CAAM,KAAG,gBAQ3C,CAAC"}

package/dist/ui/use-chat/useChatRuntime.js

@@ -4,8 +4,7 @@
 import { useChat } from "@ai-sdk/react";
 import {
   unstable_useCloudThreadListAdapter,
-  unstable_useRemoteThreadListRuntime,
-  useRuntimeAdapters
+  unstable_useRemoteThreadListRuntime
 } from "@assistant-ui/react";
 import { useAISDKRuntime } from "./useAISDKRuntime.js";
 import { AssistantChatTransport } from "./AssistantChatTransport.js";
@@ -16,7 +15,6 @@ var useChatThreadRuntime = (options) => {
     ...chatOptions
   } = options ?? {};
   const transport = transportOptions ?? new AssistantChatTransport();
-  const contextAdapters = useRuntimeAdapters();
   const chat = useChat({
     ...chatOptions,
     transport,
@@ -49,10 +47,7 @@ var useChatThreadRuntime = (options) => {
     }
   });
   const runtime = useAISDKRuntime(chat, {
-    adapters: {
-      ...contextAdapters,
-      ...adapters
-    }
+    adapters
   });
   if (transport instanceof AssistantChatTransport) {
     transport.setRuntime(runtime);

package/dist/ui/use-chat/useChatRuntime.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/ui/use-chat/useChatRuntime.tsx"],"sourcesContent":["\"use client\";\n\nimport { useChat, type UIMessage } from \"@ai-sdk/react\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport {\n  AssistantRuntime,\n  unstable_useCloudThreadListAdapter,\n  unstable_useRemoteThreadListRuntime,\n  useRuntimeAdapters,\n} from \"@assistant-ui/react\";\nimport { useAISDKRuntime, type AISDKRuntimeAdapter } from \"./useAISDKRuntime\";\nimport { ChatInit } from \"ai\";\nimport { AssistantChatTransport } from \"./AssistantChatTransport\";\n\nexport type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  ChatInit<UI_MESSAGE> & {\n    cloud?: AssistantCloud | undefined;\n    adapters?: AISDKRuntimeAdapter[\"adapters\"] | undefined;\n  };\n\nexport const useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options?: UseChatRuntimeOptions<UI_MESSAGE>,\n): AssistantRuntime => {\n  const {\n    adapters,\n    transport: transportOptions,\n    ...chatOptions\n  } = options ?? {};\n  const transport = transportOptions ?? new AssistantChatTransport();\n\n  // Get adapters from context (including history adapter from cloud)\n  const contextAdapters = useRuntimeAdapters();\n\n  const chat = useChat({\n    ...chatOptions,\n    transport,\n    onToolCall: async ({ toolCall }) => {\n      await chatOptions.onToolCall?.({ toolCall });\n\n      const tools = runtime.thread.getModelContext().tools;\n      const tool = tools?.[toolCall.toolName];\n      if (tool) {\n        try {\n          const result = await tool.execute?.(toolCall.input, {\n            toolCallId: toolCall.toolCallId,\n            abortSignal: new AbortController().signal, // dummy signal for now\n          });\n          chat.addToolResult({\n            tool: toolCall.toolName,\n            toolCallId: toolCall.toolCallId,\n            output: result,\n          });\n        } catch (error) {\n          chat.addToolResult({\n            tool: toolCall.toolName,\n            toolCallId: toolCall.toolCallId,\n            output: {\n              error: error instanceof Error ? error.message : String(error),\n            },\n          });\n        }\n      }\n    },\n  });\n\n  const runtime = useAISDKRuntime(chat as any, {\n    adapters: {\n      ...contextAdapters,\n      ...adapters,\n    },\n  });\n  if (transport instanceof AssistantChatTransport) {\n    transport.setRuntime(runtime);\n  }\n\n  return runtime;\n};\n\nexport const useChatRuntime = <UI_MESSAGE extends UIMessage = UIMessage>({\n  cloud,\n  ...options\n}: UseChatRuntimeOptions<UI_MESSAGE> = {}): AssistantRuntime => {\n  const cloudAdapter = unstable_useCloudThreadListAdapter({ cloud });\n  return unstable_useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useChatThreadRuntime(options);\n    },\n    adapter: cloudAdapter,\n  });\n};\n"],"mappings":";;;AAEA,SAAS,eAA+B;AAExC;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,uBAAiD;AAE1D,SAAS,8BAA8B;AAQhC,IAAM,uBAAuB,CAClC,YACqB;AACrB,QAAM;AAAA,IACJ;AAAA,IACA,WAAW;AAAA,IACX,GAAG;AAAA,EACL,IAAI,WAAW,CAAC;AAChB,QAAM,YAAY,oBAAoB,IAAI,uBAAuB;AAGjE,QAAM,kBAAkB,mBAAmB;AAE3C,QAAM,OAAO,QAAQ;AAAA,IACnB,GAAG;AAAA,IACH;AAAA,IACA,YAAY,OAAO,EAAE,SAAS,MAAM;AAClC,YAAM,YAAY,aAAa,EAAE,SAAS,CAAC;AAE3C,YAAM,QAAQ,QAAQ,OAAO,gBAAgB,EAAE;AAC/C,YAAM,OAAO,QAAQ,SAAS,QAAQ;AACtC,UAAI,MAAM;AACR,YAAI;AACF,gBAAM,SAAS,MAAM,KAAK,UAAU,SAAS,OAAO;AAAA,YAClD,YAAY,SAAS;AAAA,YACrB,aAAa,IAAI,gBAAgB,EAAE;AAAA;AAAA,UACrC,CAAC;AACD,eAAK,cAAc;AAAA,YACjB,MAAM,SAAS;AAAA,YACf,YAAY,SAAS;AAAA,YACrB,QAAQ;AAAA,UACV,CAAC;AAAA,QACH,SAAS,OAAO;AACd,eAAK,cAAc;AAAA,YACjB,MAAM,SAAS;AAAA,YACf,YAAY,SAAS;AAAA,YACrB,QAAQ;AAAA,cACN,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,YAC9D;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,QAAM,UAAU,gBAAgB,MAAa;AAAA,IAC3C,UAAU;AAAA,MACR,GAAG;AAAA,MACH,GAAG;AAAA,IACL;AAAA,EACF,CAAC;AACD,MAAI,qBAAqB,wBAAwB;AAC/C,cAAU,WAAW,OAAO;AAAA,EAC9B;AAEA,SAAO;AACT;AAEO,IAAM,iBAAiB,CAA2C;AAAA,EACvE;AAAA,EACA,GAAG;AACL,IAAuC,CAAC,MAAwB;AAC9D,QAAM,eAAe,mCAAmC,EAAE,MAAM,CAAC;AACjE,SAAO,oCAAoC;AAAA,IACzC,aAAa,SAAS,cAAc;AAClC,aAAO,qBAAqB,OAAO;AAAA,IACrC;AAAA,IACA,SAAS;AAAA,EACX,CAAC;AACH;","names":[]}
+{"version":3,"sources":["../../../src/ui/use-chat/useChatRuntime.tsx"],"sourcesContent":["\"use client\";\n\nimport { useChat, type UIMessage } from \"@ai-sdk/react\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport {\n  AssistantRuntime,\n  unstable_useCloudThreadListAdapter,\n  unstable_useRemoteThreadListRuntime,\n} from \"@assistant-ui/react\";\nimport { useAISDKRuntime, type AISDKRuntimeAdapter } from \"./useAISDKRuntime\";\nimport { ChatInit } from \"ai\";\nimport { AssistantChatTransport } from \"./AssistantChatTransport\";\n\nexport type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  ChatInit<UI_MESSAGE> & {\n    cloud?: AssistantCloud | undefined;\n    adapters?: AISDKRuntimeAdapter[\"adapters\"] | undefined;\n  };\n\nexport const useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options?: UseChatRuntimeOptions<UI_MESSAGE>,\n): AssistantRuntime => {\n  const {\n    adapters,\n    transport: transportOptions,\n    ...chatOptions\n  } = options ?? {};\n  const transport = transportOptions ?? new AssistantChatTransport();\n\n  const chat = useChat({\n    ...chatOptions,\n    transport,\n    onToolCall: async ({ toolCall }) => {\n      await chatOptions.onToolCall?.({ toolCall });\n\n      const tools = runtime.thread.getModelContext().tools;\n      const tool = tools?.[toolCall.toolName];\n      if (tool) {\n        try {\n          const result = await tool.execute?.(toolCall.input, {\n            toolCallId: toolCall.toolCallId,\n            abortSignal: new AbortController().signal, // dummy signal for now\n          });\n          chat.addToolResult({\n            tool: toolCall.toolName,\n            toolCallId: toolCall.toolCallId,\n            output: result,\n          });\n        } catch (error) {\n          chat.addToolResult({\n            tool: toolCall.toolName,\n            toolCallId: toolCall.toolCallId,\n            output: {\n              error: error instanceof Error ? error.message : String(error),\n            },\n          });\n        }\n      }\n    },\n  });\n\n  const runtime = useAISDKRuntime(chat as any, {\n    adapters,\n  });\n  if (transport instanceof AssistantChatTransport) {\n    transport.setRuntime(runtime);\n  }\n\n  return runtime;\n};\n\nexport const useChatRuntime = <UI_MESSAGE extends UIMessage = UIMessage>({\n  cloud,\n  ...options\n}: UseChatRuntimeOptions<UI_MESSAGE> = {}): AssistantRuntime => {\n  const cloudAdapter = unstable_useCloudThreadListAdapter({ cloud });\n  return unstable_useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useChatThreadRuntime(options);\n    },\n    adapter: cloudAdapter,\n  });\n};\n"],"mappings":";;;AAEA,SAAS,eAA+B;AAExC;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AACP,SAAS,uBAAiD;AAE1D,SAAS,8BAA8B;AAQhC,IAAM,uBAAuB,CAClC,YACqB;AACrB,QAAM;AAAA,IACJ;AAAA,IACA,WAAW;AAAA,IACX,GAAG;AAAA,EACL,IAAI,WAAW,CAAC;AAChB,QAAM,YAAY,oBAAoB,IAAI,uBAAuB;AAEjE,QAAM,OAAO,QAAQ;AAAA,IACnB,GAAG;AAAA,IACH;AAAA,IACA,YAAY,OAAO,EAAE,SAAS,MAAM;AAClC,YAAM,YAAY,aAAa,EAAE,SAAS,CAAC;AAE3C,YAAM,QAAQ,QAAQ,OAAO,gBAAgB,EAAE;AAC/C,YAAM,OAAO,QAAQ,SAAS,QAAQ;AACtC,UAAI,MAAM;AACR,YAAI;AACF,gBAAM,SAAS,MAAM,KAAK,UAAU,SAAS,OAAO;AAAA,YAClD,YAAY,SAAS;AAAA,YACrB,aAAa,IAAI,gBAAgB,EAAE;AAAA;AAAA,UACrC,CAAC;AACD,eAAK,cAAc;AAAA,YACjB,MAAM,SAAS;AAAA,YACf,YAAY,SAAS;AAAA,YACrB,QAAQ;AAAA,UACV,CAAC;AAAA,QACH,SAAS,OAAO;AACd,eAAK,cAAc;AAAA,YACjB,MAAM,SAAS;AAAA,YACf,YAAY,SAAS;AAAA,YACrB,QAAQ;AAAA,cACN,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,YAC9D;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,QAAM,UAAU,gBAAgB,MAAa;AAAA,IAC3C;AAAA,EACF,CAAC;AACD,MAAI,qBAAqB,wBAAwB;AAC/C,cAAU,WAAW,OAAO;AAAA,EAC9B;AAEA,SAAO;AACT;AAEO,IAAM,iBAAiB,CAA2C;AAAA,EACvE;AAAA,EACA,GAAG;AACL,IAAuC,CAAC,MAAwB;AAC9D,QAAM,eAAe,mCAAmC,EAAE,MAAM,CAAC;AACjE,SAAO,oCAAoC;AAAA,IACzC,aAAa,SAAS,cAAc;AAClC,aAAO,qBAAqB,OAAO;AAAA,IACrC;AAAA,IACA,SAAS;AAAA,EACX,CAAC;AACH;","names":[]}

package/dist/ui/use-chat/useExternalHistory.js

@@ -64,7 +64,7 @@ var useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storageF
       for (let i = 0; i < messages.length; i++) {
         const message = messages[i];
         if (message.status === void 0 || message.status.type === "complete" || message.status.type === "incomplete") {
-          if (historyIds.current.has(message.id)) return;
+          if (historyIds.current.has(message.id)) continue;
           historyIds.current.add(message.id);
           const parentId = i > 0 ? messages[i - 1].id : null;
           await historyAdapter?.withFormat?.(storageFormatAdapter).append({

package/dist/ui/use-chat/useExternalHistory.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/ui/use-chat/useExternalHistory.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  AssistantRuntime,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n  MessageFormatAdapter,\n  getExternalStoreMessages,\n  MessageFormatRepository,\n  ExportedMessageRepository,\n  INTERNAL,\n} from \"@assistant-ui/react\";\nimport { useRef, useEffect, useState, RefObject } from \"react\";\n\nconst { MessageRepository } = INTERNAL;\n\nexport const toExportedMessageRepository = <TMessage,>(\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  messages: MessageFormatRepository<TMessage>,\n): ExportedMessageRepository => {\n  return {\n    headId: messages.headId!,\n    messages: messages.messages.map((m) => {\n      const message = toThreadMessages([m.message])[0]!;\n      return {\n        ...m,\n        message,\n      };\n    }),\n  };\n};\n\nexport const useExternalHistory = <TMessage,>(\n  runtimeRef: RefObject<AssistantRuntime>,\n  historyAdapter: ThreadHistoryAdapter | undefined,\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  storageFormatAdapter: MessageFormatAdapter<TMessage, any>,\n  onSetMessages: (messages: TMessage[]) => void,\n) => {\n  const loadedRef = useRef(false);\n  const [isLoading, setIsLoading] = useState(true);\n  const historyIds = useRef(new Set<string>());\n\n  const onSetMessagesRef = useRef<typeof onSetMessages>(() => onSetMessages);\n  useEffect(() => {\n    onSetMessagesRef.current = onSetMessages;\n  });\n\n  // Load messages from history adapter on mount\n  useEffect(() => {\n    if (!historyAdapter || loadedRef.current) return;\n\n    const loadHistory = async () => {\n      setIsLoading(true);\n      try {\n        const repo = await historyAdapter\n          .withFormat?.(storageFormatAdapter)\n          .load();\n        if (repo && repo.messages.length > 0) {\n          const converted = toExportedMessageRepository(toThreadMessages, repo);\n          runtimeRef.current.thread.import(converted);\n\n          const tempRepo = new MessageRepository();\n          tempRepo.import(converted);\n          const messages = tempRepo.getMessages();\n\n          onSetMessagesRef.current(\n            messages.map(getExternalStoreMessages<TMessage>).flat(),\n          );\n\n          historyIds.current = new Set(\n            converted.messages.map((m) => m.message.id),\n          );\n        }\n      } catch (error) {\n        console.error(\"Failed to load message history:\", error);\n      } finally {\n        setIsLoading(false);\n      }\n    };\n\n    if (!loadedRef.current) {\n      loadedRef.current = true;\n      loadHistory();\n    }\n  }, [historyAdapter, storageFormatAdapter, toThreadMessages, runtimeRef]);\n\n  useEffect(() => {\n    return runtimeRef.current.thread.subscribe(async () => {\n      const { messages, isRunning } = runtimeRef.current.thread.getState();\n      if (isRunning) return;\n\n      for (let i = 0; i < messages.length; i++) {\n        const message = messages[i]!;\n        if (\n          message.status === undefined ||\n          message.status.type === \"complete\" ||\n          message.status.type === \"incomplete\"\n        ) {\n          if (historyIds.current.has(message.id)) return;\n          historyIds.current.add(message.id);\n\n          const parentId = i > 0 ? messages[i - 1]!.id : null;\n          await historyAdapter?.withFormat?.(storageFormatAdapter).append({\n            parentId,\n            message: getExternalStoreMessages<TMessage>(message)[0]!,\n          });\n        }\n      }\n    });\n  }, [historyAdapter, storageFormatAdapter, runtimeRef]);\n\n  return isLoading;\n};\n"],"mappings":";;;AAEA;AAAA,EAKE;AAAA,EAGA;AAAA,OACK;AACP,SAAS,QAAQ,WAAW,gBAA2B;AAEvD,IAAM,EAAE,kBAAkB,IAAI;AAEvB,IAAM,8BAA8B,CACzC,kBACA,aAC8B;AAC9B,SAAO;AAAA,IACL,QAAQ,SAAS;AAAA,IACjB,UAAU,SAAS,SAAS,IAAI,CAAC,MAAM;AACrC,YAAM,UAAU,iBAAiB,CAAC,EAAE,OAAO,CAAC,EAAE,CAAC;AAC/C,aAAO;AAAA,QACL,GAAG;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEO,IAAM,qBAAqB,CAChC,YACA,gBACA,kBACA,sBACA,kBACG;AACH,QAAM,YAAY,OAAO,KAAK;AAC9B,QAAM,CAAC,WAAW,YAAY,IAAI,SAAS,IAAI;AAC/C,QAAM,aAAa,OAAO,oBAAI,IAAY,CAAC;AAE3C,QAAM,mBAAmB,OAA6B,MAAM,aAAa;AACzE,YAAU,MAAM;AACd,qBAAiB,UAAU;AAAA,EAC7B,CAAC;AAGD,YAAU,MAAM;AACd,QAAI,CAAC,kBAAkB,UAAU,QAAS;AAE1C,UAAM,cAAc,YAAY;AAC9B,mBAAa,IAAI;AACjB,UAAI;AACF,cAAM,OAAO,MAAM,eAChB,aAAa,oBAAoB,EACjC,KAAK;AACR,YAAI,QAAQ,KAAK,SAAS,SAAS,GAAG;AACpC,gBAAM,YAAY,4BAA4B,kBAAkB,IAAI;AACpE,qBAAW,QAAQ,OAAO,OAAO,SAAS;AAE1C,gBAAM,WAAW,IAAI,kBAAkB;AACvC,mBAAS,OAAO,SAAS;AACzB,gBAAM,WAAW,SAAS,YAAY;AAEtC,2BAAiB;AAAA,YACf,SAAS,IAAI,wBAAkC,EAAE,KAAK;AAAA,UACxD;AAEA,qBAAW,UAAU,IAAI;AAAA,YACvB,UAAU,SAAS,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE;AAAA,UAC5C;AAAA,QACF;AAAA,MACF,SAAS,OAAO;AACd,gBAAQ,MAAM,mCAAmC,KAAK;AAAA,MACxD,UAAE;AACA,qBAAa,KAAK;AAAA,MACpB;AAAA,IACF;AAEA,QAAI,CAAC,UAAU,SAAS;AACtB,gBAAU,UAAU;AACpB,kBAAY;AAAA,IACd;AAAA,EACF,GAAG,CAAC,gBAAgB,sBAAsB,kBAAkB,UAAU,CAAC;AAEvE,YAAU,MAAM;AACd,WAAO,WAAW,QAAQ,OAAO,UAAU,YAAY;AACrD,YAAM,EAAE,UAAU,UAAU,IAAI,WAAW,QAAQ,OAAO,SAAS;AACnE,UAAI,UAAW;AAEf,eAAS,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;AACxC,cAAM,UAAU,SAAS,CAAC;AAC1B,YACE,QAAQ,WAAW,UACnB,QAAQ,OAAO,SAAS,cACxB,QAAQ,OAAO,SAAS,cACxB;AACA,cAAI,WAAW,QAAQ,IAAI,QAAQ,EAAE,EAAG;AACxC,qBAAW,QAAQ,IAAI,QAAQ,EAAE;AAEjC,gBAAM,WAAW,IAAI,IAAI,SAAS,IAAI,CAAC,EAAG,KAAK;AAC/C,gBAAM,gBAAgB,aAAa,oBAAoB,EAAE,OAAO;AAAA,YAC9D;AAAA,YACA,SAAS,yBAAmC,OAAO,EAAE,CAAC;AAAA,UACxD,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH,GAAG,CAAC,gBAAgB,sBAAsB,UAAU,CAAC;AAErD,SAAO;AACT;","names":[]}
+{"version":3,"sources":["../../../src/ui/use-chat/useExternalHistory.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  AssistantRuntime,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n  MessageFormatAdapter,\n  getExternalStoreMessages,\n  MessageFormatRepository,\n  ExportedMessageRepository,\n  INTERNAL,\n} from \"@assistant-ui/react\";\nimport { useRef, useEffect, useState, RefObject } from \"react\";\n\nconst { MessageRepository } = INTERNAL;\n\nexport const toExportedMessageRepository = <TMessage,>(\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  messages: MessageFormatRepository<TMessage>,\n): ExportedMessageRepository => {\n  return {\n    headId: messages.headId!,\n    messages: messages.messages.map((m) => {\n      const message = toThreadMessages([m.message])[0]!;\n      return {\n        ...m,\n        message,\n      };\n    }),\n  };\n};\n\nexport const useExternalHistory = <TMessage,>(\n  runtimeRef: RefObject<AssistantRuntime>,\n  historyAdapter: ThreadHistoryAdapter | undefined,\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  storageFormatAdapter: MessageFormatAdapter<TMessage, any>,\n  onSetMessages: (messages: TMessage[]) => void,\n) => {\n  const loadedRef = useRef(false);\n  const [isLoading, setIsLoading] = useState(true);\n  const historyIds = useRef(new Set<string>());\n\n  const onSetMessagesRef = useRef<typeof onSetMessages>(() => onSetMessages);\n  useEffect(() => {\n    onSetMessagesRef.current = onSetMessages;\n  });\n\n  // Load messages from history adapter on mount\n  useEffect(() => {\n    if (!historyAdapter || loadedRef.current) return;\n\n    const loadHistory = async () => {\n      setIsLoading(true);\n      try {\n        const repo = await historyAdapter\n          .withFormat?.(storageFormatAdapter)\n          .load();\n        if (repo && repo.messages.length > 0) {\n          const converted = toExportedMessageRepository(toThreadMessages, repo);\n          runtimeRef.current.thread.import(converted);\n\n          const tempRepo = new MessageRepository();\n          tempRepo.import(converted);\n          const messages = tempRepo.getMessages();\n\n          onSetMessagesRef.current(\n            messages.map(getExternalStoreMessages<TMessage>).flat(),\n          );\n\n          historyIds.current = new Set(\n            converted.messages.map((m) => m.message.id),\n          );\n        }\n      } catch (error) {\n        console.error(\"Failed to load message history:\", error);\n      } finally {\n        setIsLoading(false);\n      }\n    };\n\n    if (!loadedRef.current) {\n      loadedRef.current = true;\n      loadHistory();\n    }\n  }, [historyAdapter, storageFormatAdapter, toThreadMessages, runtimeRef]);\n\n  useEffect(() => {\n    return runtimeRef.current.thread.subscribe(async () => {\n      const { messages, isRunning } = runtimeRef.current.thread.getState();\n      if (isRunning) return;\n\n      for (let i = 0; i < messages.length; i++) {\n        const message = messages[i]!;\n        if (\n          message.status === undefined ||\n          message.status.type === \"complete\" ||\n          message.status.type === \"incomplete\"\n        ) {\n          if (historyIds.current.has(message.id)) continue;\n          historyIds.current.add(message.id);\n\n          const parentId = i > 0 ? messages[i - 1]!.id : null;\n          await historyAdapter?.withFormat?.(storageFormatAdapter).append({\n            parentId,\n            message: getExternalStoreMessages<TMessage>(message)[0]!,\n          });\n        }\n      }\n    });\n  }, [historyAdapter, storageFormatAdapter, runtimeRef]);\n\n  return isLoading;\n};\n"],"mappings":";;;AAEA;AAAA,EAKE;AAAA,EAGA;AAAA,OACK;AACP,SAAS,QAAQ,WAAW,gBAA2B;AAEvD,IAAM,EAAE,kBAAkB,IAAI;AAEvB,IAAM,8BAA8B,CACzC,kBACA,aAC8B;AAC9B,SAAO;AAAA,IACL,QAAQ,SAAS;AAAA,IACjB,UAAU,SAAS,SAAS,IAAI,CAAC,MAAM;AACrC,YAAM,UAAU,iBAAiB,CAAC,EAAE,OAAO,CAAC,EAAE,CAAC;AAC/C,aAAO;AAAA,QACL,GAAG;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEO,IAAM,qBAAqB,CAChC,YACA,gBACA,kBACA,sBACA,kBACG;AACH,QAAM,YAAY,OAAO,KAAK;AAC9B,QAAM,CAAC,WAAW,YAAY,IAAI,SAAS,IAAI;AAC/C,QAAM,aAAa,OAAO,oBAAI,IAAY,CAAC;AAE3C,QAAM,mBAAmB,OAA6B,MAAM,aAAa;AACzE,YAAU,MAAM;AACd,qBAAiB,UAAU;AAAA,EAC7B,CAAC;AAGD,YAAU,MAAM;AACd,QAAI,CAAC,kBAAkB,UAAU,QAAS;AAE1C,UAAM,cAAc,YAAY;AAC9B,mBAAa,IAAI;AACjB,UAAI;AACF,cAAM,OAAO,MAAM,eAChB,aAAa,oBAAoB,EACjC,KAAK;AACR,YAAI,QAAQ,KAAK,SAAS,SAAS,GAAG;AACpC,gBAAM,YAAY,4BAA4B,kBAAkB,IAAI;AACpE,qBAAW,QAAQ,OAAO,OAAO,SAAS;AAE1C,gBAAM,WAAW,IAAI,kBAAkB;AACvC,mBAAS,OAAO,SAAS;AACzB,gBAAM,WAAW,SAAS,YAAY;AAEtC,2BAAiB;AAAA,YACf,SAAS,IAAI,wBAAkC,EAAE,KAAK;AAAA,UACxD;AAEA,qBAAW,UAAU,IAAI;AAAA,YACvB,UAAU,SAAS,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE;AAAA,UAC5C;AAAA,QACF;AAAA,MACF,SAAS,OAAO;AACd,gBAAQ,MAAM,mCAAmC,KAAK;AAAA,MACxD,UAAE;AACA,qBAAa,KAAK;AAAA,MACpB;AAAA,IACF;AAEA,QAAI,CAAC,UAAU,SAAS;AACtB,gBAAU,UAAU;AACpB,kBAAY;AAAA,IACd;AAAA,EACF,GAAG,CAAC,gBAAgB,sBAAsB,kBAAkB,UAAU,CAAC;AAEvE,YAAU,MAAM;AACd,WAAO,WAAW,QAAQ,OAAO,UAAU,YAAY;AACrD,YAAM,EAAE,UAAU,UAAU,IAAI,WAAW,QAAQ,OAAO,SAAS;AACnE,UAAI,UAAW;AAEf,eAAS,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;AACxC,cAAM,UAAU,SAAS,CAAC;AAC1B,YACE,QAAQ,WAAW,UACnB,QAAQ,OAAO,SAAS,cACxB,QAAQ,OAAO,SAAS,cACxB;AACA,cAAI,WAAW,QAAQ,IAAI,QAAQ,EAAE,EAAG;AACxC,qBAAW,QAAQ,IAAI,QAAQ,EAAE;AAEjC,gBAAM,WAAW,IAAI,IAAI,SAAS,IAAI,CAAC,EAAG,KAAK;AAC/C,gBAAM,gBAAgB,aAAa,oBAAoB,EAAE,OAAO;AAAA,YAC9D;AAAA,YACA,SAAS,yBAAmC,OAAO,EAAE,CAAC;AAAA,UACxD,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH,GAAG,CAAC,gBAAgB,sBAAsB,UAAU,CAAC;AAErD,SAAO;AACT;","names":[]}

package/dist/ui/utils/convertMessage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessage.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/convertMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,SAAS,EAAE,MAAM,IAAI,CAAC;AAwJ7C,eAAO,MAAM,qBAAqB;;;;;;;0HAhIhB,sDACF;yHAIM,sDACT;;;CA+KZ,CAAC"}
+{"version":3,"file":"convertMessage.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/convertMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,SAAS,EAAE,MAAM,IAAI,CAAC;AA8I7C,eAAO,MAAM,qBAAqB;;;;;;;0HAvHb,sDACF;yHAIM,sDACT;;;CAqLf,CAAC"}

package/dist/ui/utils/convertMessage.js

@@ -7,7 +7,7 @@ var convertParts = (message) => {
   if (!message.parts || message.parts.length === 0) {
     return [];
   }
-  return message.parts.filter((p) => p.type !== "step-start").map((part) => {
+  return message.parts.filter((p) => p.type !== "step-start" && p.type !== "file").map((part) => {
     const type = part.type;
     if (type === "text") {
       return {
@@ -88,13 +88,6 @@ var convertParts = (message) => {
       );
       return null;
     }
-    if (type === "file") {
-      return {
-        type: "file",
-        data: part.url,
-        mimeType: part.mediaType
-      };
-    }
     if (type.startsWith("data-")) {
       console.warn(
         `Data part type ${type} is not yet supported in conversion`
@@ -115,14 +108,27 @@ var AISDKMessageConverter = unstable_createMessageConverter(
           id: message.id,
           createdAt,
           content: convertParts(message),
-          attachments: message.parts?.filter((p) => p.type === "file").map((part, idx) => ({
-            id: idx.toString(),
-            type: "file",
-            name: part.name ?? part.url ?? "file",
-            content: [],
-            contentType: part.mediaType ?? part.mimeType ?? "unknown/unknown",
-            status: { type: "complete" }
-          }))
+          attachments: message.parts?.filter((p) => p.type === "file").map((part, idx) => {
+            return {
+              id: idx.toString(),
+              type: part.mediaType.startsWith("image/") ? "image" : "file",
+              name: part.filename ?? "file",
+              content: [
+                part.mediaType.startsWith("image/") ? {
+                  type: "image",
+                  image: part.url,
+                  filename: part.filename
+                } : {
+                  type: "file",
+                  filename: part.filename,
+                  data: part.url,
+                  mimeType: part.mediaType
+                }
+              ],
+              contentType: part.mediaType ?? "unknown/unknown",
+              status: { type: "complete" }
+            };
+          })
         };
       case "system":
         return {

package/dist/ui/utils/convertMessage.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/ui/utils/convertMessage.ts"],"sourcesContent":["import { isToolUIPart, UIMessage } from \"ai\";\nimport {\n  unstable_createMessageConverter,\n  type ReasoningMessagePart,\n  type ToolCallMessagePart,\n  type TextMessagePart,\n  type SourceMessagePart,\n  type FileMessagePart,\n} from \"@assistant-ui/react\";\n\nconst convertParts = (message: UIMessage) => {\n  if (!message.parts || message.parts.length === 0) {\n    return [];\n  }\n\n  return message.parts\n    .filter((p) => p.type !== \"step-start\")\n    .map((part) => {\n      const type = part.type;\n\n      // Handle text parts\n      if (type === \"text\") {\n        return {\n          type: \"text\",\n          text: part.text,\n        } satisfies TextMessagePart;\n      }\n\n      // Handle reasoning parts\n      if (type === \"reasoning\") {\n        return {\n          type: \"reasoning\",\n          text: part.text,\n        } satisfies ReasoningMessagePart;\n      }\n\n      // Handle tool-* parts (AI SDK v5 tool calls)\n      if (isToolUIPart(part)) {\n        const toolName = type.replace(\"tool-\", \"\");\n        const toolCallId = part.toolCallId;\n\n        // Extract args and result based on state\n        let args: any = {};\n        let result: any = undefined;\n        let isError = false;\n\n        if (\n          part.state === \"input-streaming\" ||\n          part.state === \"input-available\"\n        ) {\n          args = part.input || {};\n        } else if (part.state === \"output-available\") {\n          args = part.input || {};\n          result = part.output;\n        } else if (part.state === \"output-error\") {\n          args = part.input || {};\n          isError = true;\n          result = { error: part.errorText };\n        }\n\n        return {\n          type: \"tool-call\",\n          toolName,\n          toolCallId,\n          argsText: JSON.stringify(args),\n          args,\n          result,\n          isError,\n        } satisfies ToolCallMessagePart;\n      }\n\n      // Handle dynamic-tool parts\n      if (type === \"dynamic-tool\") {\n        const toolName = part.toolName;\n        const toolCallId = part.toolCallId;\n\n        // Extract args and result based on state\n        let args: any = {};\n        let result: any = undefined;\n        let isError = false;\n\n        if (\n          part.state === \"input-streaming\" ||\n          part.state === \"input-available\"\n        ) {\n          args = part.input || {};\n        } else if (part.state === \"output-available\") {\n          args = part.input || {};\n          result = part.output;\n        } else if (part.state === \"output-error\") {\n          args = part.input || {};\n          isError = true;\n          result = { error: part.errorText };\n        }\n\n        return {\n          type: \"tool-call\",\n          toolName,\n          toolCallId,\n          argsText: JSON.stringify(args),\n          args,\n          result,\n          isError,\n        } satisfies ToolCallMessagePart;\n      }\n\n      // Handle source-url parts\n      if (type === \"source-url\") {\n        return {\n          type: \"source\",\n          sourceType: \"url\",\n          id: part.sourceId,\n          url: part.url,\n          title: part.title || \"\",\n        } satisfies SourceMessagePart;\n      }\n\n      // Handle source-document parts\n      if (type === \"source-document\") {\n        console.warn(\n          `Source document part type ${type} is not yet supported in conversion`,\n        );\n        return null;\n      }\n\n      // Handle file parts\n      if (type === \"file\") {\n        return {\n          type: \"file\",\n          data: part.url,\n          mimeType: part.mediaType,\n        } satisfies FileMessagePart;\n      }\n\n      // Handle data-* parts (AI SDK v5 data parts)\n      if (type.startsWith(\"data-\")) {\n        // For now, we'll skip data parts as they don't have a direct equivalent\n        // in the assistant-ui types. They could be converted to a custom message part\n        // or handled differently based on the specific use case.\n        console.warn(\n          `Data part type ${type} is not yet supported in conversion`,\n        );\n        return null;\n      }\n\n      // For unsupported types, we'll skip them instead of throwing\n      console.warn(`Unsupported message part type: ${type}`);\n      return null;\n    })\n    .filter(Boolean) as any[];\n};\n\nexport const AISDKMessageConverter = unstable_createMessageConverter(\n  (message: UIMessage) => {\n    // UIMessage doesn't have createdAt, so we'll use current date or undefined\n    const createdAt = new Date();\n    switch (message.role) {\n      case \"user\":\n        return {\n          role: \"user\",\n          id: message.id,\n          createdAt,\n          content: convertParts(message),\n          attachments: message.parts\n            ?.filter((p: any) => p.type === \"file\")\n            .map((part: any, idx) => ({\n              id: idx.toString(),\n              type: \"file\" as const,\n              name: part.name ?? part.url ?? \"file\",\n              content: [],\n              contentType: part.mediaType ?? part.mimeType ?? \"unknown/unknown\",\n              status: { type: \"complete\" as const },\n            })),\n        };\n\n      case \"system\":\n        return {\n          role: \"system\",\n          id: message.id,\n          createdAt,\n          content: convertParts(message),\n        };\n\n      case \"assistant\":\n        return {\n          role: \"assistant\",\n          id: message.id,\n          createdAt,\n          content: convertParts(message),\n          metadata: {\n            unstable_annotations: (message as any).annotations,\n            unstable_data: Array.isArray((message as any).data)\n              ? (message as any).data\n              : (message as any).data\n                ? [(message as any).data]\n                : undefined,\n            custom: {},\n          },\n        };\n\n      default:\n        console.warn(`Unsupported message role: ${message.role}`);\n        return [];\n    }\n  },\n);\n"],"mappings":";AAAA,SAAS,oBAA+B;AACxC;AAAA,EACE;AAAA,OAMK;AAEP,IAAM,eAAe,CAAC,YAAuB;AAC3C,MAAI,CAAC,QAAQ,SAAS,QAAQ,MAAM,WAAW,GAAG;AAChD,WAAO,CAAC;AAAA,EACV;AAEA,SAAO,QAAQ,MACZ,OAAO,CAAC,MAAM,EAAE,SAAS,YAAY,EACrC,IAAI,CAAC,SAAS;AACb,UAAM,OAAO,KAAK;AAGlB,QAAI,SAAS,QAAQ;AACnB,aAAO;AAAA,QACL,MAAM;AAAA,QACN,MAAM,KAAK;AAAA,MACb;AAAA,IACF;AAGA,QAAI,SAAS,aAAa;AACxB,aAAO;AAAA,QACL,MAAM;AAAA,QACN,MAAM,KAAK;AAAA,MACb;AAAA,IACF;AAGA,QAAI,aAAa,IAAI,GAAG;AACtB,YAAM,WAAW,KAAK,QAAQ,SAAS,EAAE;AACzC,YAAM,aAAa,KAAK;AAGxB,UAAI,OAAY,CAAC;AACjB,UAAI,SAAc;AAClB,UAAI,UAAU;AAEd,UACE,KAAK,UAAU,qBACf,KAAK,UAAU,mBACf;AACA,eAAO,KAAK,SAAS,CAAC;AAAA,MACxB,WAAW,KAAK,UAAU,oBAAoB;AAC5C,eAAO,KAAK,SAAS,CAAC;AACtB,iBAAS,KAAK;AAAA,MAChB,WAAW,KAAK,UAAU,gBAAgB;AACxC,eAAO,KAAK,SAAS,CAAC;AACtB,kBAAU;AACV,iBAAS,EAAE,OAAO,KAAK,UAAU;AAAA,MACnC;AAEA,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA;AAAA,QACA,UAAU,KAAK,UAAU,IAAI;AAAA,QAC7B;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAGA,QAAI,SAAS,gBAAgB;AAC3B,YAAM,WAAW,KAAK;AACtB,YAAM,aAAa,KAAK;AAGxB,UAAI,OAAY,CAAC;AACjB,UAAI,SAAc;AAClB,UAAI,UAAU;AAEd,UACE,KAAK,UAAU,qBACf,KAAK,UAAU,mBACf;AACA,eAAO,KAAK,SAAS,CAAC;AAAA,MACxB,WAAW,KAAK,UAAU,oBAAoB;AAC5C,eAAO,KAAK,SAAS,CAAC;AACtB,iBAAS,KAAK;AAAA,MAChB,WAAW,KAAK,UAAU,gBAAgB;AACxC,eAAO,KAAK,SAAS,CAAC;AACtB,kBAAU;AACV,iBAAS,EAAE,OAAO,KAAK,UAAU;AAAA,MACnC;AAEA,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA;AAAA,QACA,UAAU,KAAK,UAAU,IAAI;AAAA,QAC7B;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAGA,QAAI,SAAS,cAAc;AACzB,aAAO;AAAA,QACL,MAAM;AAAA,QACN,YAAY;AAAA,QACZ,IAAI,KAAK;AAAA,QACT,KAAK,KAAK;AAAA,QACV,OAAO,KAAK,SAAS;AAAA,MACvB;AAAA,IACF;AAGA,QAAI,SAAS,mBAAmB;AAC9B,cAAQ;AAAA,QACN,6BAA6B,IAAI;AAAA,MACnC;AACA,aAAO;AAAA,IACT;AAGA,QAAI,SAAS,QAAQ;AACnB,aAAO;AAAA,QACL,MAAM;AAAA,QACN,MAAM,KAAK;AAAA,QACX,UAAU,KAAK;AAAA,MACjB;AAAA,IACF;AAGA,QAAI,KAAK,WAAW,OAAO,GAAG;AAI5B,cAAQ;AAAA,QACN,kBAAkB,IAAI;AAAA,MACxB;AACA,aAAO;AAAA,IACT;AAGA,YAAQ,KAAK,kCAAkC,IAAI,EAAE;AACrD,WAAO;AAAA,EACT,CAAC,EACA,OAAO,OAAO;AACnB;AAEO,IAAM,wBAAwB;AAAA,EACnC,CAAC,YAAuB;AAEtB,UAAM,YAAY,oBAAI,KAAK;AAC3B,YAAQ,QAAQ,MAAM;AAAA,MACpB,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ;AAAA,UACA,SAAS,aAAa,OAAO;AAAA,UAC7B,aAAa,QAAQ,OACjB,OAAO,CAAC,MAAW,EAAE,SAAS,MAAM,EACrC,IAAI,CAAC,MAAW,SAAS;AAAA,YACxB,IAAI,IAAI,SAAS;AAAA,YACjB,MAAM;AAAA,YACN,MAAM,KAAK,QAAQ,KAAK,OAAO;AAAA,YAC/B,SAAS,CAAC;AAAA,YACV,aAAa,KAAK,aAAa,KAAK,YAAY;AAAA,YAChD,QAAQ,EAAE,MAAM,WAAoB;AAAA,UACtC,EAAE;AAAA,QACN;AAAA,MAEF,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ;AAAA,UACA,SAAS,aAAa,OAAO;AAAA,QAC/B;AAAA,MAEF,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ;AAAA,UACA,SAAS,aAAa,OAAO;AAAA,UAC7B,UAAU;AAAA,YACR,sBAAuB,QAAgB;AAAA,YACvC,eAAe,MAAM,QAAS,QAAgB,IAAI,IAC7C,QAAgB,OAChB,QAAgB,OACf,CAAE,QAAgB,IAAI,IACtB;AAAA,YACN,QAAQ,CAAC;AAAA,UACX;AAAA,QACF;AAAA,MAEF;AACE,gBAAQ,KAAK,6BAA6B,QAAQ,IAAI,EAAE;AACxD,eAAO,CAAC;AAAA,IACZ;AAAA,EACF;AACF;","names":[]}
+{"version":3,"sources":["../../../src/ui/utils/convertMessage.ts"],"sourcesContent":["import { isToolUIPart, UIMessage } from \"ai\";\nimport {\n  unstable_createMessageConverter,\n  type ReasoningMessagePart,\n  type ToolCallMessagePart,\n  type TextMessagePart,\n  type SourceMessagePart,\n} from \"@assistant-ui/react\";\n\nconst convertParts = (message: UIMessage) => {\n  if (!message.parts || message.parts.length === 0) {\n    return [];\n  }\n\n  return message.parts\n    .filter((p) => p.type !== \"step-start\" && p.type !== \"file\")\n    .map((part) => {\n      const type = part.type;\n\n      // Handle text parts\n      if (type === \"text\") {\n        return {\n          type: \"text\",\n          text: part.text,\n        } satisfies TextMessagePart;\n      }\n\n      // Handle reasoning parts\n      if (type === \"reasoning\") {\n        return {\n          type: \"reasoning\",\n          text: part.text,\n        } satisfies ReasoningMessagePart;\n      }\n\n      // Handle tool-* parts (AI SDK v5 tool calls)\n      if (isToolUIPart(part)) {\n        const toolName = type.replace(\"tool-\", \"\");\n        const toolCallId = part.toolCallId;\n\n        // Extract args and result based on state\n        let args: any = {};\n        let result: any = undefined;\n        let isError = false;\n\n        if (\n          part.state === \"input-streaming\" ||\n          part.state === \"input-available\"\n        ) {\n          args = part.input || {};\n        } else if (part.state === \"output-available\") {\n          args = part.input || {};\n          result = part.output;\n        } else if (part.state === \"output-error\") {\n          args = part.input || {};\n          isError = true;\n          result = { error: part.errorText };\n        }\n\n        return {\n          type: \"tool-call\",\n          toolName,\n          toolCallId,\n          argsText: JSON.stringify(args),\n          args,\n          result,\n          isError,\n        } satisfies ToolCallMessagePart;\n      }\n\n      // Handle dynamic-tool parts\n      if (type === \"dynamic-tool\") {\n        const toolName = part.toolName;\n        const toolCallId = part.toolCallId;\n\n        // Extract args and result based on state\n        let args: any = {};\n        let result: any = undefined;\n        let isError = false;\n\n        if (\n          part.state === \"input-streaming\" ||\n          part.state === \"input-available\"\n        ) {\n          args = part.input || {};\n        } else if (part.state === \"output-available\") {\n          args = part.input || {};\n          result = part.output;\n        } else if (part.state === \"output-error\") {\n          args = part.input || {};\n          isError = true;\n          result = { error: part.errorText };\n        }\n\n        return {\n          type: \"tool-call\",\n          toolName,\n          toolCallId,\n          argsText: JSON.stringify(args),\n          args,\n          result,\n          isError,\n        } satisfies ToolCallMessagePart;\n      }\n\n      // Handle source-url parts\n      if (type === \"source-url\") {\n        return {\n          type: \"source\",\n          sourceType: \"url\",\n          id: part.sourceId,\n          url: part.url,\n          title: part.title || \"\",\n        } satisfies SourceMessagePart;\n      }\n\n      // Handle source-document parts\n      if (type === \"source-document\") {\n        console.warn(\n          `Source document part type ${type} is not yet supported in conversion`,\n        );\n        return null;\n      }\n\n      // Handle data-* parts (AI SDK v5 data parts)\n      if (type.startsWith(\"data-\")) {\n        // For now, we'll skip data parts as they don't have a direct equivalent\n        // in the assistant-ui types. They could be converted to a custom message part\n        // or handled differently based on the specific use case.\n        console.warn(\n          `Data part type ${type} is not yet supported in conversion`,\n        );\n        return null;\n      }\n\n      // For unsupported types, we'll skip them instead of throwing\n      console.warn(`Unsupported message part type: ${type}`);\n      return null;\n    })\n    .filter(Boolean) as any[];\n};\n\nexport const AISDKMessageConverter = unstable_createMessageConverter(\n  (message: UIMessage) => {\n    // UIMessage doesn't have createdAt, so we'll use current date or undefined\n    const createdAt = new Date();\n    switch (message.role) {\n      case \"user\":\n        return {\n          role: \"user\",\n          id: message.id,\n          createdAt,\n          content: convertParts(message),\n          attachments: message.parts\n            ?.filter((p) => p.type === \"file\")\n            .map((part, idx) => {\n              return {\n                id: idx.toString(),\n                type: part.mediaType.startsWith(\"image/\") ? \"image\" : \"file\",\n                name: part.filename ?? \"file\",\n                content: [\n                  part.mediaType.startsWith(\"image/\")\n                    ? {\n                        type: \"image\",\n                        image: part.url,\n                        filename: part.filename!,\n                      }\n                    : {\n                        type: \"file\",\n                        filename: part.filename!,\n                        data: part.url,\n                        mimeType: part.mediaType,\n                      },\n                ],\n                contentType: part.mediaType ?? \"unknown/unknown\",\n                status: { type: \"complete\" as const },\n              };\n            }),\n        };\n\n      case \"system\":\n        return {\n          role: \"system\",\n          id: message.id,\n          createdAt,\n          content: convertParts(message),\n        };\n\n      case \"assistant\":\n        return {\n          role: \"assistant\",\n          id: message.id,\n          createdAt,\n          content: convertParts(message),\n          metadata: {\n            unstable_annotations: (message as any).annotations,\n            unstable_data: Array.isArray((message as any).data)\n              ? (message as any).data\n              : (message as any).data\n                ? [(message as any).data]\n                : undefined,\n            custom: {},\n          },\n        };\n\n      default:\n        console.warn(`Unsupported message role: ${message.role}`);\n        return [];\n    }\n  },\n);\n"],"mappings":";AAAA,SAAS,oBAA+B;AACxC;AAAA,EACE;AAAA,OAKK;AAEP,IAAM,eAAe,CAAC,YAAuB;AAC3C,MAAI,CAAC,QAAQ,SAAS,QAAQ,MAAM,WAAW,GAAG;AAChD,WAAO,CAAC;AAAA,EACV;AAEA,SAAO,QAAQ,MACZ,OAAO,CAAC,MAAM,EAAE,SAAS,gBAAgB,EAAE,SAAS,MAAM,EAC1D,IAAI,CAAC,SAAS;AACb,UAAM,OAAO,KAAK;AAGlB,QAAI,SAAS,QAAQ;AACnB,aAAO;AAAA,QACL,MAAM;AAAA,QACN,MAAM,KAAK;AAAA,MACb;AAAA,IACF;AAGA,QAAI,SAAS,aAAa;AACxB,aAAO;AAAA,QACL,MAAM;AAAA,QACN,MAAM,KAAK;AAAA,MACb;AAAA,IACF;AAGA,QAAI,aAAa,IAAI,GAAG;AACtB,YAAM,WAAW,KAAK,QAAQ,SAAS,EAAE;AACzC,YAAM,aAAa,KAAK;AAGxB,UAAI,OAAY,CAAC;AACjB,UAAI,SAAc;AAClB,UAAI,UAAU;AAEd,UACE,KAAK,UAAU,qBACf,KAAK,UAAU,mBACf;AACA,eAAO,KAAK,SAAS,CAAC;AAAA,MACxB,WAAW,KAAK,UAAU,oBAAoB;AAC5C,eAAO,KAAK,SAAS,CAAC;AACtB,iBAAS,KAAK;AAAA,MAChB,WAAW,KAAK,UAAU,gBAAgB;AACxC,eAAO,KAAK,SAAS,CAAC;AACtB,kBAAU;AACV,iBAAS,EAAE,OAAO,KAAK,UAAU;AAAA,MACnC;AAEA,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA;AAAA,QACA,UAAU,KAAK,UAAU,IAAI;AAAA,QAC7B;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAGA,QAAI,SAAS,gBAAgB;AAC3B,YAAM,WAAW,KAAK;AACtB,YAAM,aAAa,KAAK;AAGxB,UAAI,OAAY,CAAC;AACjB,UAAI,SAAc;AAClB,UAAI,UAAU;AAEd,UACE,KAAK,UAAU,qBACf,KAAK,UAAU,mBACf;AACA,eAAO,KAAK,SAAS,CAAC;AAAA,MACxB,WAAW,KAAK,UAAU,oBAAoB;AAC5C,eAAO,KAAK,SAAS,CAAC;AACtB,iBAAS,KAAK;AAAA,MAChB,WAAW,KAAK,UAAU,gBAAgB;AACxC,eAAO,KAAK,SAAS,CAAC;AACtB,kBAAU;AACV,iBAAS,EAAE,OAAO,KAAK,UAAU;AAAA,MACnC;AAEA,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA;AAAA,QACA,UAAU,KAAK,UAAU,IAAI;AAAA,QAC7B;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAGA,QAAI,SAAS,cAAc;AACzB,aAAO;AAAA,QACL,MAAM;AAAA,QACN,YAAY;AAAA,QACZ,IAAI,KAAK;AAAA,QACT,KAAK,KAAK;AAAA,QACV,OAAO,KAAK,SAAS;AAAA,MACvB;AAAA,IACF;AAGA,QAAI,SAAS,mBAAmB;AAC9B,cAAQ;AAAA,QACN,6BAA6B,IAAI;AAAA,MACnC;AACA,aAAO;AAAA,IACT;AAGA,QAAI,KAAK,WAAW,OAAO,GAAG;AAI5B,cAAQ;AAAA,QACN,kBAAkB,IAAI;AAAA,MACxB;AACA,aAAO;AAAA,IACT;AAGA,YAAQ,KAAK,kCAAkC,IAAI,EAAE;AACrD,WAAO;AAAA,EACT,CAAC,EACA,OAAO,OAAO;AACnB;AAEO,IAAM,wBAAwB;AAAA,EACnC,CAAC,YAAuB;AAEtB,UAAM,YAAY,oBAAI,KAAK;AAC3B,YAAQ,QAAQ,MAAM;AAAA,MACpB,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ;AAAA,UACA,SAAS,aAAa,OAAO;AAAA,UAC7B,aAAa,QAAQ,OACjB,OAAO,CAAC,MAAM,EAAE,SAAS,MAAM,EAChC,IAAI,CAAC,MAAM,QAAQ;AAClB,mBAAO;AAAA,cACL,IAAI,IAAI,SAAS;AAAA,cACjB,MAAM,KAAK,UAAU,WAAW,QAAQ,IAAI,UAAU;AAAA,cACtD,MAAM,KAAK,YAAY;AAAA,cACvB,SAAS;AAAA,gBACP,KAAK,UAAU,WAAW,QAAQ,IAC9B;AAAA,kBACE,MAAM;AAAA,kBACN,OAAO,KAAK;AAAA,kBACZ,UAAU,KAAK;AAAA,gBACjB,IACA;AAAA,kBACE,MAAM;AAAA,kBACN,UAAU,KAAK;AAAA,kBACf,MAAM,KAAK;AAAA,kBACX,UAAU,KAAK;AAAA,gBACjB;AAAA,cACN;AAAA,cACA,aAAa,KAAK,aAAa;AAAA,cAC/B,QAAQ,EAAE,MAAM,WAAoB;AAAA,YACtC;AAAA,UACF,CAAC;AAAA,QACL;AAAA,MAEF,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ;AAAA,UACA,SAAS,aAAa,OAAO;AAAA,QAC/B;AAAA,MAEF,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ;AAAA,UACA,SAAS,aAAa,OAAO;AAAA,UAC7B,UAAU;AAAA,YACR,sBAAuB,QAAgB;AAAA,YACvC,eAAe,MAAM,QAAS,QAAgB,IAAI,IAC7C,QAAgB,OAChB,QAAgB,OACf,CAAE,QAAgB,IAAI,IACtB;AAAA,YACN,QAAQ,CAAC;AAAA,UACX;AAAA,QACF;AAAA,MAEF;AACE,gBAAQ,KAAK,6BAA6B,QAAQ,IAAI,EAAE;AACxD,eAAO,CAAC;AAAA,IACZ;AAAA,EACF;AACF;","names":[]}

package/dist/ui/utils/toCreateMessage.d.ts

@@ -1,4 +1,4 @@
 import { AppendMessage } from "@assistant-ui/react";
 import { CreateUIMessage, UIMessage } from "ai";
-export declare const toCreateMessage: <UI_MESSAGE extends UIMessage = UIMessage>(message: AppendMessage) => Promise<CreateUIMessage<UI_MESSAGE>>;
+export declare const toCreateMessage: <UI_MESSAGE extends UIMessage = UIMessage>(message: AppendMessage) => CreateUIMessage<UI_MESSAGE>;
 //# sourceMappingURL=toCreateMessage.d.ts.map

package/dist/ui/utils/toCreateMessage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"toCreateMessage.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/toCreateMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EACL,eAAe,EAIf,SAAS,EAGV,MAAM,IAAI,CAAC;AAEZ,eAAO,MAAM,eAAe,GAAU,UAAU,SAAS,SAAS,GAAG,SAAS,EAC5E,SAAS,aAAa,KACrB,OAAO,CAAC,eAAe,CAAC,UAAU,CAAC,CA+CrC,CAAC"}
+{"version":3,"file":"toCreateMessage.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/toCreateMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EACL,eAAe,EAGf,SAAS,EAGV,MAAM,IAAI,CAAC;AAEZ,eAAO,MAAM,eAAe,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EACtE,SAAS,aAAa,KACrB,eAAe,CAAC,UAAU,CA0C5B,CAAC"}

package/dist/ui/utils/toCreateMessage.js

@@ -2,47 +2,47 @@
 import {
   generateId
 } from "ai";
-var toCreateMessage = async (message) => {
-  const textParts = message.content.filter((part) => part.type === "text").map((t) => t.text).join("\n\n");
-  const parts = [
-    {
-      type: "text",
-      text: textParts
-    }
+var toCreateMessage = (message) => {
+  const inputParts = [
+    ...message.content.filter((c) => c.type !== "file"),
+    ...message.attachments?.flatMap(
+      (a) => a.content.map((c) => ({
+        ...c,
+        filename: a.name
+      }))
+    ) ?? []
   ];
-  const imageParts = message.content.filter((part) => part.type === "image").map(
-    (part) => ({
-      type: "file",
-      mediaType: "image/png",
-      // Default to PNG, could be made more dynamic
-      url: part.image
-    })
-  );
-  parts.push(...imageParts);
-  const attachmentParts = await Promise.all(
-    (message.attachments ?? []).map(async (m) => {
-      if (m.file == null) throw new Error("Attachment did not contain a file");
-      return {
-        type: "file",
-        mediaType: m.file.type,
-        filename: m.file.name,
-        url: await getFileDataURL(m.file)
-      };
-    })
-  );
-  parts.push(...attachmentParts);
+  const parts = inputParts.map((part) => {
+    switch (part.type) {
+      case "text":
+        return {
+          type: "text",
+          text: part.text
+        };
+      case "image":
+        return {
+          type: "file",
+          url: part.image,
+          ...part.filename && { filename: part.filename },
+          mediaType: "image/png"
+        };
+      case "file":
+        return {
+          type: "file",
+          url: part.data,
+          mediaType: part.mimeType,
+          ...part.filename && { filename: part.filename }
+        };
+      default:
+        throw new Error(`Unsupported part type: ${part.type}`);
+    }
+  });
   return {
     id: generateId(),
     role: message.role,
     parts
   };
 };
-var getFileDataURL = (file) => new Promise((resolve, reject) => {
-  const reader = new FileReader();
-  reader.onload = () => resolve(reader.result);
-  reader.onerror = (error) => reject(error);
-  reader.readAsDataURL(file);
-});
 export {
   toCreateMessage
 };

package/dist/ui/utils/toCreateMessage.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/ui/utils/toCreateMessage.ts"],"sourcesContent":["import { AppendMessage } from \"@assistant-ui/react\";\nimport {\n  CreateUIMessage,\n  FileUIPart,\n  generateId,\n  UIDataTypes,\n  UIMessage,\n  UIMessagePart,\n  UITools,\n} from \"ai\";\n\nexport const toCreateMessage = async <UI_MESSAGE extends UIMessage = UIMessage>(\n  message: AppendMessage,\n): Promise<CreateUIMessage<UI_MESSAGE>> => {\n  const textParts = message.content\n    .filter((part) => part.type === \"text\")\n    .map((t) => t.text)\n    .join(\"\\n\\n\");\n\n  const parts: UIMessagePart<UIDataTypes, UITools>[] = [\n    {\n      type: \"text\",\n      text: textParts,\n    },\n  ];\n\n  // Add image parts\n  const imageParts = message.content\n    .filter((part) => part.type === \"image\")\n    .map(\n      (part) =>\n        ({\n          type: \"file\",\n          mediaType: \"image/png\", // Default to PNG, could be made more dynamic\n          url: part.image,\n        }) satisfies FileUIPart,\n    );\n\n  parts.push(...imageParts);\n\n  // Add attachment parts\n  const attachmentParts = await Promise.all(\n    (message.attachments ?? []).map(async (m) => {\n      if (m.file == null) throw new Error(\"Attachment did not contain a file\");\n      return {\n        type: \"file\",\n        mediaType: m.file.type,\n        filename: m.file.name,\n        url: await getFileDataURL(m.file),\n      } satisfies FileUIPart;\n    }),\n  );\n\n  parts.push(...attachmentParts);\n\n  return {\n    id: generateId(),\n    role: message.role,\n    parts,\n  } satisfies CreateUIMessage<UIMessage> as CreateUIMessage<UI_MESSAGE>;\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"],"mappings":";AACA;AAAA,EAGE;AAAA,OAKK;AAEA,IAAM,kBAAkB,OAC7B,YACyC;AACzC,QAAM,YAAY,QAAQ,QACvB,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,MAAM,EAAE,IAAI,EACjB,KAAK,MAAM;AAEd,QAAM,QAA+C;AAAA,IACnD;AAAA,MACE,MAAM;AAAA,MACN,MAAM;AAAA,IACR;AAAA,EACF;AAGA,QAAM,aAAa,QAAQ,QACxB,OAAO,CAAC,SAAS,KAAK,SAAS,OAAO,EACtC;AAAA,IACC,CAAC,UACE;AAAA,MACC,MAAM;AAAA,MACN,WAAW;AAAA;AAAA,MACX,KAAK,KAAK;AAAA,IACZ;AAAA,EACJ;AAEF,QAAM,KAAK,GAAG,UAAU;AAGxB,QAAM,kBAAkB,MAAM,QAAQ;AAAA,KACnC,QAAQ,eAAe,CAAC,GAAG,IAAI,OAAO,MAAM;AAC3C,UAAI,EAAE,QAAQ,KAAM,OAAM,IAAI,MAAM,mCAAmC;AACvE,aAAO;AAAA,QACL,MAAM;AAAA,QACN,WAAW,EAAE,KAAK;AAAA,QAClB,UAAU,EAAE,KAAK;AAAA,QACjB,KAAK,MAAM,eAAe,EAAE,IAAI;AAAA,MAClC;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,KAAK,GAAG,eAAe;AAE7B,SAAO;AAAA,IACL,IAAI,WAAW;AAAA,IACf,MAAM,QAAQ;AAAA,IACd;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;","names":[]}
+{"version":3,"sources":["../../../src/ui/utils/toCreateMessage.ts"],"sourcesContent":["import { AppendMessage } from \"@assistant-ui/react\";\nimport {\n  CreateUIMessage,\n  generateId,\n  UIDataTypes,\n  UIMessage,\n  UIMessagePart,\n  UITools,\n} from \"ai\";\n\nexport const toCreateMessage = <UI_MESSAGE extends UIMessage = UIMessage>(\n  message: AppendMessage,\n): CreateUIMessage<UI_MESSAGE> => {\n  const inputParts = [\n    ...message.content.filter((c) => c.type !== \"file\"),\n    ...(message.attachments?.flatMap((a) =>\n      a.content.map((c) => ({\n        ...c,\n        filename: a.name,\n      })),\n    ) ?? []),\n  ];\n\n  const parts = inputParts.map((part): UIMessagePart<UIDataTypes, UITools> => {\n    switch (part.type) {\n      case \"text\":\n        return {\n          type: \"text\",\n          text: part.text,\n        };\n      case \"image\":\n        return {\n          type: \"file\",\n          url: part.image,\n          ...(part.filename && { filename: part.filename }),\n          mediaType: \"image/png\",\n        };\n      case \"file\":\n        return {\n          type: \"file\",\n          url: part.data,\n          mediaType: part.mimeType,\n          ...(part.filename && { filename: part.filename }),\n        };\n      default:\n        throw new Error(`Unsupported part type: ${part.type}`);\n    }\n  });\n\n  return {\n    id: generateId(),\n    role: message.role,\n    parts,\n  } satisfies CreateUIMessage<UIMessage> as CreateUIMessage<UI_MESSAGE>;\n};\n"],"mappings":";AACA;AAAA,EAEE;AAAA,OAKK;AAEA,IAAM,kBAAkB,CAC7B,YACgC;AAChC,QAAM,aAAa;AAAA,IACjB,GAAG,QAAQ,QAAQ,OAAO,CAAC,MAAM,EAAE,SAAS,MAAM;AAAA,IAClD,GAAI,QAAQ,aAAa;AAAA,MAAQ,CAAC,MAChC,EAAE,QAAQ,IAAI,CAAC,OAAO;AAAA,QACpB,GAAG;AAAA,QACH,UAAU,EAAE;AAAA,MACd,EAAE;AAAA,IACJ,KAAK,CAAC;AAAA,EACR;AAEA,QAAM,QAAQ,WAAW,IAAI,CAAC,SAA8C;AAC1E,YAAQ,KAAK,MAAM;AAAA,MACjB,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,MAAM,KAAK;AAAA,QACb;AAAA,MACF,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,KAAK,KAAK;AAAA,UACV,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;AAAA,UAC/C,WAAW;AAAA,QACb;AAAA,MACF,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,KAAK,KAAK;AAAA,UACV,WAAW,KAAK;AAAA,UAChB,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;AAAA,QACjD;AAAA,MACF;AACE,cAAM,IAAI,MAAM,0BAA0B,KAAK,IAAI,EAAE;AAAA,IACzD;AAAA,EACF,CAAC;AAED,SAAO;AAAA,IACL,IAAI,WAAW;AAAA,IACf,MAAM,QAAQ;AAAA,IACd;AAAA,EACF;AACF;","names":[]}

package/dist/ui/utils/vercelAttachmentAdapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vercelAttachmentAdapter.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/vercelAttachmentAdapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAGxD,eAAO,MAAM,uBAAuB,EAAE,iBAyBrC,CAAC"}
+{"version":3,"file":"vercelAttachmentAdapter.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/vercelAttachmentAdapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAaxD,eAAO,MAAM,uBAAuB,EAAE,iBAgCrC,CAAC"}

package/dist/ui/utils/vercelAttachmentAdapter.js

@@ -1,11 +1,17 @@
 // src/ui/utils/vercelAttachmentAdapter.ts
 import { generateId } from "ai";
+var getFileDataURL = (file) => new Promise((resolve, reject) => {
+  const reader = new FileReader();
+  reader.onload = () => resolve(reader.result);
+  reader.onerror = (error) => reject(error);
+  reader.readAsDataURL(file);
+});
 var vercelAttachmentAdapter = {
   accept: "image/*, text/plain, text/html, text/markdown, text/csv, text/xml, text/json, text/css",
   async add({ file }) {
     return {
       id: generateId(),
-      type: "file",
+      type: file.type.startsWith("image/") ? "image" : "file",
       name: file.name,
       file,
       contentType: file.type,
@@ -17,7 +23,14 @@ var vercelAttachmentAdapter = {
     return {
       ...attachment,
       status: { type: "complete" },
-      content: []
+      content: [
+        {
+          type: "file",
+          mimeType: attachment.contentType,
+          filename: attachment.name,
+          data: await getFileDataURL(attachment.file)
+        }
+      ]
     };
   },
   async remove() {

package/dist/ui/utils/vercelAttachmentAdapter.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/ui/utils/vercelAttachmentAdapter.ts"],"sourcesContent":["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      contentType: file.type,\n      content: [],\n      status: { type: \"requires-action\", reason: \"composer-send\" },\n    };\n  },\n  async send(attachment) {\n    // noop\n    return {\n      ...attachment,\n      status: { type: \"complete\" },\n      content: [],\n    };\n  },\n  async remove() {\n    // noop\n  },\n};\n"],"mappings":";AACA,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,aAAa,KAAK;AAAA,MAClB,SAAS,CAAC;AAAA,MACV,QAAQ,EAAE,MAAM,mBAAmB,QAAQ,gBAAgB;AAAA,IAC7D;AAAA,EACF;AAAA,EACA,MAAM,KAAK,YAAY;AAErB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,QAAQ,EAAE,MAAM,WAAW;AAAA,MAC3B,SAAS,CAAC;AAAA,IACZ;AAAA,EACF;AAAA,EACA,MAAM,SAAS;AAAA,EAEf;AACF;","names":[]}
+{"version":3,"sources":["../../../src/ui/utils/vercelAttachmentAdapter.ts"],"sourcesContent":["import { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { generateId } from \"ai\";\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\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.type.startsWith(\"image/\") ? \"image\" : \"file\",\n      name: file.name,\n      file,\n      contentType: file.type,\n      content: [],\n      status: { type: \"requires-action\", reason: \"composer-send\" },\n    };\n  },\n  async send(attachment) {\n    // noop\n    return {\n      ...attachment,\n      status: { type: \"complete\" },\n      content: [\n        {\n          type: \"file\",\n          mimeType: attachment.contentType,\n          filename: attachment.name,\n          data: await getFileDataURL(attachment.file),\n        },\n      ],\n    };\n  },\n  async remove() {\n    // noop\n  },\n};\n"],"mappings":";AACA,SAAS,kBAAkB;AAE3B,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;AAEI,IAAM,0BAA6C;AAAA,EACxD,QACE;AAAA,EACF,MAAM,IAAI,EAAE,KAAK,GAAG;AAClB,WAAO;AAAA,MACL,IAAI,WAAW;AAAA,MACf,MAAM,KAAK,KAAK,WAAW,QAAQ,IAAI,UAAU;AAAA,MACjD,MAAM,KAAK;AAAA,MACX;AAAA,MACA,aAAa,KAAK;AAAA,MAClB,SAAS,CAAC;AAAA,MACV,QAAQ,EAAE,MAAM,mBAAmB,QAAQ,gBAAgB;AAAA,IAC7D;AAAA,EACF;AAAA,EACA,MAAM,KAAK,YAAY;AAErB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,QAAQ,EAAE,MAAM,WAAW;AAAA,MAC3B,SAAS;AAAA,QACP;AAAA,UACE,MAAM;AAAA,UACN,UAAU,WAAW;AAAA,UACrB,UAAU,WAAW;AAAA,UACrB,MAAM,MAAM,eAAe,WAAW,IAAI;AAAA,QAC5C;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EACA,MAAM,SAAS;AAAA,EAEf;AACF;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ai-sdk",
-  "version": "1.0.4",
+  "version": "1.0.6",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -29,7 +29,7 @@
     "zustand": "^5.0.7"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.10.43",
+    "@assistant-ui/react": "^0.10.50",
     "@types/react": "*",
     "assistant-cloud": "*",
     "react": "^18 || ^19 || ^19.0.0-rc"
@@ -49,8 +49,8 @@
     "eslint-config-next": "15.4.6",
     "react": "19.1.1",
     "tsx": "^4.20.4",
-    "@assistant-ui/x-buildutils": "0.0.1",
-    "@assistant-ui/react": "0.10.43"
+    "@assistant-ui/react": "0.10.50",
+    "@assistant-ui/x-buildutils": "0.0.1"
   },
   "publishConfig": {
     "access": "public",

package/src/ui/use-chat/useAISDKRuntime.tsx

@@ -8,6 +8,7 @@ import {
   AssistantRuntime,
   ThreadMessage,
   MessageFormatAdapter,
+  useRuntimeAdapters,
 } from "@assistant-ui/react";
 import { sliceMessagesUntil } from "../utils/sliceMessagesUntil";
 import { toCreateMessage } from "../utils/toCreateMessage";
@@ -33,6 +34,7 @@ export const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
   chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>,
   adapter: AISDKRuntimeAdapter = {},
 ) => {
+  const contextAdapters = useRuntimeAdapters();
   const messages = AISDKMessageConverter.useThreadMessages({
     isRunning:
       chatHelpers.status === "submitted" || chatHelpers.status == "streaming",
@@ -72,7 +74,7 @@ export const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
       ),
     onCancel: async () => chatHelpers.stop(),
     onNew: async (message) => {
-      const createMessage = await toCreateMessage<UI_MESSAGE>(message);
+      const createMessage = toCreateMessage<UI_MESSAGE>(message);
       await chatHelpers.sendMessage(createMessage, {
         metadata: message.runConfig,
       });
@@ -84,7 +86,7 @@ export const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
       );
       chatHelpers.setMessages(newMessages);
 
-      const createMessage = await toCreateMessage<UI_MESSAGE>(message);
+      const createMessage = toCreateMessage<UI_MESSAGE>(message);
       await chatHelpers.sendMessage(createMessage, {
         metadata: message.runConfig,
       });
@@ -104,6 +106,7 @@ export const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
     },
     adapters: {
       attachments: vercelAttachmentAdapter,
+      ...contextAdapters,
       ...adapter.adapters,
     },
     isLoading,

package/src/ui/use-chat/useChatRuntime.tsx

@@ -6,7 +6,6 @@ import {
   AssistantRuntime,
   unstable_useCloudThreadListAdapter,
   unstable_useRemoteThreadListRuntime,
-  useRuntimeAdapters,
 } from "@assistant-ui/react";
 import { useAISDKRuntime, type AISDKRuntimeAdapter } from "./useAISDKRuntime";
 import { ChatInit } from "ai";
@@ -28,9 +27,6 @@ export const useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
   } = options ?? {};
   const transport = transportOptions ?? new AssistantChatTransport();
 
-  // Get adapters from context (including history adapter from cloud)
-  const contextAdapters = useRuntimeAdapters();
-
   const chat = useChat({
     ...chatOptions,
     transport,
@@ -64,10 +60,7 @@ export const useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
   });
 
   const runtime = useAISDKRuntime(chat as any, {
-    adapters: {
-      ...contextAdapters,
-      ...adapters,
-    },
+    adapters,
   });
   if (transport instanceof AssistantChatTransport) {
     transport.setRuntime(runtime);

package/src/ui/use-chat/useExternalHistory.tsx

@@ -97,7 +97,7 @@ export const useExternalHistory = <TMessage,>(
           message.status.type === "complete" ||
           message.status.type === "incomplete"
         ) {
-          if (historyIds.current.has(message.id)) return;
+          if (historyIds.current.has(message.id)) continue;
           historyIds.current.add(message.id);
 
           const parentId = i > 0 ? messages[i - 1]!.id : null;

package/src/ui/utils/convertMessage.ts

@@ -5,7 +5,6 @@ import {
   type ToolCallMessagePart,
   type TextMessagePart,
   type SourceMessagePart,
-  type FileMessagePart,
 } from "@assistant-ui/react";
 
 const convertParts = (message: UIMessage) => {
@@ -14,7 +13,7 @@ const convertParts = (message: UIMessage) => {
   }
 
   return message.parts
-    .filter((p) => p.type !== "step-start")
+    .filter((p) => p.type !== "step-start" && p.type !== "file")
     .map((part) => {
       const type = part.type;
 
@@ -123,15 +122,6 @@ const convertParts = (message: UIMessage) => {
         return null;
       }
 
-      // Handle file parts
-      if (type === "file") {
-        return {
-          type: "file",
-          data: part.url,
-          mimeType: part.mediaType,
-        } satisfies FileMessagePart;
-      }
-
       // Handle data-* parts (AI SDK v5 data parts)
       if (type.startsWith("data-")) {
         // For now, we'll skip data parts as they don't have a direct equivalent
@@ -162,15 +152,30 @@ export const AISDKMessageConverter = unstable_createMessageConverter(
           createdAt,
           content: convertParts(message),
           attachments: message.parts
-            ?.filter((p: any) => p.type === "file")
-            .map((part: any, idx) => ({
-              id: idx.toString(),
-              type: "file" as const,
-              name: part.name ?? part.url ?? "file",
-              content: [],
-              contentType: part.mediaType ?? part.mimeType ?? "unknown/unknown",
-              status: { type: "complete" as const },
-            })),
+            ?.filter((p) => p.type === "file")
+            .map((part, idx) => {
+              return {
+                id: idx.toString(),
+                type: part.mediaType.startsWith("image/") ? "image" : "file",
+                name: part.filename ?? "file",
+                content: [
+                  part.mediaType.startsWith("image/")
+                    ? {
+                        type: "image",
+                        image: part.url,
+                        filename: part.filename!,
+                      }
+                    : {
+                        type: "file",
+                        filename: part.filename!,
+                        data: part.url,
+                        mimeType: part.mediaType,
+                      },
+                ],
+                contentType: part.mediaType ?? "unknown/unknown",
+                status: { type: "complete" as const },
+              };
+            }),
         };
 
       case "system":

package/src/ui/utils/toCreateMessage.ts

@@ -1,7 +1,6 @@
 import { AppendMessage } from "@assistant-ui/react";
 import {
   CreateUIMessage,
-  FileUIPart,
   generateId,
   UIDataTypes,
   UIMessage,
@@ -9,49 +8,44 @@ import {
   UITools,
 } from "ai";
 
-export const toCreateMessage = async <UI_MESSAGE extends UIMessage = UIMessage>(
+export const toCreateMessage = <UI_MESSAGE extends UIMessage = UIMessage>(
   message: AppendMessage,
-): Promise<CreateUIMessage<UI_MESSAGE>> => {
-  const textParts = message.content
-    .filter((part) => part.type === "text")
-    .map((t) => t.text)
-    .join("\n\n");
-
-  const parts: UIMessagePart<UIDataTypes, UITools>[] = [
-    {
-      type: "text",
-      text: textParts,
-    },
+): CreateUIMessage<UI_MESSAGE> => {
+  const inputParts = [
+    ...message.content.filter((c) => c.type !== "file"),
+    ...(message.attachments?.flatMap((a) =>
+      a.content.map((c) => ({
+        ...c,
+        filename: a.name,
+      })),
+    ) ?? []),
   ];
 
-  // Add image parts
-  const imageParts = message.content
-    .filter((part) => part.type === "image")
-    .map(
-      (part) =>
-        ({
+  const parts = inputParts.map((part): UIMessagePart<UIDataTypes, UITools> => {
+    switch (part.type) {
+      case "text":
+        return {
+          type: "text",
+          text: part.text,
+        };
+      case "image":
+        return {
           type: "file",
-          mediaType: "image/png", // Default to PNG, could be made more dynamic
           url: part.image,
-        }) satisfies FileUIPart,
-    );
-
-  parts.push(...imageParts);
-
-  // Add attachment parts
-  const attachmentParts = await Promise.all(
-    (message.attachments ?? []).map(async (m) => {
-      if (m.file == null) throw new Error("Attachment did not contain a file");
-      return {
-        type: "file",
-        mediaType: m.file.type,
-        filename: m.file.name,
-        url: await getFileDataURL(m.file),
-      } satisfies FileUIPart;
-    }),
-  );
-
-  parts.push(...attachmentParts);
+          ...(part.filename && { filename: part.filename }),
+          mediaType: "image/png",
+        };
+      case "file":
+        return {
+          type: "file",
+          url: part.data,
+          mediaType: part.mimeType,
+          ...(part.filename && { filename: part.filename }),
+        };
+      default:
+        throw new Error(`Unsupported part type: ${part.type}`);
+    }
+  });
 
   return {
     id: generateId(),
@@ -59,13 +53,3 @@ export const toCreateMessage = async <UI_MESSAGE extends UIMessage = UIMessage>(
     parts,
   } satisfies CreateUIMessage<UIMessage> as CreateUIMessage<UI_MESSAGE>;
 };
-
-const getFileDataURL = (file: File) =>
-  new Promise<string>((resolve, reject) => {
-    const reader = new FileReader();
-
-    reader.onload = () => resolve(reader.result as string);
-    reader.onerror = (error) => reject(error);
-
-    reader.readAsDataURL(file);
-  });

package/src/ui/utils/vercelAttachmentAdapter.ts

@@ -1,13 +1,23 @@
 import { AttachmentAdapter } from "@assistant-ui/react";
 import { generateId } from "ai";
 
+const getFileDataURL = (file: File) =>
+  new Promise<string>((resolve, reject) => {
+    const reader = new FileReader();
+
+    reader.onload = () => resolve(reader.result as string);
+    reader.onerror = (error) => reject(error);
+
+    reader.readAsDataURL(file);
+  });
+
 export const vercelAttachmentAdapter: AttachmentAdapter = {
   accept:
     "image/*, text/plain, text/html, text/markdown, text/csv, text/xml, text/json, text/css",
   async add({ file }) {
     return {
       id: generateId(),
-      type: "file",
+      type: file.type.startsWith("image/") ? "image" : "file",
       name: file.name,
       file,
       contentType: file.type,
@@ -20,7 +30,14 @@ export const vercelAttachmentAdapter: AttachmentAdapter = {
     return {
       ...attachment,
       status: { type: "complete" },
-      content: [],
+      content: [
+        {
+          type: "file",
+          mimeType: attachment.contentType,
+          filename: attachment.name,
+          data: await getFileDataURL(attachment.file),
+        },
+      ],
     };
   },
   async remove() {