@assistant-ui/react-ai-sdk 0.10.15 → 0.11.0-alpha.0-alpha

package/dist/frontendTools.d.ts

@@ -4,7 +4,7 @@ export declare const frontendTools: (tools: Record<string, {
     parameters: JSONSchema7;
 }>) => {
     [k: string]: {
-        parameters: import("@ai-sdk/ui-utils").Schema<unknown>;
+        parameters: import("ai").Schema<unknown>;
         description?: string;
     };
 };

package/dist/frontendTools.js

@@ -1,5 +1,5 @@
 // src/frontendTools.ts
-import { jsonSchema } from "@ai-sdk/ui-utils";
+import { jsonSchema } from "ai";
 var frontendTools = (tools) => Object.fromEntries(
   Object.entries(tools).map(([name, tool]) => [
     name,

package/dist/frontendTools.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/frontendTools.ts"],"sourcesContent":["import { jsonSchema } from \"@ai-sdk/ui-utils\";\nimport type { JSONSchema7 } from \"json-schema\";\n\nexport const frontendTools = (\n  tools: Record<string, { description?: string; parameters: JSONSchema7 }>,\n) =>\n  Object.fromEntries(\n    Object.entries(tools).map(([name, tool]) => [\n      name,\n      {\n        ...(tool.description ? { description: tool.description } : undefined),\n        parameters: jsonSchema(tool.parameters),\n      },\n    ]),\n  );\n"],"mappings":";AAAA,SAAS,kBAAkB;AAGpB,IAAM,gBAAgB,CAC3B,UAEA,OAAO;AAAA,EACL,OAAO,QAAQ,KAAK,EAAE,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM;AAAA,IAC1C;AAAA,IACA;AAAA,MACE,GAAI,KAAK,cAAc,EAAE,aAAa,KAAK,YAAY,IAAI;AAAA,MAC3D,YAAY,WAAW,KAAK,UAAU;AAAA,IACxC;AAAA,EACF,CAAC;AACH;","names":[]}
+{"version":3,"sources":["../src/frontendTools.ts"],"sourcesContent":["import { jsonSchema } from \"ai\";\nimport type { JSONSchema7 } from \"json-schema\";\n\nexport const frontendTools = (\n  tools: Record<string, { description?: string; parameters: JSONSchema7 }>,\n) =>\n  Object.fromEntries(\n    Object.entries(tools).map(([name, tool]) => [\n      name,\n      {\n        ...(tool.description ? { description: tool.description } : undefined),\n        parameters: jsonSchema(tool.parameters),\n      },\n    ]),\n  );\n"],"mappings":";AAAA,SAAS,kBAAkB;AAGpB,IAAM,gBAAgB,CAC3B,UAEA,OAAO;AAAA,EACL,OAAO,QAAQ,KAAK,EAAE,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM;AAAA,IAC1C;AAAA,IACA;AAAA,MACE,GAAI,KAAK,cAAc,EAAE,aAAa,KAAK,YAAY,IAAI;AAAA,MAC3D,YAAY,WAAW,KAAK,UAAU;AAAA,IACxC;AAAA,EACF,CAAC;AACH;","names":[]}

package/dist/index.d.ts

@@ -1,7 +1,3 @@
-export * from "./rsc";
 export * from "./ui";
-export * from "./useChatRuntime";
-export * from "./useCloudRuntime";
-export { toLanguageModelMessages, toLanguageModelTools, fromLanguageModelMessages, fromLanguageModelTools, useDangerousInBrowserRuntime, } from "@assistant-ui/react-edge";
 export { frontendTools } from "./frontendTools";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,OAAO,CAAC;AACtB,cAAc,MAAM,CAAC;AACrB,cAAc,kBAAkB,CAAC;AACjC,cAAc,mBAAmB,CAAC;AAElC,OAAO,EACL,uBAAuB,EACvB,oBAAoB,EACpB,yBAAyB,EACzB,sBAAsB,EACtB,4BAA4B,GAC7B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,MAAM,CAAC;AACrB,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/index.js

@@ -1,22 +1,7 @@
 // src/index.ts
-export * from "./rsc/index.js";
 export * from "./ui/index.js";
-export * from "./useChatRuntime.js";
-export * from "./useCloudRuntime.js";
-import {
-  toLanguageModelMessages,
-  toLanguageModelTools,
-  fromLanguageModelMessages,
-  fromLanguageModelTools,
-  useDangerousInBrowserRuntime
-} from "@assistant-ui/react-edge";
 import { frontendTools } from "./frontendTools.js";
 export {
-  fromLanguageModelMessages,
-  fromLanguageModelTools,
-  frontendTools,
-  toLanguageModelMessages,
-  toLanguageModelTools,
-  useDangerousInBrowserRuntime
+  frontendTools
 };
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts"],"sourcesContent":["export * from \"./rsc\";\nexport * from \"./ui\";\nexport * from \"./useChatRuntime\";\nexport * from \"./useCloudRuntime\";\n\nexport {\n  toLanguageModelMessages,\n  toLanguageModelTools,\n  fromLanguageModelMessages,\n  fromLanguageModelTools,\n  useDangerousInBrowserRuntime,\n} from \"@assistant-ui/react-edge\";\nexport { frontendTools } from \"./frontendTools\";\n"],"mappings":";AAAA,cAAc;AACd,cAAc;AACd,cAAc;AACd,cAAc;AAEd;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,qBAAqB;","names":[]}
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["export * from \"./ui\";\nexport { frontendTools } from \"./frontendTools\";\n"],"mappings":";AAAA,cAAc;AACd,SAAS,qBAAqB;","names":[]}

package/dist/ui/getVercelAIMessages.d.ts

@@ -1,4 +1,4 @@
 import { type ThreadMessage } from "@assistant-ui/react";
-import type { Message } from "@ai-sdk/ui-utils";
-export declare const getVercelAIMessages: (message: ThreadMessage) => Message[];
+import type { UIMessage } from "ai";
+export declare const getVercelAIMessages: (message: ThreadMessage) => UIMessage[];
 //# sourceMappingURL=getVercelAIMessages.d.ts.map

package/dist/ui/getVercelAIMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"getVercelAIMessages.d.ts","sourceRoot":"","sources":["../../src/ui/getVercelAIMessages.tsx"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,aAAa,EACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAEhD,eAAO,MAAM,mBAAmB,GAAI,SAAS,aAAa,KACZ,OAAO,EACpD,CAAC"}
+{"version":3,"file":"getVercelAIMessages.d.ts","sourceRoot":"","sources":["../../src/ui/getVercelAIMessages.tsx"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,aAAa,EACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAEpC,eAAO,MAAM,mBAAmB,GAAI,SAAS,aAAa,KACZ,SAAS,EACtD,CAAC"}

package/dist/ui/getVercelAIMessages.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/ui/getVercelAIMessages.tsx"],"sourcesContent":["import {\n  getExternalStoreMessages,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\nimport type { Message } from \"@ai-sdk/ui-utils\";\n\nexport const getVercelAIMessages = (message: ThreadMessage) => {\n  return getExternalStoreMessages(message) as Message[];\n};\n"],"mappings":";AAAA;AAAA,EACE;AAAA,OAEK;AAGA,IAAM,sBAAsB,CAAC,YAA2B;AAC7D,SAAO,yBAAyB,OAAO;AACzC;","names":[]}
+{"version":3,"sources":["../../src/ui/getVercelAIMessages.tsx"],"sourcesContent":["import {\n  getExternalStoreMessages,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\nimport type { UIMessage } from \"ai\";\n\nexport const getVercelAIMessages = (message: ThreadMessage) => {\n  return getExternalStoreMessages(message) as UIMessage[];\n};\n"],"mappings":";AAAA;AAAA,EACE;AAAA,OAEK;AAGA,IAAM,sBAAsB,CAAC,YAA2B;AAC7D,SAAO,yBAAyB,OAAO;AACzC;","names":[]}

package/dist/ui/index.d.ts

@@ -1,5 +1,2 @@
-export { useVercelUseChatRuntime } from "./use-chat/useVercelUseChatRuntime";
-export { useVercelUseAssistantRuntime } from "./use-assistant/useVercelUseAssistantRuntime";
-export { getVercelAIMessages } from "./getVercelAIMessages";
-export { AISDKMessageConverter } from "./utils/convertMessage";
+export { useAISDKRuntime } from "./use-chat/useAISDKRuntime";
 //# sourceMappingURL=index.d.ts.map

package/dist/ui/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ui/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,MAAM,oCAAoC,CAAC;AAC7E,OAAO,EAAE,4BAA4B,EAAE,MAAM,8CAA8C,CAAC;AAC5F,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ui/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC"}

package/dist/ui/index.js

@@ -1,12 +1,6 @@
 // src/ui/index.ts
-import { useVercelUseChatRuntime } from "./use-chat/useVercelUseChatRuntime.js";
-import { useVercelUseAssistantRuntime } from "./use-assistant/useVercelUseAssistantRuntime.js";
-import { getVercelAIMessages } from "./getVercelAIMessages.js";
-import { AISDKMessageConverter } from "./utils/convertMessage.js";
+import { useAISDKRuntime } from "./use-chat/useAISDKRuntime.js";
 export {
-  AISDKMessageConverter,
-  getVercelAIMessages,
-  useVercelUseAssistantRuntime,
-  useVercelUseChatRuntime
+  useAISDKRuntime
 };
 //# sourceMappingURL=index.js.map

package/dist/ui/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/ui/index.ts"],"sourcesContent":["export { useVercelUseChatRuntime } from \"./use-chat/useVercelUseChatRuntime\";\nexport { useVercelUseAssistantRuntime } from \"./use-assistant/useVercelUseAssistantRuntime\";\nexport { getVercelAIMessages } from \"./getVercelAIMessages\";\nexport { AISDKMessageConverter } from \"./utils/convertMessage\";\n"],"mappings":";AAAA,SAAS,+BAA+B;AACxC,SAAS,oCAAoC;AAC7C,SAAS,2BAA2B;AACpC,SAAS,6BAA6B;","names":[]}
+{"version":3,"sources":["../../src/ui/index.ts"],"sourcesContent":["export { useAISDKRuntime } from \"./use-chat/useAISDKRuntime\";\n"],"mappings":";AAAA,SAAS,uBAAuB;","names":[]}

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

@@ -0,0 +1,7 @@
+import type { useChat } from "@ai-sdk/react";
+import { ExternalStoreAdapter } from "@assistant-ui/react";
+export type AISDKRuntimeAdapter = {
+    adapters?: Omit<NonNullable<ExternalStoreAdapter["adapters"]>, "attachments"> | undefined;
+};
+export declare const useAISDKRuntime: (chatHelpers: ReturnType<typeof useChat>, adapter?: AISDKRuntimeAdapter) => import("@assistant-ui/react").AssistantRuntime;
+//# sourceMappingURL=useAISDKRuntime.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"useAISDKRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.tsx"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAM7C,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAG3D,MAAM,MAAM,mBAAmB,GAAG;IAChC,QAAQ,CAAC,EACL,IAAI,CAAC,WAAW,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC,EAAE,aAAa,CAAC,GAClE,SAAS,CAAC;CACf,CAAC;AAEF,eAAO,MAAM,eAAe,GAC1B,aAAa,UAAU,CAAC,OAAO,OAAO,CAAC,EACvC,UAAS,mBAAwB,mDA+ClC,CAAC"}

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

@@ -0,0 +1,53 @@
+"use client";
+
+// src/ui/use-chat/useAISDKRuntime.tsx
+import { useExternalStoreRuntime } from "@assistant-ui/react";
+import { sliceMessagesUntil } from "../utils/sliceMessagesUntil.js";
+import { toCreateMessage } from "../utils/toCreateMessage.js";
+import { vercelAttachmentAdapter } from "../utils/vercelAttachmentAdapter.js";
+import { getVercelAIMessages } from "../getVercelAIMessages.js";
+import { AISDKMessageConverter } from "../utils/convertMessage.js";
+var useAISDKRuntime = (chatHelpers, adapter = {}) => {
+  const messages = AISDKMessageConverter.useThreadMessages({
+    isRunning: chatHelpers.status === "submitted" || chatHelpers.status == "streaming",
+    messages: chatHelpers.messages
+  });
+  const runtime = useExternalStoreRuntime({
+    isRunning: chatHelpers.status === "submitted" || chatHelpers.status === "streaming",
+    messages,
+    setMessages: (messages2) => chatHelpers.setMessages(messages2.map(getVercelAIMessages).flat()),
+    onCancel: async () => chatHelpers.stop(),
+    onNew: async (message) => {
+      await chatHelpers.sendMessage(await toCreateMessage(message));
+    },
+    onEdit: async (message) => {
+      const newMessages = sliceMessagesUntil(
+        chatHelpers.messages,
+        message.parentId
+      );
+      chatHelpers.setMessages(newMessages);
+      await chatHelpers.sendMessage(await toCreateMessage(message));
+    },
+    onReload: async (parentId) => {
+      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);
+      chatHelpers.setMessages(newMessages);
+      await chatHelpers.regenerate();
+    },
+    onAddToolResult: ({ toolCallId, result }) => {
+      chatHelpers.addToolResult({
+        tool: toolCallId,
+        toolCallId,
+        output: result
+      });
+    },
+    adapters: {
+      attachments: vercelAttachmentAdapter,
+      ...adapter.adapters
+    }
+  });
+  return runtime;
+};
+export {
+  useAISDKRuntime
+};
+//# sourceMappingURL=useAISDKRuntime.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/ui/use-chat/useAISDKRuntime.tsx"],"sourcesContent":["\"use client\";\n\nimport type { useChat } from \"@ai-sdk/react\";\nimport { useExternalStoreRuntime } from \"@assistant-ui/react\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../getVercelAIMessages\";\nimport { ExternalStoreAdapter } from \"@assistant-ui/react\";\nimport { AISDKMessageConverter } from \"../utils/convertMessage\";\n\nexport type AISDKRuntimeAdapter = {\n  adapters?:\n    | Omit<NonNullable<ExternalStoreAdapter[\"adapters\"]>, \"attachments\">\n    | undefined;\n};\n\nexport const useAISDKRuntime = (\n  chatHelpers: ReturnType<typeof useChat>,\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 runtime = useExternalStoreRuntime({\n    isRunning:\n      chatHelpers.status === \"submitted\" || chatHelpers.status === \"streaming\",\n    messages,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(messages.map(getVercelAIMessages).flat()),\n    onCancel: async () => chatHelpers.stop(),\n    onNew: async (message) => {\n      await chatHelpers.sendMessage(await toCreateMessage(message));\n    },\n    onEdit: async (message) => {\n      const newMessages = sliceMessagesUntil(\n        chatHelpers.messages,\n        message.parentId,\n      );\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.sendMessage(await toCreateMessage(message));\n    },\n    onReload: async (parentId: string | null) => {\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.regenerate();\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  });\n\n  return runtime;\n};\n"],"mappings":";;;AAGA,SAAS,+BAA+B;AACxC,SAAS,0BAA0B;AACnC,SAAS,uBAAuB;AAChC,SAAS,+BAA+B;AACxC,SAAS,2BAA2B;AAEpC,SAAS,6BAA6B;AAQ/B,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,UAAU,wBAAwB;AAAA,IACtC,WACE,YAAY,WAAW,eAAe,YAAY,WAAW;AAAA,IAC/D;AAAA,IACA,aAAa,CAACA,cACZ,YAAY,YAAYA,UAAS,IAAI,mBAAmB,EAAE,KAAK,CAAC;AAAA,IAClE,UAAU,YAAY,YAAY,KAAK;AAAA,IACvC,OAAO,OAAO,YAAY;AACxB,YAAM,YAAY,YAAY,MAAM,gBAAgB,OAAO,CAAC;AAAA,IAC9D;AAAA,IACA,QAAQ,OAAO,YAAY;AACzB,YAAM,cAAc;AAAA,QAClB,YAAY;AAAA,QACZ,QAAQ;AAAA,MACV;AACA,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,YAAY,MAAM,gBAAgB,OAAO,CAAC;AAAA,IAC9D;AAAA,IACA,UAAU,OAAO,aAA4B;AAC3C,YAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;AACrE,kBAAY,YAAY,WAAW;AAEnC,YAAM,YAAY,WAAW;AAAA,IAC/B;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,EACF,CAAC;AAED,SAAO;AACT;","names":["messages"]}

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

@@ -1,14 +1,14 @@
-import { Message } from "@ai-sdk/ui-utils";
+import { UIMessage } from "ai";
 export declare const AISDKMessageConverter: {
     useThreadMessages: ({ messages, isRunning, joinStrategy, }: {
-        messages: Message[];
+        messages: UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[];
         isRunning: boolean;
         joinStrategy?: "concat-content" | "none" | undefined;
     }) => import("@assistant-ui/react").ThreadMessage[];
-    toThreadMessages: (messages: Message[]) => import("@assistant-ui/react").ThreadMessage[];
+    toThreadMessages: (messages: UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[]) => import("@assistant-ui/react").ThreadMessage[];
     toOriginalMessages: (input: import("@assistant-ui/react").ThreadState | import("@assistant-ui/react").ThreadMessage | import("@assistant-ui/react").ThreadMessage["content"][number]) => unknown[];
     toOriginalMessage: (input: import("@assistant-ui/react").ThreadState | import("@assistant-ui/react").ThreadMessage | import("@assistant-ui/react").ThreadMessage["content"][number]) => {};
-    useOriginalMessage: () => Message;
-    useOriginalMessages: () => Message[];
+    useOriginalMessage: () => UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>;
+    useOriginalMessages: () => UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[];
 };
 //# sourceMappingURL=convertMessage.d.ts.map

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,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAsE3C,eAAO,MAAM,qBAAqB;;;;;;;0HAjDpB,sDACR;yHAES,sDACI;;;CAuIlB,CAAC"}
+{"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"}

package/dist/ui/utils/convertMessage.js

@@ -1,122 +1,151 @@
 // src/ui/utils/convertMessage.ts
+import { isToolUIPart } from "ai";
 import {
   unstable_createMessageConverter
 } from "@assistant-ui/react";
 var convertParts = (message) => {
-  if (message.parts && message.parts.length > 0) {
-    return message.parts.filter((p) => p.type !== "step-start").map((part) => {
-      const type = part.type;
-      switch (type) {
-        case "text":
-          return {
-            type: "text",
-            text: part.text
-          };
-        case "tool-invocation":
-          return {
-            type: "tool-call",
-            toolName: part.toolInvocation.toolName,
-            toolCallId: part.toolInvocation.toolCallId,
-            argsText: JSON.stringify(part.toolInvocation.args),
-            args: part.toolInvocation.args,
-            result: part.toolInvocation.state === "result" && part.toolInvocation.result
-          };
-        case "reasoning":
-          return {
-            type: "reasoning",
-            text: part.reasoning
-          };
-        case "source":
-          return {
-            type: "source",
-            ...part.source
-          };
-        case "file":
-          return {
-            type: "file",
-            data: part.data,
-            mimeType: part.mimeType
-          };
-        default: {
-          const _unsupported = type;
-          throw new Error(
-            `You have a message with an unsupported part type. The type ${_unsupported} is not supported.`
-          );
-        }
-      }
-    });
+  if (!message.parts || message.parts.length === 0) {
+    return [];
   }
-  return message.content ? [
-    {
-      type: "text",
-      text: message.content
+  return message.parts.filter((p) => p.type !== "step-start").map((part) => {
+    const type = part.type;
+    if (type === "text") {
+      return {
+        type: "text",
+        text: part.text
+      };
+    }
+    if (type === "reasoning") {
+      return {
+        type: "reasoning",
+        text: part.text
+      };
+    }
+    if (isToolUIPart(part)) {
+      const toolName = type.replace("tool-", "");
+      const toolCallId = part.toolCallId;
+      let args = {};
+      let result = void 0;
+      let isError = false;
+      if (part.state === "input-streaming" || part.state === "input-available") {
+        args = part.input || {};
+      } else if (part.state === "output-available") {
+        args = part.input || {};
+        result = part.output;
+      } else if (part.state === "output-error") {
+        args = part.input || {};
+        isError = true;
+        result = { error: part.errorText };
+      }
+      return {
+        type: "tool-call",
+        toolName,
+        toolCallId,
+        argsText: JSON.stringify(args),
+        args,
+        result,
+        isError
+      };
+    }
+    if (type === "dynamic-tool") {
+      const toolName = part.toolName;
+      const toolCallId = part.toolCallId;
+      let args = {};
+      let result = void 0;
+      let isError = false;
+      if (part.state === "input-streaming" || part.state === "input-available") {
+        args = part.input || {};
+      } else if (part.state === "output-available") {
+        args = part.input || {};
+        result = part.output;
+      } else if (part.state === "output-error") {
+        args = part.input || {};
+        isError = true;
+        result = { error: part.errorText };
+      }
+      return {
+        type: "tool-call",
+        toolName,
+        toolCallId,
+        argsText: JSON.stringify(args),
+        args,
+        result,
+        isError
+      };
+    }
+    if (type === "source-url") {
+      return {
+        type: "source",
+        sourceType: "url",
+        id: part.sourceId,
+        url: part.url,
+        title: part.title || ""
+      };
     }
-  ] : [];
+    if (type === "source-document") {
+      console.warn(
+        `Source document part type ${type} is not yet supported in conversion`
+      );
+      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`
+      );
+      return null;
+    }
+    console.warn(`Unsupported message part type: ${type}`);
+    return null;
+  }).filter(Boolean);
 };
 var AISDKMessageConverter = unstable_createMessageConverter(
   (message) => {
+    const createdAt = /* @__PURE__ */ new Date();
     switch (message.role) {
       case "user":
         return {
           role: "user",
           id: message.id,
-          createdAt: message.createdAt,
+          createdAt,
           content: convertParts(message),
-          attachments: message.experimental_attachments?.map(
-            (attachment, idx) => ({
-              id: idx.toString(),
-              type: "file",
-              name: attachment.name ?? attachment.url,
-              content: [],
-              contentType: attachment.contentType ?? "unknown/unknown",
-              status: { type: "complete" }
-            })
-          )
+          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" }
+          }))
         };
       case "system":
         return {
           role: "system",
           id: message.id,
-          createdAt: message.createdAt,
+          createdAt,
           content: convertParts(message)
         };
       case "assistant":
         return {
           role: "assistant",
           id: message.id,
-          createdAt: message.createdAt,
+          createdAt,
           content: convertParts(message),
           metadata: {
             unstable_annotations: message.annotations,
-            unstable_data: Array.isArray(message.data) ? message.data : message.data ? [message.data] : void 0
+            unstable_data: Array.isArray(message.data) ? message.data : message.data ? [message.data] : void 0,
+            custom: {}
           }
         };
-      case "data": {
-        if (!message.data || !(typeof message.data === "object") || Array.isArray(message.data))
-          return [];
-        const data = message.data;
-        if (data.type === "tool-call") {
-          return {
-            role: "assistant",
-            id: message.id,
-            createdAt: message.createdAt,
-            content: [data]
-          };
-        } else if (data.type === "tool-result") {
-          return {
-            role: "tool",
-            id: message.id,
-            toolCallId: data.toolCallId,
-            result: data.result
-          };
-        }
-        return [];
-      }
       default:
-        const _unsupported = message.role;
-        throw new Error(
-          `You have a message with an unsupported role. The role ${_unsupported} is not supported.`
-        );
+        console.warn(`Unsupported message role: ${message.role}`);
+        return [];
     }
   }
 );

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/ui/utils/convertMessage.ts"],"sourcesContent":["import { Message } from \"@ai-sdk/ui-utils\";\nimport {\n  unstable_createMessageConverter,\n  type ReasoningMessagePart,\n  type ToolCallMessagePart,\n  type TextMessagePart,\n  type CompleteAttachment,\n  type SourceMessagePart,\n  type FileMessagePart,\n} from \"@assistant-ui/react\";\n\nconst convertParts = (message: Message) => {\n  if (message.parts && message.parts.length > 0) {\n    return message.parts\n      .filter((p) => p.type !== \"step-start\")\n      .map((part) => {\n        const type = part.type;\n        switch (type) {\n          case \"text\":\n            return {\n              type: \"text\",\n              text: part.text,\n            } satisfies TextMessagePart;\n          case \"tool-invocation\":\n            return {\n              type: \"tool-call\",\n              toolName: part.toolInvocation.toolName,\n              toolCallId: part.toolInvocation.toolCallId,\n              argsText: JSON.stringify(part.toolInvocation.args),\n              args: part.toolInvocation.args,\n              result:\n                part.toolInvocation.state === \"result\" &&\n                part.toolInvocation.result,\n            } satisfies ToolCallMessagePart;\n          case \"reasoning\":\n            return {\n              type: \"reasoning\",\n              text: part.reasoning,\n            } satisfies ReasoningMessagePart;\n          case \"source\":\n            return {\n              type: \"source\",\n              ...part.source,\n            } satisfies SourceMessagePart;\n          case \"file\":\n            return {\n              type: \"file\",\n              data: part.data,\n              mimeType: part.mimeType,\n            } satisfies FileMessagePart;\n          default: {\n            const _unsupported: never = type;\n            throw new Error(\n              `You have a message with an unsupported part type. The type ${_unsupported} is not supported.`,\n            );\n          }\n        }\n      });\n  }\n\n  return message.content\n    ? [\n        {\n          type: \"text\",\n          text: message.content,\n        } satisfies TextMessagePart,\n      ]\n    : [];\n};\n\nexport const AISDKMessageConverter = unstable_createMessageConverter(\n  (message: Message) => {\n    switch (message.role) {\n      case \"user\":\n        return {\n          role: \"user\",\n          id: message.id,\n          createdAt: message.createdAt,\n          content: convertParts(message),\n          attachments: message.experimental_attachments?.map(\n            (attachment, idx) =>\n              ({\n                id: idx.toString(),\n                type: \"file\",\n                name: attachment.name ?? attachment.url,\n                content: [],\n                contentType: attachment.contentType ?? \"unknown/unknown\",\n                status: { type: \"complete\" },\n              }) satisfies CompleteAttachment,\n          ),\n        };\n\n      case \"system\":\n        return {\n          role: \"system\",\n          id: message.id,\n          createdAt: message.createdAt,\n          content: convertParts(message),\n        };\n\n      case \"assistant\":\n        return {\n          role: \"assistant\",\n          id: message.id,\n          createdAt: message.createdAt,\n          content: convertParts(message),\n          metadata: {\n            unstable_annotations: message.annotations,\n            unstable_data: Array.isArray(message.data)\n              ? message.data\n              : message.data\n                ? [message.data]\n                : undefined,\n          },\n        };\n\n      case \"data\": {\n        type MaybeSupportedDataMessage =\n          | { type?: \"unsafe_other\" }\n          | ToolCallMessagePart\n          | {\n              type: \"tool-result\";\n              toolCallId: string;\n              result: any;\n            };\n\n        if (\n          !message.data ||\n          !(typeof message.data === \"object\") ||\n          Array.isArray(message.data)\n        )\n          return [];\n\n        const data = message.data as MaybeSupportedDataMessage;\n\n        if (data.type === \"tool-call\") {\n          return {\n            role: \"assistant\",\n            id: message.id,\n            createdAt: message.createdAt,\n            content: [data],\n          };\n        } else if (data.type === \"tool-result\") {\n          return {\n            role: \"tool\",\n            id: message.id,\n            toolCallId: data.toolCallId,\n            result: data.result,\n          };\n        }\n        return [];\n      }\n\n      default:\n        const _unsupported: \"function\" | \"tool\" = message.role;\n        throw new Error(\n          `You have a message with an unsupported role. The role ${_unsupported} is not supported.`,\n        );\n    }\n  },\n);\n"],"mappings":";AACA;AAAA,EACE;AAAA,OAOK;AAEP,IAAM,eAAe,CAAC,YAAqB;AACzC,MAAI,QAAQ,SAAS,QAAQ,MAAM,SAAS,GAAG;AAC7C,WAAO,QAAQ,MACZ,OAAO,CAAC,MAAM,EAAE,SAAS,YAAY,EACrC,IAAI,CAAC,SAAS;AACb,YAAM,OAAO,KAAK;AAClB,cAAQ,MAAM;AAAA,QACZ,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,MAAM,KAAK;AAAA,UACb;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,UAAU,KAAK,eAAe;AAAA,YAC9B,YAAY,KAAK,eAAe;AAAA,YAChC,UAAU,KAAK,UAAU,KAAK,eAAe,IAAI;AAAA,YACjD,MAAM,KAAK,eAAe;AAAA,YAC1B,QACE,KAAK,eAAe,UAAU,YAC9B,KAAK,eAAe;AAAA,UACxB;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,MAAM,KAAK;AAAA,UACb;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,GAAG,KAAK;AAAA,UACV;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,MAAM,KAAK;AAAA,YACX,UAAU,KAAK;AAAA,UACjB;AAAA,QACF,SAAS;AACP,gBAAM,eAAsB;AAC5B,gBAAM,IAAI;AAAA,YACR,8DAA8D,YAAY;AAAA,UAC5E;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACL;AAEA,SAAO,QAAQ,UACX;AAAA,IACE;AAAA,MACE,MAAM;AAAA,MACN,MAAM,QAAQ;AAAA,IAChB;AAAA,EACF,IACA,CAAC;AACP;AAEO,IAAM,wBAAwB;AAAA,EACnC,CAAC,YAAqB;AACpB,YAAQ,QAAQ,MAAM;AAAA,MACpB,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ,WAAW,QAAQ;AAAA,UACnB,SAAS,aAAa,OAAO;AAAA,UAC7B,aAAa,QAAQ,0BAA0B;AAAA,YAC7C,CAAC,YAAY,SACV;AAAA,cACC,IAAI,IAAI,SAAS;AAAA,cACjB,MAAM;AAAA,cACN,MAAM,WAAW,QAAQ,WAAW;AAAA,cACpC,SAAS,CAAC;AAAA,cACV,aAAa,WAAW,eAAe;AAAA,cACvC,QAAQ,EAAE,MAAM,WAAW;AAAA,YAC7B;AAAA,UACJ;AAAA,QACF;AAAA,MAEF,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ,WAAW,QAAQ;AAAA,UACnB,SAAS,aAAa,OAAO;AAAA,QAC/B;AAAA,MAEF,KAAK;AACH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,IAAI,QAAQ;AAAA,UACZ,WAAW,QAAQ;AAAA,UACnB,SAAS,aAAa,OAAO;AAAA,UAC7B,UAAU;AAAA,YACR,sBAAsB,QAAQ;AAAA,YAC9B,eAAe,MAAM,QAAQ,QAAQ,IAAI,IACrC,QAAQ,OACR,QAAQ,OACN,CAAC,QAAQ,IAAI,IACb;AAAA,UACR;AAAA,QACF;AAAA,MAEF,KAAK,QAAQ;AAUX,YACE,CAAC,QAAQ,QACT,EAAE,OAAO,QAAQ,SAAS,aAC1B,MAAM,QAAQ,QAAQ,IAAI;AAE1B,iBAAO,CAAC;AAEV,cAAM,OAAO,QAAQ;AAErB,YAAI,KAAK,SAAS,aAAa;AAC7B,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,IAAI,QAAQ;AAAA,YACZ,WAAW,QAAQ;AAAA,YACnB,SAAS,CAAC,IAAI;AAAA,UAChB;AAAA,QACF,WAAW,KAAK,SAAS,eAAe;AACtC,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,IAAI,QAAQ;AAAA,YACZ,YAAY,KAAK;AAAA,YACjB,QAAQ,KAAK;AAAA,UACf;AAAA,QACF;AACA,eAAO,CAAC;AAAA,MACV;AAAA,MAEA;AACE,cAAM,eAAoC,QAAQ;AAClD,cAAM,IAAI;AAAA,UACR,yDAAyD,YAAY;AAAA,QACvE;AAAA,IACJ;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  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":[]}

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

@@ -1,3 +1,3 @@
-import type { Message } from "@ai-sdk/ui-utils";
-export declare const sliceMessagesUntil: (messages: Message[], messageId: string | null) => Message[];
+import type { UIMessage } from "ai";
+export declare const sliceMessagesUntil: (messages: UIMessage[], messageId: string | null) => UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[];
 //# sourceMappingURL=sliceMessagesUntil.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"sliceMessagesUntil.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/sliceMessagesUntil.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAEhD,eAAO,MAAM,kBAAkB,GAC7B,UAAU,OAAO,EAAE,EACnB,WAAW,MAAM,GAAG,IAAI,cAezB,CAAC"}
+{"version":3,"file":"sliceMessagesUntil.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/sliceMessagesUntil.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAEpC,eAAO,MAAM,kBAAkB,GAC7B,UAAU,SAAS,EAAE,EACrB,WAAW,MAAM,GAAG,IAAI,yEAezB,CAAC"}

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/ui/utils/sliceMessagesUntil.tsx"],"sourcesContent":["import type { Message } from \"@ai-sdk/ui-utils\";\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 likely 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"],"mappings":";AAEO,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;","names":[]}
+{"version":3,"sources":["../../../src/ui/utils/sliceMessagesUntil.tsx"],"sourcesContent":["import type { UIMessage } from \"ai\";\n\nexport const sliceMessagesUntil = (\n  messages: UIMessage[],\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 likely 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"],"mappings":";AAEO,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;","names":[]}

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

@@ -1,4 +1,4 @@
 import { AppendMessage } from "@assistant-ui/react";
-import { CreateMessage } from "@ai-sdk/ui-utils";
-export declare const toCreateMessage: (message: AppendMessage) => Promise<CreateMessage>;
+import { CreateUIMessage, UIMessage } from "ai";
+export declare const toCreateMessage: (message: AppendMessage) => Promise<CreateUIMessage<UIMessage>>;
 //# 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,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEjD,eAAO,MAAM,eAAe,GAC1B,SAAS,aAAa,KACrB,OAAO,CAAC,aAAa,CA4BvB,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,GAC1B,SAAS,aAAa,KACrB,OAAO,CAAC,eAAe,CAAC,SAAS,CAAC,CA8CpC,CAAC"}

package/dist/ui/utils/toCreateMessage.js

@@ -1,24 +1,36 @@
 // src/ui/utils/toCreateMessage.ts
 var toCreateMessage = async (message) => {
-  const content = message.content.filter((part) => part.type === "text").map((t) => t.text).join("\n\n");
-  const images = message.content.filter((part) => part.type === "image").map((part) => ({ url: part.image }));
+  const textParts = message.content.filter((part) => part.type === "text").map((t) => t.text).join("\n\n");
+  const parts = [
+    {
+      type: "text",
+      text: textParts
+    }
+  ];
+  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);
   return {
     role: message.role,
-    content,
-    experimental_attachments: [
-      ...images,
-      ...await Promise.all(
-        (message.attachments ?? []).map(async (m) => {
-          if (m.file == null)
-            throw new Error("Attachment did not contain a file");
-          return {
-            contentType: m.file.type,
-            name: m.file.name,
-            url: await getFileDataURL(m.file)
-          };
-        })
-      )
-    ]
+    parts
   };
 };
 var getFileDataURL = (file) => new Promise((resolve, reject) => {