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

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 { CreateMessage } from \"@ai-sdk/ui-utils\";\n\nexport const toCreateMessage = async (\n  message: AppendMessage,\n): Promise<CreateMessage> => {\n  const content = message.content\n    .filter((part) => part.type === \"text\")\n    .map((t) => t.text)\n    .join(\"\\n\\n\");\n\n  const images = message.content\n    .filter((part) => part.type === \"image\")\n    .map((part) => ({ url: part.image }));\n\n  return {\n    role: message.role,\n    content,\n    experimental_attachments: [\n      ...images,\n      ...(await Promise.all(\n        (message.attachments ?? []).map(async (m) => {\n          if (m.file == null)\n            throw new Error(\"Attachment did not contain a file\");\n          return {\n            contentType: m.file.type,\n            name: m.file.name,\n            url: await getFileDataURL(m.file),\n          };\n        }),\n      )),\n    ],\n  };\n};\n\nconst getFileDataURL = (file: File) =>\n  new Promise<string>((resolve, reject) => {\n    const reader = new FileReader();\n\n    reader.onload = () => resolve(reader.result as string);\n    reader.onerror = (error) => reject(error);\n\n    reader.readAsDataURL(file);\n  });\n"],"mappings":";AAGO,IAAM,kBAAkB,OAC7B,YAC2B;AAC3B,QAAM,UAAU,QAAQ,QACrB,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,MAAM,EAAE,IAAI,EACjB,KAAK,MAAM;AAEd,QAAM,SAAS,QAAQ,QACpB,OAAO,CAAC,SAAS,KAAK,SAAS,OAAO,EACtC,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,MAAM,EAAE;AAEtC,SAAO;AAAA,IACL,MAAM,QAAQ;AAAA,IACd;AAAA,IACA,0BAA0B;AAAA,MACxB,GAAG;AAAA,MACH,GAAI,MAAM,QAAQ;AAAA,SACf,QAAQ,eAAe,CAAC,GAAG,IAAI,OAAO,MAAM;AAC3C,cAAI,EAAE,QAAQ;AACZ,kBAAM,IAAI,MAAM,mCAAmC;AACrD,iBAAO;AAAA,YACL,aAAa,EAAE,KAAK;AAAA,YACpB,MAAM,EAAE,KAAK;AAAA,YACb,KAAK,MAAM,eAAe,EAAE,IAAI;AAAA,UAClC;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,iBAAiB,CAAC,SACtB,IAAI,QAAgB,CAAC,SAAS,WAAW;AACvC,QAAM,SAAS,IAAI,WAAW;AAE9B,SAAO,SAAS,MAAM,QAAQ,OAAO,MAAgB;AACrD,SAAO,UAAU,CAAC,UAAU,OAAO,KAAK;AAExC,SAAO,cAAc,IAAI;AAC3B,CAAC;","names":[]}
+{"version":3,"sources":["../../../src/ui/utils/toCreateMessage.ts"],"sourcesContent":["import { AppendMessage } from \"@assistant-ui/react\";\nimport {\n  CreateUIMessage,\n  FileUIPart,\n  UIDataTypes,\n  UIMessage,\n  UIMessagePart,\n  UITools,\n} from \"ai\";\n\nexport const toCreateMessage = async (\n  message: AppendMessage,\n): Promise<CreateUIMessage<UIMessage>> => {\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    role: message.role,\n    parts,\n  };\n};\n\nconst getFileDataURL = (file: File) =>\n  new Promise<string>((resolve, reject) => {\n    const reader = new FileReader();\n\n    reader.onload = () => resolve(reader.result as string);\n    reader.onerror = (error) => reject(error);\n\n    reader.readAsDataURL(file);\n  });\n"],"mappings":";AAUO,IAAM,kBAAkB,OAC7B,YACwC;AACxC,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,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":[]}

package/dist/ui/utils/vercelAttachmentAdapter.js

@@ -1,5 +1,5 @@
 // src/ui/utils/vercelAttachmentAdapter.ts
-import { generateId } from "@ai-sdk/ui-utils";
+import { generateId } from "ai";
 var vercelAttachmentAdapter = {
   accept: "image/*, text/plain, text/html, text/markdown, text/csv, text/xml, text/json, text/css",
   async add({ file }) {

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-sdk/ui-utils\";\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\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":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ai-sdk",
-  "version": "0.10.15",
+  "version": "0.11.0-alpha.0-alpha",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -18,16 +18,19 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@ai-sdk/react": "*",
-    "@ai-sdk/ui-utils": "*",
+    "@ai-sdk/provider": "^2.0.0",
+    "@ai-sdk/react": "^2.0.0",
+    "ai": "^5.0.0",
     "@radix-ui/react-use-callback-ref": "^1.1.1",
     "@types/json-schema": "^7.0.15",
-    "zod": "^3.25.67",
-    "zustand": "^5.0.6",
-    "@assistant-ui/react-edge": "0.2.13"
+    "assistant-stream": "^0.2.21",
+    "json-schema": "^0.4.0",
+    "zod": "^4.0.14",
+    "zod-to-json-schema": "^3.24.6",
+    "zustand": "^5.0.7"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.10.25",
+    "@assistant-ui/react": "^0.10.33",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc"
   },
@@ -37,12 +40,13 @@
     }
   },
   "devDependencies": {
-    "@types/node": "^24.0.4",
-    "@types/react": "^19.1.8",
+    "@types/node": "^24.1.0",
+    "@types/react": "^19.1.9",
     "eslint": "^9",
-    "eslint-config-next": "15.3.4",
-    "react": "^19.1.0",
+    "eslint-config-next": "15.4.5",
+    "react": "19.1.0",
     "tsx": "^4.20.3",
+    "@assistant-ui/react": "0.10.33",
     "@assistant-ui/x-buildutils": "0.0.1"
   },
   "publishConfig": {
@@ -51,7 +55,7 @@
   },
   "repository": {
     "type": "git",
-    "url": "https://github.com/assistant-ui/assistant-ui/tree/main/packages/react-ai-sdk"
+    "url": "https://github.com/assistant-ui/assistant-ui/tree/main/packages/react-ai-sdk-v5"
   },
   "bugs": {
     "url": "https://github.com/assistant-ui/assistant-ui/issues"

package/src/frontendTools.ts

@@ -1,4 +1,4 @@
-import { jsonSchema } from "@ai-sdk/ui-utils";
+import { jsonSchema } from "ai";
 import type { JSONSchema7 } from "json-schema";
 
 export const frontendTools = (

package/src/index.ts

@@ -1,13 +1,2 @@
-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";

package/src/ui/getVercelAIMessages.tsx

@@ -2,8 +2,8 @@ import {
   getExternalStoreMessages,
   type ThreadMessage,
 } from "@assistant-ui/react";
-import type { Message } from "@ai-sdk/ui-utils";
+import type { UIMessage } from "ai";
 
 export const getVercelAIMessages = (message: ThreadMessage) => {
-  return getExternalStoreMessages(message) as Message[];
+  return getExternalStoreMessages(message) as UIMessage[];
 };

package/src/ui/index.ts

@@ -1,4 +1 @@
-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";

package/src/ui/use-chat/{useVercelUseChatRuntime.tsx → useAISDKRuntime.tsx}

@@ -2,44 +2,38 @@
 
 import type { useChat } from "@ai-sdk/react";
 import { useExternalStoreRuntime } from "@assistant-ui/react";
-import { useInputSync } from "../utils/useInputSync";
 import { sliceMessagesUntil } from "../utils/sliceMessagesUntil";
 import { toCreateMessage } from "../utils/toCreateMessage";
 import { vercelAttachmentAdapter } from "../utils/vercelAttachmentAdapter";
 import { getVercelAIMessages } from "../getVercelAIMessages";
 import { ExternalStoreAdapter } from "@assistant-ui/react";
-import { useState } from "react";
-import { generateId } from "@ai-sdk/ui-utils";
 import { AISDKMessageConverter } from "../utils/convertMessage";
 
-export type VercelUseChatAdapter = {
+export type AISDKRuntimeAdapter = {
   adapters?:
     | Omit<NonNullable<ExternalStoreAdapter["adapters"]>, "attachments">
     | undefined;
-  unstable_joinStrategy?: "concat-content" | "none";
 };
 
-export const useVercelUseChatRuntime = (
+export const useAISDKRuntime = (
   chatHelpers: ReturnType<typeof useChat>,
-  adapter: VercelUseChatAdapter = {},
+  adapter: AISDKRuntimeAdapter = {},
 ) => {
   const messages = AISDKMessageConverter.useThreadMessages({
     isRunning:
       chatHelpers.status === "submitted" || chatHelpers.status == "streaming",
     messages: chatHelpers.messages,
-    joinStrategy: adapter.unstable_joinStrategy,
   });
 
-  const [threadId, setThreadId] = useState<string>(generateId());
-
   const runtime = useExternalStoreRuntime({
-    isRunning: chatHelpers.isLoading,
+    isRunning:
+      chatHelpers.status === "submitted" || chatHelpers.status === "streaming",
     messages,
     setMessages: (messages) =>
       chatHelpers.setMessages(messages.map(getVercelAIMessages).flat()),
     onCancel: async () => chatHelpers.stop(),
     onNew: async (message) => {
-      await chatHelpers.append(await toCreateMessage(message));
+      await chatHelpers.sendMessage(await toCreateMessage(message));
     },
     onEdit: async (message) => {
       const newMessages = sliceMessagesUntil(
@@ -48,46 +42,26 @@ export const useVercelUseChatRuntime = (
       );
       chatHelpers.setMessages(newMessages);
 
-      await chatHelpers.append(await toCreateMessage(message));
+      await chatHelpers.sendMessage(await toCreateMessage(message));
     },
     onReload: async (parentId: string | null) => {
       const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);
       chatHelpers.setMessages(newMessages);
 
-      await chatHelpers.reload();
+      await chatHelpers.regenerate();
     },
     onAddToolResult: ({ toolCallId, result }) => {
-      chatHelpers.addToolResult({ toolCallId, result });
+      chatHelpers.addToolResult({
+        tool: toolCallId,
+        toolCallId,
+        output: result,
+      });
     },
     adapters: {
       attachments: vercelAttachmentAdapter,
       ...adapter.adapters,
-      threadList: new Proxy(adapter.adapters?.threadList ?? {}, {
-        get(target, prop, receiver) {
-          if (prop === "threadId") {
-            return target.threadId ?? threadId;
-          }
-          if (prop === "onSwitchToNewThread") {
-            return () => {
-              chatHelpers.messages = [];
-              chatHelpers.input = "";
-              chatHelpers.setMessages([]);
-              chatHelpers.setInput("");
-              setThreadId(generateId());
-
-              if (typeof target.onSwitchToNewThread === "function") {
-                return target.onSwitchToNewThread.call(target);
-              }
-            };
-          }
-
-          return Reflect.get(target, prop, receiver);
-        },
-      }),
     },
   });
 
-  useInputSync(chatHelpers, runtime);
-
   return runtime;
 };

package/src/ui/utils/convertMessage.ts

@@ -1,100 +1,183 @@
-import { Message } from "@ai-sdk/ui-utils";
+import { isToolUIPart, UIMessage } from "ai";
 import {
   unstable_createMessageConverter,
   type ReasoningMessagePart,
   type ToolCallMessagePart,
   type TextMessagePart,
-  type CompleteAttachment,
   type SourceMessagePart,
   type FileMessagePart,
 } from "@assistant-ui/react";
 
-const convertParts = (message: 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,
-            } satisfies TextMessagePart;
-          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,
-            } satisfies ToolCallMessagePart;
-          case "reasoning":
-            return {
-              type: "reasoning",
-              text: part.reasoning,
-            } satisfies ReasoningMessagePart;
-          case "source":
-            return {
-              type: "source",
-              ...part.source,
-            } satisfies SourceMessagePart;
-          case "file":
-            return {
-              type: "file",
-              data: part.data,
-              mimeType: part.mimeType,
-            } satisfies FileMessagePart;
-          default: {
-            const _unsupported: never = type;
-            throw new Error(
-              `You have a message with an unsupported part type. The type ${_unsupported} is not supported.`,
-            );
-          }
-        }
-      });
+const convertParts = (message: UIMessage) => {
+  if (!message.parts || message.parts.length === 0) {
+    return [];
   }
 
-  return message.content
-    ? [
-        {
+  return message.parts
+    .filter((p) => p.type !== "step-start")
+    .map((part) => {
+      const type = part.type;
+
+      // Handle text parts
+      if (type === "text") {
+        return {
           type: "text",
-          text: message.content,
-        } satisfies TextMessagePart,
-      ]
-    : [];
+          text: part.text,
+        } satisfies TextMessagePart;
+      }
+
+      // Handle reasoning parts
+      if (type === "reasoning") {
+        return {
+          type: "reasoning",
+          text: part.text,
+        } satisfies ReasoningMessagePart;
+      }
+
+      // Handle tool-* parts (AI SDK v5 tool calls)
+      if (isToolUIPart(part)) {
+        const toolName = type.replace("tool-", "");
+        const toolCallId = part.toolCallId;
+
+        // Extract args and result based on state
+        let args: any = {};
+        let result: any = undefined;
+        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,
+        } satisfies ToolCallMessagePart;
+      }
+
+      // Handle dynamic-tool parts
+      if (type === "dynamic-tool") {
+        const toolName = part.toolName;
+        const toolCallId = part.toolCallId;
+
+        // Extract args and result based on state
+        let args: any = {};
+        let result: any = undefined;
+        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,
+        } satisfies ToolCallMessagePart;
+      }
+
+      // Handle source-url parts
+      if (type === "source-url") {
+        return {
+          type: "source",
+          sourceType: "url",
+          id: part.sourceId,
+          url: part.url,
+          title: part.title || "",
+        } satisfies SourceMessagePart;
+      }
+
+      // Handle source-document parts
+      if (type === "source-document") {
+        console.warn(
+          `Source document part type ${type} is not yet supported in conversion`,
+        );
+        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
+        // in the assistant-ui types. They could be converted to a custom message part
+        // or handled differently based on the specific use case.
+        console.warn(
+          `Data part type ${type} is not yet supported in conversion`,
+        );
+        return null;
+      }
+
+      // For unsupported types, we'll skip them instead of throwing
+      console.warn(`Unsupported message part type: ${type}`);
+      return null;
+    })
+    .filter(Boolean) as any[];
 };
 
 export const AISDKMessageConverter = unstable_createMessageConverter(
-  (message: Message) => {
+  (message: UIMessage) => {
+    // UIMessage doesn't have createdAt, so we'll use current date or undefined
+    const createdAt = 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" },
-              }) satisfies CompleteAttachment,
-          ),
+          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 },
+            })),
         };
 
       case "system":
         return {
           role: "system",
           id: message.id,
-          createdAt: message.createdAt,
+          createdAt,
           content: convertParts(message),
         };
 
@@ -102,60 +185,22 @@ export const AISDKMessageConverter = unstable_createMessageConverter(
         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]
+            unstable_annotations: (message as any).annotations,
+            unstable_data: Array.isArray((message as any).data)
+              ? (message as any).data
+              : (message as any).data
+                ? [(message as any).data]
                 : undefined,
+            custom: {},
           },
         };
 
-      case "data": {
-        type MaybeSupportedDataMessage =
-          | { type?: "unsafe_other" }
-          | ToolCallMessagePart
-          | {
-              type: "tool-result";
-              toolCallId: string;
-              result: any;
-            };
-
-        if (
-          !message.data ||
-          !(typeof message.data === "object") ||
-          Array.isArray(message.data)
-        )
-          return [];
-
-        const data = message.data as MaybeSupportedDataMessage;
-
-        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: "function" | "tool" = 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/src/ui/utils/sliceMessagesUntil.tsx

@@ -1,7 +1,7 @@
-import type { Message } from "@ai-sdk/ui-utils";
+import type { UIMessage } from "ai";
 
 export const sliceMessagesUntil = (
-  messages: Message[],
+  messages: UIMessage[],
   messageId: string | null,
 ) => {
   if (messageId == null) return [];

package/src/ui/utils/toCreateMessage.ts

@@ -1,35 +1,60 @@
 import { AppendMessage } from "@assistant-ui/react";
-import { CreateMessage } from "@ai-sdk/ui-utils";
+import {
+  CreateUIMessage,
+  FileUIPart,
+  UIDataTypes,
+  UIMessage,
+  UIMessagePart,
+  UITools,
+} from "ai";
 
 export const toCreateMessage = async (
   message: AppendMessage,
-): Promise<CreateMessage> => {
-  const content = message.content
+): Promise<CreateUIMessage<UIMessage>> => {
+  const textParts = message.content
     .filter((part) => part.type === "text")
     .map((t) => t.text)
     .join("\n\n");
 
-  const images = message.content
+  const parts: UIMessagePart<UIDataTypes, UITools>[] = [
+    {
+      type: "text",
+      text: textParts,
+    },
+  ];
+
+  // Add image parts
+  const imageParts = message.content
     .filter((part) => part.type === "image")
-    .map((part) => ({ url: part.image }));
+    .map(
+      (part) =>
+        ({
+          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);
 
   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,
   };
 };
 

package/src/ui/utils/vercelAttachmentAdapter.ts

@@ -1,5 +1,5 @@
 import { AttachmentAdapter } from "@assistant-ui/react";
-import { generateId } from "@ai-sdk/ui-utils";
+import { generateId } from "ai";
 
 export const vercelAttachmentAdapter: AttachmentAdapter = {
   accept:

package/dist/rsc/RSCDisplay.d.ts

@@ -1,3 +0,0 @@
-import { FC } from "react";
-export declare const RSCDisplay: FC;
-//# sourceMappingURL=RSCDisplay.d.ts.map

package/dist/rsc/RSCDisplay.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"RSCDisplay.d.ts","sourceRoot":"","sources":["../../src/rsc/RSCDisplay.tsx"],"names":[],"mappings":"AAUA,OAAO,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC;AAE3B,eAAO,MAAM,UAAU,EAAE,EAUxB,CAAC"}