@assistant-ui/ai-sdk 0.0.1

package/dist/utils/getVercelAIMessages.d.ts

@@ -0,0 +1,7 @@
+import { UIMessage } from "ai";
+import { ThreadMessage } from "@assistant-ui/core";
+//#region src/utils/getVercelAIMessages.d.ts
+declare const getVercelAIMessages: <UI_MESSAGE extends UIMessage = UIMessage>(message: ThreadMessage) => UI_MESSAGE[];
+//#endregion
+export { getVercelAIMessages };
+//# sourceMappingURL=getVercelAIMessages.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"getVercelAIMessages.d.ts","names":[],"sources":["../../src/utils/getVercelAIMessages.ts"],"mappings":";;;cAMa,sBAAuB,mBAAmB,YAAY,WACjE,SAAS,kBAEmC"}

package/dist/utils/getVercelAIMessages.js

@@ -0,0 +1,9 @@
+import { getExternalStoreMessages } from "@assistant-ui/core";
+//#region src/utils/getVercelAIMessages.ts
+const getVercelAIMessages = (message) => {
+	return getExternalStoreMessages(message);
+};
+//#endregion
+export { getVercelAIMessages };
+
+//# sourceMappingURL=getVercelAIMessages.js.map

package/dist/utils/getVercelAIMessages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getVercelAIMessages.js","names":["getExternalStoreMessages","ThreadMessage","UIMessage","getVercelAIMessages","message","UI_MESSAGE"],"sources":["../../src/utils/getVercelAIMessages.ts"],"sourcesContent":["import {\n  getExternalStoreMessages,\n  type ThreadMessage,\n} from \"@assistant-ui/core\";\nimport type { UIMessage } from \"ai\";\n\nexport const getVercelAIMessages = <UI_MESSAGE extends UIMessage = UIMessage>(\n  message: ThreadMessage,\n) => {\n  return getExternalStoreMessages(message) as UI_MESSAGE[];\n};\n"],"mappings":";;AAMA,MAAaG,uBACXC,YACG;CACH,OAAOJ,yBAAyBI,OAAO;AACzC"}

package/dist/utils/sliceMessagesUntil.d.ts

@@ -0,0 +1,6 @@
+import { UIMessage } from "ai";
+//#region src/utils/sliceMessagesUntil.d.ts
+declare const sliceMessagesUntil: <UI_MESSAGE extends UIMessage = UIMessage>(messages: UI_MESSAGE[], messageId: string | null) => UI_MESSAGE[];
+//#endregion
+export { sliceMessagesUntil };
+//# sourceMappingURL=sliceMessagesUntil.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"sliceMessagesUntil.d.ts","names":[],"sources":["../../src/utils/sliceMessagesUntil.ts"],"mappings":";;cAEa,qBAAsB,mBAAmB,YAAY,WAChE,UAAU,cACV,6BAAwB"}

package/dist/utils/sliceMessagesUntil.js

@@ -0,0 +1,12 @@
+//#region src/utils/sliceMessagesUntil.ts
+const sliceMessagesUntil = (messages, messageId) => {
+	if (messageId == null) return [];
+	let messageIdx = messages.findIndex((m) => m.id === messageId);
+	if (messageIdx === -1) throw new Error(`sliceMessagesUntil: Message "${messageId}" not found in AI SDK messages. This is likely an internal bug in assistant-ui.`);
+	while (messages[messageIdx + 1]?.role === "assistant") messageIdx++;
+	return messages.slice(0, messageIdx + 1);
+};
+//#endregion
+export { sliceMessagesUntil };
+
+//# sourceMappingURL=sliceMessagesUntil.js.map

package/dist/utils/sliceMessagesUntil.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sliceMessagesUntil.js","names":["UIMessage","sliceMessagesUntil","messages","UI_MESSAGE","messageId","messageIdx","findIndex","m","id","Error","role","slice"],"sources":["../../src/utils/sliceMessagesUntil.ts"],"sourcesContent":["import type { UIMessage } from \"ai\";\n\nexport const sliceMessagesUntil = <UI_MESSAGE extends UIMessage = UIMessage>(\n  messages: UI_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      `sliceMessagesUntil: Message \"${messageId}\" not found in AI SDK messages. 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":";AAEA,MAAaC,sBACXC,UACAE,cACG;CACH,IAAIA,aAAa,MAAM,OAAO,CAAA;CAE9B,IAAIC,aAAaH,SAASI,WAAWC,MAAMA,EAAEC,OAAOJ,SAAS;CAC7D,IAAIC,eAAe,IACjB,MAAM,IAAII,MACR,gCAAgCL,UAAS,gFAC3C;CAEF,OAAOF,SAASG,aAAa,EAAE,EAAEK,SAAS,aACxCL;CAGF,OAAOH,SAASS,MAAM,GAAGN,aAAa,CAAC;AACzC"}

package/package.json

@@ -0,0 +1,100 @@
+{
+  "name": "@assistant-ui/ai-sdk",
+  "version": "0.0.1",
+  "description": "Vercel AI SDK adapter for assistant-ui",
+  "keywords": [
+    "ai-sdk",
+    "vercel",
+    "assistant-ui",
+    "react",
+    "ai",
+    "chat",
+    "openai",
+    "streaming"
+  ],
+  "author": "AgentbaseAI Inc.",
+  "license": "MIT",
+  "type": "module",
+  "imports": {
+    "#mcp-stdio": {
+      "types": "./src/tools/mcp-stdio.node.ts",
+      "react-native": "./dist/tools/mcp-stdio.unsupported.js",
+      "edge-light": "./dist/tools/mcp-stdio.unsupported.js",
+      "workerd": "./dist/tools/mcp-stdio.unsupported.js",
+      "browser": "./dist/tools/mcp-stdio.unsupported.js",
+      "node": "./dist/tools/mcp-stdio.node.js",
+      "default": "./dist/tools/mcp-stdio.unsupported.js"
+    }
+  },
+  "exports": {
+    ".": {
+      "react-native": {
+        "types": "./dist/index.native.d.ts",
+        "default": "./dist/index.native.js"
+      },
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "main": "./dist/index.js",
+  "react-native": "./dist/index.native.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist",
+    "src",
+    "README.md"
+  ],
+  "sideEffects": false,
+  "dependencies": {
+    "@ai-sdk/mcp": "^2.0.31",
+    "@ai-sdk/react": "^4.0.65",
+    "@assistant-ui/core": "^0.3.14",
+    "@assistant-ui/store": "^0.3.10",
+    "@assistant-ui/tap": "^0.9.13",
+    "ai": "^7.0.62",
+    "assistant-cloud": "*",
+    "assistant-stream": "^0.3.38"
+  },
+  "peerDependencies": {
+    "@types/react": "*",
+    "react": "^18 || ^19"
+  },
+  "peerDependenciesMeta": {
+    "@types/react": {
+      "optional": true
+    },
+    "react": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@testing-library/react": "^16.3.2",
+    "@types/json-schema": "^7.0.15",
+    "@types/react": "^19.2.18",
+    "@types/react-dom": "^19.2.4",
+    "jsdom": "^30.0.1",
+    "react": "^19.2.8",
+    "vitest": "^4.1.10",
+    "@assistant-ui/x-buildutils": "0.0.23"
+  },
+  "publishConfig": {
+    "access": "public",
+    "provenance": true
+  },
+  "homepage": "https://www.assistant-ui.com/",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/assistant-ui/assistant-ui.git",
+    "directory": "packages/ai-sdk"
+  },
+  "bugs": {
+    "url": "https://github.com/assistant-ui/assistant-ui/issues"
+  },
+  "scripts": {
+    "build": "aui-build",
+    "test": "vitest run",
+    "test:react-compiler": "vitest run --config vitest.react-compiler.config.ts",
+    "test:standalone": "vitest run --project standalone",
+    "test:watch": "vitest"
+  }
+}

package/src/__tests__/standalone.e2e.ts

@@ -0,0 +1,59 @@
+import { describe, expect, it, vi } from "vitest";
+import { flushTapSync } from "@assistant-ui/tap";
+import { AuiConfig, createAssistantClient } from "@assistant-ui/store/client";
+import * as aiSdk from "../index";
+import { createControlledTransport } from "../runtime/__tests__/controlled-transport";
+
+describe("react-less standalone graph", () => {
+  it("resolves react to the standalone shim", async () => {
+    const react = await import("react");
+    expect(() => react.useState(0)).toThrow(/standalone-shim/);
+  });
+
+  it("loads the whole barrel under the standalone shim", () => {
+    expect(typeof aiSdk.AISDKChat).toBe("function");
+    expect(typeof aiSdk.useChatRuntime).toBe("function");
+    expect(typeof aiSdk.useAISDKRuntime).toBe("function");
+    expect(typeof aiSdk.frontendTools).toBe("function");
+    expect(typeof aiSdk.AssistantChatTransport).toBe("function");
+  });
+
+  it("streams a chat round trip with the shim as the only react", async () => {
+    const { transport, emit, close } = createControlledTransport();
+    const handle = createAssistantClient(
+      AuiConfig({ threads: aiSdk.AISDKChat({ transport }) }),
+    );
+    try {
+      handle.subscribe(() => {});
+      const aui = handle.getClient();
+
+      flushTapSync(() => aui.composer.setText("hi"));
+      flushTapSync(() => aui.composer.send());
+
+      await vi.waitFor(() => {
+        expect(aui.thread.getState().isRunning).toBe(true);
+      });
+
+      emit(
+        { type: "start" },
+        { type: "text-start", id: "t1" },
+        { type: "text-delta", id: "t1", delta: "hello " },
+        { type: "text-delta", id: "t1", delta: "world" },
+        { type: "text-end", id: "t1" },
+        { type: "finish" },
+      );
+      close();
+
+      await vi.waitFor(() => {
+        const state = aui.thread.getState();
+        expect(state.isRunning).toBe(false);
+        expect(state.messages).toHaveLength(2);
+        expect(state.messages.at(-1)?.content).toContainEqual(
+          expect.objectContaining({ type: "text", text: "hello world" }),
+        );
+      });
+    } finally {
+      handle.destroy();
+    }
+  });
+});

package/src/adapters/aiSDKFormatAdapter.ts

@@ -0,0 +1,41 @@
+import type { UIMessage } from "ai";
+import type {
+  MessageFormatAdapter,
+  MessageFormatItem,
+  MessageStorageEntry,
+} from "@assistant-ui/core";
+
+// Storage format for AI SDK messages - just the UIMessage
+export type AISDKStorageFormat = Omit<UIMessage, "id">;
+
+export const aiSDKV6FormatAdapter: MessageFormatAdapter<
+  UIMessage,
+  AISDKStorageFormat
+> = {
+  format: "ai-sdk/v6",
+
+  encode({
+    message: { id, parts, ...message },
+  }: MessageFormatItem<UIMessage>): AISDKStorageFormat {
+    return {
+      ...message,
+      parts,
+    };
+  },
+
+  decode(
+    stored: MessageStorageEntry<AISDKStorageFormat>,
+  ): MessageFormatItem<UIMessage> {
+    return {
+      parentId: stored.parent_id,
+      message: {
+        id: stored.id,
+        ...stored.content,
+      },
+    };
+  },
+
+  getId(message: UIMessage): string {
+    return message.id;
+  },
+};

package/src/adapters/vercelAttachmentAdapter.test.ts

@@ -0,0 +1,29 @@
+import { afterEach, describe, expect, it } from "vitest";
+import { vercelAttachmentAdapter } from "./vercelAttachmentAdapter";
+
+const originalFileReader = globalThis.FileReader;
+
+afterEach(() => {
+  globalThis.FileReader = originalFileReader;
+});
+
+describe("vercelAttachmentAdapter", () => {
+  it("sends a file as a data URL via the shared core encoder", async () => {
+    globalThis.FileReader = undefined as unknown as typeof FileReader;
+
+    const file = new File(["hello"], "a.txt", { type: "text/plain" });
+    const result = await vercelAttachmentAdapter.send({
+      id: "1",
+      type: "file",
+      name: file.name,
+      file,
+      contentType: file.type,
+      content: [],
+      status: { type: "requires-action", reason: "composer-send" },
+    });
+
+    const part = result.content[0];
+    if (part?.type !== "file") throw new Error("expected file content part");
+    expect(part.data).toBe("data:text/plain;base64,aGVsbG8=");
+  });
+});

package/src/adapters/vercelAttachmentAdapter.ts

@@ -0,0 +1,36 @@
+import type { AttachmentAdapter } from "@assistant-ui/core";
+import { getFileDataURL } from "@assistant-ui/core/internal";
+import { generateId } from "ai";
+
+export const vercelAttachmentAdapter: AttachmentAdapter = {
+  accept: "*",
+  async add({ file }) {
+    return {
+      id: generateId(),
+      type: file.type.startsWith("image/") ? "image" : "file",
+      name: file.name,
+      file,
+      contentType: file.type,
+      content: [],
+      status: { type: "requires-action", reason: "composer-send" },
+    };
+  },
+  async send(attachment) {
+    // noop
+    return {
+      ...attachment,
+      status: { type: "complete" },
+      content: [
+        {
+          type: "file",
+          mimeType: attachment.contentType ?? "",
+          filename: attachment.name,
+          data: await getFileDataURL(attachment.file),
+        },
+      ],
+    };
+  },
+  async remove() {
+    // noop
+  },
+};

package/src/aiSDKExtras.ts

@@ -0,0 +1,11 @@
+import { createRuntimeExtras } from "@assistant-ui/core/react";
+import type { UseChatHelpers } from "@ai-sdk/react";
+import type { UIMessage } from "ai";
+
+export type AISDKRuntimeExtras = {
+  chat: UseChatHelpers<UIMessage>;
+  error: Error | undefined;
+};
+
+export const aiSDKExtras =
+  createRuntimeExtras<AISDKRuntimeExtras>("useAISDKRuntime");