@assistant-ui/react-data-stream 0.10.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 AgentbaseAI Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,3 @@
+# `@assistant-ui/react-data-stream`
+
+Data Stream protocol integration for `@assistant-ui/react`.

package/dist/converters/index.d.ts

@@ -0,0 +1,2 @@
+export { toLanguageModelMessages } from "./toLanguageModelMessages";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/converters/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC"}

package/dist/converters/index.js

@@ -0,0 +1,6 @@
+// src/converters/index.ts
+import { toLanguageModelMessages } from "./toLanguageModelMessages.js";
+export {
+  toLanguageModelMessages
+};
+//# sourceMappingURL=index.js.map

package/dist/converters/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/converters/index.ts"],"sourcesContent":["export { toLanguageModelMessages } from \"./toLanguageModelMessages\";\n"],"mappings":";AAAA,SAAS,+BAA+B;","names":[]}

package/dist/converters/toLanguageModelMessages.d.ts

@@ -0,0 +1,9 @@
+import { LanguageModelV1Message } from "@ai-sdk/provider";
+import { ThreadMessage } from "@assistant-ui/react";
+/**
+ * @deprecated This is an internal API and may change without notice.
+ */
+export declare function toLanguageModelMessages(message: readonly ThreadMessage[], options?: {
+    unstable_includeId?: boolean | undefined;
+}): LanguageModelV1Message[];
+//# sourceMappingURL=toLanguageModelMessages.d.ts.map

package/dist/converters/toLanguageModelMessages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toLanguageModelMessages.d.ts","sourceRoot":"","sources":["../../src/converters/toLanguageModelMessages.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,sBAAsB,EAIvB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAEL,aAAa,EAEd,MAAM,qBAAqB,CAAC;AAiE7B;;GAEG;AACH,wBAAgB,uBAAuB,CACrC,OAAO,EAAE,SAAS,aAAa,EAAE,EACjC,OAAO,GAAE;IAAE,kBAAkB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAA;CAAO,GACzD,sBAAsB,EAAE,CAsG1B"}

package/dist/converters/toLanguageModelMessages.js

@@ -0,0 +1,144 @@
+// src/converters/toLanguageModelMessages.ts
+var assistantMessageSplitter = () => {
+  const stash = [];
+  let assistantMessage = {
+    role: "assistant",
+    content: []
+  };
+  let toolMessage = {
+    role: "tool",
+    content: []
+  };
+  return {
+    addTextMessagePart: (part) => {
+      if (toolMessage.content.length > 0) {
+        stash.push(assistantMessage);
+        stash.push(toolMessage);
+        assistantMessage = {
+          role: "assistant",
+          content: []
+        };
+        toolMessage = {
+          role: "tool",
+          content: []
+        };
+      }
+      assistantMessage.content.push(part);
+    },
+    addToolCallPart: (part) => {
+      assistantMessage.content.push({
+        type: "tool-call",
+        toolCallId: part.toolCallId,
+        toolName: part.toolName,
+        args: part.args
+      });
+      toolMessage.content.push({
+        type: "tool-result",
+        toolCallId: part.toolCallId,
+        toolName: part.toolName,
+        ..."artifact" in part ? { artifact: part.artifact } : {},
+        result: part.result === void 0 ? "Error: tool is has no configured code to run" : part.result,
+        isError: part.isError ?? part.result === void 0
+      });
+    },
+    getMessages: () => {
+      if (toolMessage.content.length > 0) {
+        return [...stash, assistantMessage, toolMessage];
+      }
+      return [...stash, assistantMessage];
+    }
+  };
+};
+function toLanguageModelMessages(message, options = {}) {
+  const includeId = options.unstable_includeId ?? false;
+  return message.flatMap((message2) => {
+    const role = message2.role;
+    switch (role) {
+      case "system": {
+        return [
+          {
+            ...includeId ? { unstable_id: message2.id } : {},
+            role: "system",
+            content: message2.content[0].text
+          }
+        ];
+      }
+      case "user": {
+        const attachments = "attachments" in message2 ? message2.attachments : [];
+        const content = [
+          ...message2.content,
+          ...attachments.map((a) => a.content).flat()
+        ];
+        const msg = {
+          ...includeId ? { unstable_id: message2.id } : {},
+          role: "user",
+          content: content.map(
+            (part) => {
+              const type = part.type;
+              switch (type) {
+                case "text": {
+                  return part;
+                }
+                case "image": {
+                  return {
+                    type: "image",
+                    image: new URL(part.image)
+                  };
+                }
+                case "file": {
+                  return {
+                    type: "file",
+                    data: new URL(part.data),
+                    mimeType: part.mimeType
+                  };
+                }
+                default: {
+                  const unhandledType = type;
+                  throw new Error(
+                    `Unspported message part type: ${unhandledType}`
+                  );
+                }
+              }
+            }
+          )
+        };
+        return [msg];
+      }
+      case "assistant": {
+        const splitter = assistantMessageSplitter();
+        for (const part of message2.content) {
+          const type = part.type;
+          switch (type) {
+            case "reasoning":
+            case "source":
+            case "file":
+            case "image": {
+              break;
+            }
+            case "text": {
+              splitter.addTextMessagePart(part);
+              break;
+            }
+            case "tool-call": {
+              splitter.addToolCallPart(part);
+              break;
+            }
+            default: {
+              const unhandledType = type;
+              throw new Error(`Unhandled message part type: ${unhandledType}`);
+            }
+          }
+        }
+        return splitter.getMessages();
+      }
+      default: {
+        const unhandledRole = role;
+        throw new Error(`Unknown message role: ${unhandledRole}`);
+      }
+    }
+  });
+}
+export {
+  toLanguageModelMessages
+};
+//# sourceMappingURL=toLanguageModelMessages.js.map

package/dist/converters/toLanguageModelMessages.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/converters/toLanguageModelMessages.ts"],"sourcesContent":["import {\n  LanguageModelV1FilePart,\n  LanguageModelV1ImagePart,\n  LanguageModelV1Message,\n  LanguageModelV1TextPart,\n  LanguageModelV1ToolCallPart,\n  LanguageModelV1ToolResultPart,\n} from \"@ai-sdk/provider\";\nimport {\n  TextMessagePart,\n  ThreadMessage,\n  ToolCallMessagePart,\n} from \"@assistant-ui/react\";\n\nconst assistantMessageSplitter = () => {\n  const stash: LanguageModelV1Message[] = [];\n  let assistantMessage = {\n    role: \"assistant\" as const,\n    content: [] as (LanguageModelV1TextPart | LanguageModelV1ToolCallPart)[],\n  };\n  let toolMessage = {\n    role: \"tool\" as const,\n    content: [] as LanguageModelV1ToolResultPart[],\n  };\n\n  return {\n    addTextMessagePart: (part: TextMessagePart) => {\n      if (toolMessage.content.length > 0) {\n        stash.push(assistantMessage);\n        stash.push(toolMessage);\n\n        assistantMessage = {\n          role: \"assistant\" as const,\n          content: [] as (\n            | LanguageModelV1TextPart\n            | LanguageModelV1ToolCallPart\n          )[],\n        };\n\n        toolMessage = {\n          role: \"tool\" as const,\n          content: [] as LanguageModelV1ToolResultPart[],\n        };\n      }\n\n      assistantMessage.content.push(part);\n    },\n    addToolCallPart: (part: ToolCallMessagePart) => {\n      assistantMessage.content.push({\n        type: \"tool-call\",\n        toolCallId: part.toolCallId,\n        toolName: part.toolName,\n        args: part.args,\n      });\n\n      toolMessage.content.push({\n        type: \"tool-result\",\n        toolCallId: part.toolCallId,\n        toolName: part.toolName,\n        ...(\"artifact\" in part ? { artifact: part.artifact } : {}),\n        result:\n          part.result === undefined\n            ? \"Error: tool is has no configured code to run\"\n            : part.result,\n        isError: part.isError ?? part.result === undefined,\n      });\n    },\n    getMessages: () => {\n      if (toolMessage.content.length > 0) {\n        return [...stash, assistantMessage, toolMessage];\n      }\n\n      return [...stash, assistantMessage];\n    },\n  };\n};\n\n/**\n * @deprecated This is an internal API and may change without notice.\n */\nexport function toLanguageModelMessages(\n  message: readonly ThreadMessage[],\n  options: { unstable_includeId?: boolean | undefined } = {},\n): LanguageModelV1Message[] {\n  const includeId = options.unstable_includeId ?? false;\n  return message.flatMap((message) => {\n    const role = message.role;\n    switch (role) {\n      case \"system\": {\n        return [\n          {\n            ...(includeId\n              ? { unstable_id: (message as ThreadMessage).id }\n              : {}),\n            role: \"system\",\n            content: message.content[0].text,\n          },\n        ];\n      }\n\n      case \"user\": {\n        const attachments = \"attachments\" in message ? message.attachments : [];\n        const content = [\n          ...message.content,\n          ...attachments.map((a) => a.content).flat(),\n        ];\n        const msg: LanguageModelV1Message = {\n          ...(includeId ? { unstable_id: (message as ThreadMessage).id } : {}),\n          role: \"user\",\n          content: content.map(\n            (\n              part,\n            ):\n              | LanguageModelV1TextPart\n              | LanguageModelV1ImagePart\n              | LanguageModelV1FilePart => {\n              const type = part.type;\n              switch (type) {\n                case \"text\": {\n                  return part;\n                }\n\n                case \"image\": {\n                  return {\n                    type: \"image\",\n                    image: new URL(part.image),\n                  };\n                }\n\n                case \"file\": {\n                  return {\n                    type: \"file\",\n                    data: new URL(part.data),\n                    mimeType: part.mimeType,\n                  };\n                }\n\n                default: {\n                  const unhandledType: \"audio\" = type;\n                  throw new Error(\n                    `Unspported message part type: ${unhandledType}`,\n                  );\n                }\n              }\n            },\n          ),\n        };\n        return [msg];\n      }\n\n      case \"assistant\": {\n        const splitter = assistantMessageSplitter();\n        for (const part of message.content) {\n          const type = part.type;\n          switch (type) {\n            case \"reasoning\":\n            case \"source\":\n            case \"file\":\n            case \"image\": {\n              break; // reasoning, source, file, and image parts are omitted\n            }\n\n            case \"text\": {\n              splitter.addTextMessagePart(part);\n              break;\n            }\n            case \"tool-call\": {\n              splitter.addToolCallPart(part);\n              break;\n            }\n            default: {\n              const unhandledType: never = type;\n              throw new Error(`Unhandled message part type: ${unhandledType}`);\n            }\n          }\n        }\n        return splitter.getMessages();\n      }\n\n      default: {\n        const unhandledRole: never = role;\n        throw new Error(`Unknown message role: ${unhandledRole}`);\n      }\n    }\n  });\n}\n"],"mappings":";AAcA,IAAM,2BAA2B,MAAM;AACrC,QAAM,QAAkC,CAAC;AACzC,MAAI,mBAAmB;AAAA,IACrB,MAAM;AAAA,IACN,SAAS,CAAC;AAAA,EACZ;AACA,MAAI,cAAc;AAAA,IAChB,MAAM;AAAA,IACN,SAAS,CAAC;AAAA,EACZ;AAEA,SAAO;AAAA,IACL,oBAAoB,CAAC,SAA0B;AAC7C,UAAI,YAAY,QAAQ,SAAS,GAAG;AAClC,cAAM,KAAK,gBAAgB;AAC3B,cAAM,KAAK,WAAW;AAEtB,2BAAmB;AAAA,UACjB,MAAM;AAAA,UACN,SAAS,CAAC;AAAA,QAIZ;AAEA,sBAAc;AAAA,UACZ,MAAM;AAAA,UACN,SAAS,CAAC;AAAA,QACZ;AAAA,MACF;AAEA,uBAAiB,QAAQ,KAAK,IAAI;AAAA,IACpC;AAAA,IACA,iBAAiB,CAAC,SAA8B;AAC9C,uBAAiB,QAAQ,KAAK;AAAA,QAC5B,MAAM;AAAA,QACN,YAAY,KAAK;AAAA,QACjB,UAAU,KAAK;AAAA,QACf,MAAM,KAAK;AAAA,MACb,CAAC;AAED,kBAAY,QAAQ,KAAK;AAAA,QACvB,MAAM;AAAA,QACN,YAAY,KAAK;AAAA,QACjB,UAAU,KAAK;AAAA,QACf,GAAI,cAAc,OAAO,EAAE,UAAU,KAAK,SAAS,IAAI,CAAC;AAAA,QACxD,QACE,KAAK,WAAW,SACZ,iDACA,KAAK;AAAA,QACX,SAAS,KAAK,WAAW,KAAK,WAAW;AAAA,MAC3C,CAAC;AAAA,IACH;AAAA,IACA,aAAa,MAAM;AACjB,UAAI,YAAY,QAAQ,SAAS,GAAG;AAClC,eAAO,CAAC,GAAG,OAAO,kBAAkB,WAAW;AAAA,MACjD;AAEA,aAAO,CAAC,GAAG,OAAO,gBAAgB;AAAA,IACpC;AAAA,EACF;AACF;AAKO,SAAS,wBACd,SACA,UAAwD,CAAC,GAC/B;AAC1B,QAAM,YAAY,QAAQ,sBAAsB;AAChD,SAAO,QAAQ,QAAQ,CAACA,aAAY;AAClC,UAAM,OAAOA,SAAQ;AACrB,YAAQ,MAAM;AAAA,MACZ,KAAK,UAAU;AACb,eAAO;AAAA,UACL;AAAA,YACE,GAAI,YACA,EAAE,aAAcA,SAA0B,GAAG,IAC7C,CAAC;AAAA,YACL,MAAM;AAAA,YACN,SAASA,SAAQ,QAAQ,CAAC,EAAE;AAAA,UAC9B;AAAA,QACF;AAAA,MACF;AAAA,MAEA,KAAK,QAAQ;AACX,cAAM,cAAc,iBAAiBA,WAAUA,SAAQ,cAAc,CAAC;AACtE,cAAM,UAAU;AAAA,UACd,GAAGA,SAAQ;AAAA,UACX,GAAG,YAAY,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK;AAAA,QAC5C;AACA,cAAM,MAA8B;AAAA,UAClC,GAAI,YAAY,EAAE,aAAcA,SAA0B,GAAG,IAAI,CAAC;AAAA,UAClE,MAAM;AAAA,UACN,SAAS,QAAQ;AAAA,YACf,CACE,SAI6B;AAC7B,oBAAM,OAAO,KAAK;AAClB,sBAAQ,MAAM;AAAA,gBACZ,KAAK,QAAQ;AACX,yBAAO;AAAA,gBACT;AAAA,gBAEA,KAAK,SAAS;AACZ,yBAAO;AAAA,oBACL,MAAM;AAAA,oBACN,OAAO,IAAI,IAAI,KAAK,KAAK;AAAA,kBAC3B;AAAA,gBACF;AAAA,gBAEA,KAAK,QAAQ;AACX,yBAAO;AAAA,oBACL,MAAM;AAAA,oBACN,MAAM,IAAI,IAAI,KAAK,IAAI;AAAA,oBACvB,UAAU,KAAK;AAAA,kBACjB;AAAA,gBACF;AAAA,gBAEA,SAAS;AACP,wBAAM,gBAAyB;AAC/B,wBAAM,IAAI;AAAA,oBACR,iCAAiC,aAAa;AAAA,kBAChD;AAAA,gBACF;AAAA,cACF;AAAA,YACF;AAAA,UACF;AAAA,QACF;AACA,eAAO,CAAC,GAAG;AAAA,MACb;AAAA,MAEA,KAAK,aAAa;AAChB,cAAM,WAAW,yBAAyB;AAC1C,mBAAW,QAAQA,SAAQ,SAAS;AAClC,gBAAM,OAAO,KAAK;AAClB,kBAAQ,MAAM;AAAA,YACZ,KAAK;AAAA,YACL,KAAK;AAAA,YACL,KAAK;AAAA,YACL,KAAK,SAAS;AACZ;AAAA,YACF;AAAA,YAEA,KAAK,QAAQ;AACX,uBAAS,mBAAmB,IAAI;AAChC;AAAA,YACF;AAAA,YACA,KAAK,aAAa;AAChB,uBAAS,gBAAgB,IAAI;AAC7B;AAAA,YACF;AAAA,YACA,SAAS;AACP,oBAAM,gBAAuB;AAC7B,oBAAM,IAAI,MAAM,gCAAgC,aAAa,EAAE;AAAA,YACjE;AAAA,UACF;AAAA,QACF;AACA,eAAO,SAAS,YAAY;AAAA,MAC9B;AAAA,MAEA,SAAS;AACP,cAAM,gBAAuB;AAC7B,cAAM,IAAI,MAAM,yBAAyB,aAAa,EAAE;AAAA,MAC1D;AAAA,IACF;AAAA,EACF,CAAC;AACH;","names":["message"]}

package/dist/frontendTools.d.ts

@@ -0,0 +1,11 @@
+import type { JSONSchema7 } from "json-schema";
+export declare const frontendTools: (tools: Record<string, {
+    description?: string;
+    parameters: JSONSchema7;
+}>) => {
+    [k: string]: {
+        parameters: import("@ai-sdk/ui-utils").Schema<unknown>;
+        description?: string;
+    };
+};
+//# sourceMappingURL=frontendTools.d.ts.map

package/dist/frontendTools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"frontendTools.d.ts","sourceRoot":"","sources":["../src/frontendTools.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE/C,eAAO,MAAM,aAAa,GACxB,OAAO,MAAM,CAAC,MAAM,EAAE;IAAE,WAAW,CAAC,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,WAAW,CAAA;CAAE,CAAC;;;;;CAUvE,CAAC"}

package/dist/frontendTools.js

@@ -0,0 +1,15 @@
+// src/frontendTools.ts
+import { jsonSchema } from "@ai-sdk/ui-utils";
+var frontendTools = (tools) => Object.fromEntries(
+  Object.entries(tools).map(([name, tool]) => [
+    name,
+    {
+      ...tool.description ? { description: tool.description } : void 0,
+      parameters: jsonSchema(tool.parameters)
+    }
+  ])
+);
+export {
+  frontendTools
+};
+//# sourceMappingURL=frontendTools.js.map

package/dist/frontendTools.js.map

@@ -0,0 +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":[]}

package/dist/index.d.ts

@@ -0,0 +1,5 @@
+export * from "./useDataStreamRuntime";
+export * from "./useCloudRuntime";
+export * from "./converters";
+export { frontendTools } from "./frontendTools";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,wBAAwB,CAAC;AACvC,cAAc,mBAAmB,CAAC;AAClC,cAAc,cAAc,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/index.js

@@ -0,0 +1,9 @@
+// src/index.ts
+export * from "./useDataStreamRuntime.js";
+export * from "./useCloudRuntime.js";
+export * from "./converters/index.js";
+import { frontendTools } from "./frontendTools.js";
+export {
+  frontendTools
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["export * from \"./useDataStreamRuntime\";\nexport * from \"./useCloudRuntime\";\nexport * from \"./converters\";\nexport { frontendTools } from \"./frontendTools\";\n"],"mappings":";AAAA,cAAc;AACd,cAAc;AACd,cAAc;AACd,SAAS,qBAAqB;","names":[]}

package/dist/useCloudRuntime.d.ts

@@ -0,0 +1,12 @@
+import { AssistantCloud } from "@assistant-ui/react";
+import { UseDataStreamRuntimeOptions } from "./useDataStreamRuntime";
+type UseCloudRuntimeOptions = Omit<UseDataStreamRuntimeOptions, "api"> & {
+    cloud: AssistantCloud;
+    assistantId: string;
+};
+/**
+ * @deprecated This is under active development and not yet ready for prod use.
+ */
+export declare const useCloudRuntime: (options: UseCloudRuntimeOptions) => import("@assistant-ui/react").AssistantRuntime;
+export {};
+//# sourceMappingURL=useCloudRuntime.d.ts.map

package/dist/useCloudRuntime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useCloudRuntime.d.ts","sourceRoot":"","sources":["../src/useCloudRuntime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAEL,2BAA2B,EAC5B,MAAM,wBAAwB,CAAC;AAEhC,KAAK,sBAAsB,GAAG,IAAI,CAAC,2BAA2B,EAAE,KAAK,CAAC,GAAG;IACvE,KAAK,EAAE,cAAc,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,eAAe,GAAI,SAAS,sBAAsB,mDAS9D,CAAC"}

package/dist/useCloudRuntime.js

@@ -0,0 +1,17 @@
+// src/useCloudRuntime.ts
+import {
+  useDataStreamRuntime
+} from "./useDataStreamRuntime.js";
+var useCloudRuntime = (options) => {
+  const opts = options.cloud.runs.__internal_getAssistantOptions(
+    options.assistantId
+  );
+  return useDataStreamRuntime({
+    ...options,
+    ...opts
+  });
+};
+export {
+  useCloudRuntime
+};
+//# sourceMappingURL=useCloudRuntime.js.map

package/dist/useCloudRuntime.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/useCloudRuntime.ts"],"sourcesContent":["import { AssistantCloud } from \"@assistant-ui/react\";\nimport {\n  useDataStreamRuntime,\n  UseDataStreamRuntimeOptions,\n} from \"./useDataStreamRuntime\";\n\ntype UseCloudRuntimeOptions = Omit<UseDataStreamRuntimeOptions, \"api\"> & {\n  cloud: AssistantCloud;\n  assistantId: string;\n};\n\n/**\n * @deprecated This is under active development and not yet ready for prod use.\n */\nexport const useCloudRuntime = (options: UseCloudRuntimeOptions) => {\n  const opts = options.cloud.runs.__internal_getAssistantOptions(\n    options.assistantId,\n  );\n\n  return useDataStreamRuntime({\n    ...options,\n    ...opts,\n  });\n};\n"],"mappings":";AACA;AAAA,EACE;AAAA,OAEK;AAUA,IAAM,kBAAkB,CAAC,YAAoC;AAClE,QAAM,OAAO,QAAQ,MAAM,KAAK;AAAA,IAC9B,QAAQ;AAAA,EACV;AAEA,SAAO,qBAAqB;AAAA,IAC1B,GAAG;AAAA,IACH,GAAG;AAAA,EACL,CAAC;AACH;","names":[]}

package/dist/useDataStreamRuntime.d.ts

@@ -0,0 +1,16 @@
+import { AssistantRuntime, LocalRuntimeOptions, ThreadMessage } from "@assistant-ui/react";
+type HeadersValue = Record<string, string> | Headers;
+export type UseDataStreamRuntimeOptions = {
+    api: string;
+    onResponse?: (response: Response) => void | Promise<void>;
+    onFinish?: (message: ThreadMessage) => void;
+    onError?: (error: Error) => void;
+    onCancel?: () => void;
+    credentials?: RequestCredentials;
+    headers?: HeadersValue | (() => Promise<HeadersValue>);
+    body?: object;
+    sendExtraMessageFields?: boolean;
+} & LocalRuntimeOptions;
+export declare const useDataStreamRuntime: (options: UseDataStreamRuntimeOptions) => AssistantRuntime;
+export {};
+//# sourceMappingURL=useDataStreamRuntime.d.ts.map

package/dist/useDataStreamRuntime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useDataStreamRuntime.d.ts","sourceRoot":"","sources":["../src/useDataStreamRuntime.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,gBAAgB,EAIhB,mBAAmB,EACnB,aAAa,EAGd,MAAM,qBAAqB,CAAC;AAY7B,KAAK,YAAY,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC;AAErD,MAAM,MAAM,2BAA2B,GAAG;IACxC,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1D,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,IAAI,CAAC;IAC5C,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACjC,QAAQ,CAAC,EAAE,MAAM,IAAI,CAAC;IACtB,WAAW,CAAC,EAAE,kBAAkB,CAAC;IACjC,OAAO,CAAC,EAAE,YAAY,GAAG,CAAC,MAAM,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC;IACvD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,sBAAsB,CAAC,EAAE,OAAO,CAAC;CAClC,GAAG,mBAAmB,CAAC;AAgHxB,eAAO,MAAM,oBAAoB,GAC/B,SAAS,2BAA2B,KACnC,gBAQF,CAAC"}

package/dist/useDataStreamRuntime.js

@@ -0,0 +1,105 @@
+"use client";
+
+// src/useDataStreamRuntime.ts
+import { toLanguageModelMessages } from "./converters/index.js";
+import {
+  INTERNAL,
+  useLocalRuntime
+} from "@assistant-ui/react";
+import { z } from "zod";
+import {
+  AssistantMessageAccumulator,
+  DataStreamDecoder,
+  unstable_toolResultStream
+} from "assistant-stream";
+import { asAsyncIterableStream } from "assistant-stream/utils";
+var { splitLocalRuntimeOptions } = INTERNAL;
+var toAISDKTools = (tools) => {
+  return Object.fromEntries(
+    Object.entries(tools).map(([name, tool]) => [
+      name,
+      {
+        ...tool.description ? { description: tool.description } : void 0,
+        parameters: tool.parameters instanceof z.ZodType ? z.toJSONSchema(tool.parameters) : tool.parameters
+      }
+    ])
+  );
+};
+var getEnabledTools = (tools) => {
+  return Object.fromEntries(
+    Object.entries(tools).filter(
+      ([, tool]) => !tool.disabled && tool.type !== "backend"
+    )
+  );
+};
+var DataStreamRuntimeAdapter = class {
+  constructor(options) {
+    this.options = options;
+  }
+  async *run({
+    messages,
+    runConfig,
+    abortSignal,
+    context,
+    unstable_assistantMessageId,
+    unstable_getMessage
+  }) {
+    const headersValue = typeof this.options.headers === "function" ? await this.options.headers() : this.options.headers;
+    abortSignal.addEventListener(
+      "abort",
+      () => {
+        if (!abortSignal.reason?.detach) this.options.onCancel?.();
+      },
+      { once: true }
+    );
+    const headers = new Headers(headersValue);
+    headers.set("Content-Type", "application/json");
+    const result = await fetch(this.options.api, {
+      method: "POST",
+      headers,
+      credentials: this.options.credentials ?? "same-origin",
+      body: JSON.stringify({
+        system: context.system,
+        messages: toLanguageModelMessages(messages, {
+          unstable_includeId: this.options.sendExtraMessageFields
+        }),
+        tools: toAISDKTools(
+          getEnabledTools(context.tools ?? {})
+        ),
+        ...unstable_assistantMessageId ? { unstable_assistantMessageId } : {},
+        runConfig,
+        state: unstable_getMessage().metadata.unstable_state || void 0,
+        ...context.callSettings,
+        ...context.config,
+        ...this.options.body
+      }),
+      signal: abortSignal
+    });
+    await this.options.onResponse?.(result);
+    try {
+      if (!result.ok) {
+        throw new Error(`Status ${result.status}: ${await result.text()}`);
+      }
+      if (!result.body) {
+        throw new Error("Response body is null");
+      }
+      const stream = result.body.pipeThrough(new DataStreamDecoder()).pipeThrough(unstable_toolResultStream(context.tools, abortSignal)).pipeThrough(new AssistantMessageAccumulator());
+      yield* asAsyncIterableStream(stream);
+      this.options.onFinish?.(unstable_getMessage());
+    } catch (error) {
+      this.options.onError?.(error);
+      throw error;
+    }
+  }
+};
+var useDataStreamRuntime = (options) => {
+  const { localRuntimeOptions, otherOptions } = splitLocalRuntimeOptions(options);
+  return useLocalRuntime(
+    new DataStreamRuntimeAdapter(otherOptions),
+    localRuntimeOptions
+  );
+};
+export {
+  useDataStreamRuntime
+};
+//# sourceMappingURL=useDataStreamRuntime.js.map

package/dist/useDataStreamRuntime.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/useDataStreamRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { toLanguageModelMessages } from \"./converters\";\nimport {\n  AssistantRuntime,\n  ChatModelAdapter,\n  ChatModelRunOptions,\n  INTERNAL,\n  LocalRuntimeOptions,\n  ThreadMessage,\n  Tool,\n  useLocalRuntime,\n} from \"@assistant-ui/react\";\nimport { z } from \"zod\";\nimport { JSONSchema7 } from \"json-schema\";\nimport {\n  AssistantMessageAccumulator,\n  DataStreamDecoder,\n  unstable_toolResultStream,\n} from \"assistant-stream\";\nimport { asAsyncIterableStream } from \"assistant-stream/utils\";\n\nconst { splitLocalRuntimeOptions } = INTERNAL;\n\ntype HeadersValue = Record<string, string> | Headers;\n\nexport type UseDataStreamRuntimeOptions = {\n  api: string;\n  onResponse?: (response: Response) => void | Promise<void>;\n  onFinish?: (message: ThreadMessage) => void;\n  onError?: (error: Error) => void;\n  onCancel?: () => void;\n  credentials?: RequestCredentials;\n  headers?: HeadersValue | (() => Promise<HeadersValue>);\n  body?: object;\n  sendExtraMessageFields?: boolean;\n} & LocalRuntimeOptions;\n\ntype DataStreamRuntimeRequestOptions = {\n  messages: any[];\n  tools: any;\n  system?: string | undefined;\n  runConfig?: any;\n  unstable_assistantMessageId?: string;\n  state?: any;\n};\n\nconst toAISDKTools = (tools: Record<string, Tool>) => {\n  return Object.fromEntries(\n    Object.entries(tools).map(([name, tool]) => [\n      name,\n      {\n        ...(tool.description ? { description: tool.description } : undefined),\n        parameters: (tool.parameters instanceof z.ZodType\n          ? z.toJSONSchema(tool.parameters)\n          : tool.parameters) as JSONSchema7,\n      },\n    ]),\n  );\n};\n\nconst getEnabledTools = (tools: Record<string, Tool>) => {\n  return Object.fromEntries(\n    Object.entries(tools).filter(\n      ([, tool]) => !tool.disabled && tool.type !== \"backend\",\n    ),\n  );\n};\n\nclass DataStreamRuntimeAdapter implements ChatModelAdapter {\n  constructor(\n    private options: Omit<\n      UseDataStreamRuntimeOptions,\n      keyof LocalRuntimeOptions\n    >,\n  ) {}\n\n  async *run({\n    messages,\n    runConfig,\n    abortSignal,\n    context,\n    unstable_assistantMessageId,\n    unstable_getMessage,\n  }: ChatModelRunOptions) {\n    const headersValue =\n      typeof this.options.headers === \"function\"\n        ? await this.options.headers()\n        : this.options.headers;\n\n    abortSignal.addEventListener(\n      \"abort\",\n      () => {\n        if (!abortSignal.reason?.detach) this.options.onCancel?.();\n      },\n      { once: true },\n    );\n\n    const headers = new Headers(headersValue);\n    headers.set(\"Content-Type\", \"application/json\");\n\n    const result = await fetch(this.options.api, {\n      method: \"POST\",\n      headers,\n      credentials: this.options.credentials ?? \"same-origin\",\n      body: JSON.stringify({\n        system: context.system,\n        messages: toLanguageModelMessages(messages, {\n          unstable_includeId: this.options.sendExtraMessageFields,\n        }) as DataStreamRuntimeRequestOptions[\"messages\"],\n        tools: toAISDKTools(\n          getEnabledTools(context.tools ?? {}),\n        ) as unknown as DataStreamRuntimeRequestOptions[\"tools\"],\n        ...(unstable_assistantMessageId ? { unstable_assistantMessageId } : {}),\n        runConfig,\n        state: unstable_getMessage().metadata.unstable_state || undefined,\n        ...context.callSettings,\n        ...context.config,\n        ...this.options.body,\n      } satisfies DataStreamRuntimeRequestOptions),\n      signal: abortSignal,\n    });\n\n    await this.options.onResponse?.(result);\n\n    try {\n      if (!result.ok) {\n        throw new Error(`Status ${result.status}: ${await result.text()}`);\n      }\n      if (!result.body) {\n        throw new Error(\"Response body is null\");\n      }\n\n      const stream = result.body\n        .pipeThrough(new DataStreamDecoder())\n        .pipeThrough(unstable_toolResultStream(context.tools, abortSignal))\n        .pipeThrough(new AssistantMessageAccumulator());\n\n      yield* asAsyncIterableStream(stream);\n\n      this.options.onFinish?.(unstable_getMessage());\n    } catch (error: unknown) {\n      this.options.onError?.(error as Error);\n      throw error;\n    }\n  }\n}\n\nexport const useDataStreamRuntime = (\n  options: UseDataStreamRuntimeOptions,\n): AssistantRuntime => {\n  const { localRuntimeOptions, otherOptions } =\n    splitLocalRuntimeOptions(options);\n\n  return useLocalRuntime(\n    new DataStreamRuntimeAdapter(otherOptions),\n    localRuntimeOptions,\n  );\n};\n"],"mappings":";;;AAEA,SAAS,+BAA+B;AACxC;AAAA,EAIE;AAAA,EAIA;AAAA,OACK;AACP,SAAS,SAAS;AAElB;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,6BAA6B;AAEtC,IAAM,EAAE,yBAAyB,IAAI;AAyBrC,IAAM,eAAe,CAAC,UAAgC;AACpD,SAAO,OAAO;AAAA,IACZ,OAAO,QAAQ,KAAK,EAAE,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM;AAAA,MAC1C;AAAA,MACA;AAAA,QACE,GAAI,KAAK,cAAc,EAAE,aAAa,KAAK,YAAY,IAAI;AAAA,QAC3D,YAAa,KAAK,sBAAsB,EAAE,UACtC,EAAE,aAAa,KAAK,UAAU,IAC9B,KAAK;AAAA,MACX;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEA,IAAM,kBAAkB,CAAC,UAAgC;AACvD,SAAO,OAAO;AAAA,IACZ,OAAO,QAAQ,KAAK,EAAE;AAAA,MACpB,CAAC,CAAC,EAAE,IAAI,MAAM,CAAC,KAAK,YAAY,KAAK,SAAS;AAAA,IAChD;AAAA,EACF;AACF;AAEA,IAAM,2BAAN,MAA2D;AAAA,EACzD,YACU,SAIR;AAJQ;AAAA,EAIP;AAAA,EAEH,OAAO,IAAI;AAAA,IACT;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAAwB;AACtB,UAAM,eACJ,OAAO,KAAK,QAAQ,YAAY,aAC5B,MAAM,KAAK,QAAQ,QAAQ,IAC3B,KAAK,QAAQ;AAEnB,gBAAY;AAAA,MACV;AAAA,MACA,MAAM;AACJ,YAAI,CAAC,YAAY,QAAQ,OAAQ,MAAK,QAAQ,WAAW;AAAA,MAC3D;AAAA,MACA,EAAE,MAAM,KAAK;AAAA,IACf;AAEA,UAAM,UAAU,IAAI,QAAQ,YAAY;AACxC,YAAQ,IAAI,gBAAgB,kBAAkB;AAE9C,UAAM,SAAS,MAAM,MAAM,KAAK,QAAQ,KAAK;AAAA,MAC3C,QAAQ;AAAA,MACR;AAAA,MACA,aAAa,KAAK,QAAQ,eAAe;AAAA,MACzC,MAAM,KAAK,UAAU;AAAA,QACnB,QAAQ,QAAQ;AAAA,QAChB,UAAU,wBAAwB,UAAU;AAAA,UAC1C,oBAAoB,KAAK,QAAQ;AAAA,QACnC,CAAC;AAAA,QACD,OAAO;AAAA,UACL,gBAAgB,QAAQ,SAAS,CAAC,CAAC;AAAA,QACrC;AAAA,QACA,GAAI,8BAA8B,EAAE,4BAA4B,IAAI,CAAC;AAAA,QACrE;AAAA,QACA,OAAO,oBAAoB,EAAE,SAAS,kBAAkB;AAAA,QACxD,GAAG,QAAQ;AAAA,QACX,GAAG,QAAQ;AAAA,QACX,GAAG,KAAK,QAAQ;AAAA,MAClB,CAA2C;AAAA,MAC3C,QAAQ;AAAA,IACV,CAAC;AAED,UAAM,KAAK,QAAQ,aAAa,MAAM;AAEtC,QAAI;AACF,UAAI,CAAC,OAAO,IAAI;AACd,cAAM,IAAI,MAAM,UAAU,OAAO,MAAM,KAAK,MAAM,OAAO,KAAK,CAAC,EAAE;AAAA,MACnE;AACA,UAAI,CAAC,OAAO,MAAM;AAChB,cAAM,IAAI,MAAM,uBAAuB;AAAA,MACzC;AAEA,YAAM,SAAS,OAAO,KACnB,YAAY,IAAI,kBAAkB,CAAC,EACnC,YAAY,0BAA0B,QAAQ,OAAO,WAAW,CAAC,EACjE,YAAY,IAAI,4BAA4B,CAAC;AAEhD,aAAO,sBAAsB,MAAM;AAEnC,WAAK,QAAQ,WAAW,oBAAoB,CAAC;AAAA,IAC/C,SAAS,OAAgB;AACvB,WAAK,QAAQ,UAAU,KAAc;AACrC,YAAM;AAAA,IACR;AAAA,EACF;AACF;AAEO,IAAM,uBAAuB,CAClC,YACqB;AACrB,QAAM,EAAE,qBAAqB,aAAa,IACxC,yBAAyB,OAAO;AAElC,SAAO;AAAA,IACL,IAAI,yBAAyB,YAAY;AAAA,IACzC;AAAA,EACF;AACF;","names":[]}

package/package.json

@@ -0,0 +1,66 @@
+{
+  "name": "@assistant-ui/react-data-stream",
+  "version": "0.10.0",
+  "license": "MIT",
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist",
+    "src",
+    "README.md"
+  ],
+  "sideEffects": false,
+  "dependencies": {
+    "@ai-sdk/provider": "^1.1.3",
+    "@ai-sdk/react": "^1",
+    "@ai-sdk/ui-utils": "^1",
+    "@radix-ui/react-use-callback-ref": "^1.1.1",
+    "@types/json-schema": "^7.0.15",
+    "assistant-stream": "^0.2.22",
+    "json-schema": "^0.4.0",
+    "zod": "^4.0.14",
+    "zustand": "^5.0.7"
+  },
+  "peerDependencies": {
+    "@assistant-ui/react": "^0.10.40",
+    "@types/react": "*",
+    "react": "^18 || ^19 || ^19.0.0-rc"
+  },
+  "peerDependenciesMeta": {
+    "@types/react": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@types/node": "^24.1.0",
+    "@types/react": "^19.1.9",
+    "eslint": "^9",
+    "eslint-config-next": "15.4.5",
+    "react": "19.1.1",
+    "tsx": "^4.20.3",
+    "@assistant-ui/react": "0.10.41",
+    "@assistant-ui/x-buildutils": "0.0.1"
+  },
+  "publishConfig": {
+    "access": "public",
+    "provenance": true
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/assistant-ui/assistant-ui/tree/main/packages/react-ai-sdk"
+  },
+  "bugs": {
+    "url": "https://github.com/assistant-ui/assistant-ui/issues"
+  },
+  "scripts": {
+    "build": "tsx scripts/build.mts",
+    "lint": "eslint ."
+  }
+}

package/src/converters/index.ts

@@ -0,0 +1 @@
+export { toLanguageModelMessages } from "./toLanguageModelMessages";

package/src/converters/toLanguageModelMessages.ts

@@ -0,0 +1,186 @@
+import {
+  LanguageModelV1FilePart,
+  LanguageModelV1ImagePart,
+  LanguageModelV1Message,
+  LanguageModelV1TextPart,
+  LanguageModelV1ToolCallPart,
+  LanguageModelV1ToolResultPart,
+} from "@ai-sdk/provider";
+import {
+  TextMessagePart,
+  ThreadMessage,
+  ToolCallMessagePart,
+} from "@assistant-ui/react";
+
+const assistantMessageSplitter = () => {
+  const stash: LanguageModelV1Message[] = [];
+  let assistantMessage = {
+    role: "assistant" as const,
+    content: [] as (LanguageModelV1TextPart | LanguageModelV1ToolCallPart)[],
+  };
+  let toolMessage = {
+    role: "tool" as const,
+    content: [] as LanguageModelV1ToolResultPart[],
+  };
+
+  return {
+    addTextMessagePart: (part: TextMessagePart) => {
+      if (toolMessage.content.length > 0) {
+        stash.push(assistantMessage);
+        stash.push(toolMessage);
+
+        assistantMessage = {
+          role: "assistant" as const,
+          content: [] as (
+            | LanguageModelV1TextPart
+            | LanguageModelV1ToolCallPart
+          )[],
+        };
+
+        toolMessage = {
+          role: "tool" as const,
+          content: [] as LanguageModelV1ToolResultPart[],
+        };
+      }
+
+      assistantMessage.content.push(part);
+    },
+    addToolCallPart: (part: ToolCallMessagePart) => {
+      assistantMessage.content.push({
+        type: "tool-call",
+        toolCallId: part.toolCallId,
+        toolName: part.toolName,
+        args: part.args,
+      });
+
+      toolMessage.content.push({
+        type: "tool-result",
+        toolCallId: part.toolCallId,
+        toolName: part.toolName,
+        ...("artifact" in part ? { artifact: part.artifact } : {}),
+        result:
+          part.result === undefined
+            ? "Error: tool is has no configured code to run"
+            : part.result,
+        isError: part.isError ?? part.result === undefined,
+      });
+    },
+    getMessages: () => {
+      if (toolMessage.content.length > 0) {
+        return [...stash, assistantMessage, toolMessage];
+      }
+
+      return [...stash, assistantMessage];
+    },
+  };
+};
+
+/**
+ * @deprecated This is an internal API and may change without notice.
+ */
+export function toLanguageModelMessages(
+  message: readonly ThreadMessage[],
+  options: { unstable_includeId?: boolean | undefined } = {},
+): LanguageModelV1Message[] {
+  const includeId = options.unstable_includeId ?? false;
+  return message.flatMap((message) => {
+    const role = message.role;
+    switch (role) {
+      case "system": {
+        return [
+          {
+            ...(includeId
+              ? { unstable_id: (message as ThreadMessage).id }
+              : {}),
+            role: "system",
+            content: message.content[0].text,
+          },
+        ];
+      }
+
+      case "user": {
+        const attachments = "attachments" in message ? message.attachments : [];
+        const content = [
+          ...message.content,
+          ...attachments.map((a) => a.content).flat(),
+        ];
+        const msg: LanguageModelV1Message = {
+          ...(includeId ? { unstable_id: (message as ThreadMessage).id } : {}),
+          role: "user",
+          content: content.map(
+            (
+              part,
+            ):
+              | LanguageModelV1TextPart
+              | LanguageModelV1ImagePart
+              | LanguageModelV1FilePart => {
+              const type = part.type;
+              switch (type) {
+                case "text": {
+                  return part;
+                }
+
+                case "image": {
+                  return {
+                    type: "image",
+                    image: new URL(part.image),
+                  };
+                }
+
+                case "file": {
+                  return {
+                    type: "file",
+                    data: new URL(part.data),
+                    mimeType: part.mimeType,
+                  };
+                }
+
+                default: {
+                  const unhandledType: "audio" = type;
+                  throw new Error(
+                    `Unspported message part type: ${unhandledType}`,
+                  );
+                }
+              }
+            },
+          ),
+        };
+        return [msg];
+      }
+
+      case "assistant": {
+        const splitter = assistantMessageSplitter();
+        for (const part of message.content) {
+          const type = part.type;
+          switch (type) {
+            case "reasoning":
+            case "source":
+            case "file":
+            case "image": {
+              break; // reasoning, source, file, and image parts are omitted
+            }
+
+            case "text": {
+              splitter.addTextMessagePart(part);
+              break;
+            }
+            case "tool-call": {
+              splitter.addToolCallPart(part);
+              break;
+            }
+            default: {
+              const unhandledType: never = type;
+              throw new Error(`Unhandled message part type: ${unhandledType}`);
+            }
+          }
+        }
+        return splitter.getMessages();
+      }
+
+      default: {
+        const unhandledRole: never = role;
+        throw new Error(`Unknown message role: ${unhandledRole}`);
+      }
+    }
+  });
+}

package/src/frontendTools.ts

@@ -0,0 +1,15 @@
+import { jsonSchema } from "@ai-sdk/ui-utils";
+import type { JSONSchema7 } from "json-schema";
+
+export const frontendTools = (
+  tools: Record<string, { description?: string; parameters: JSONSchema7 }>,
+) =>
+  Object.fromEntries(
+    Object.entries(tools).map(([name, tool]) => [
+      name,
+      {
+        ...(tool.description ? { description: tool.description } : undefined),
+        parameters: jsonSchema(tool.parameters),
+      },
+    ]),
+  );

package/src/index.ts

@@ -0,0 +1,4 @@
+export * from "./useDataStreamRuntime";
+export * from "./useCloudRuntime";
+export * from "./converters";
+export { frontendTools } from "./frontendTools";

package/src/useCloudRuntime.ts

@@ -0,0 +1,24 @@
+import { AssistantCloud } from "@assistant-ui/react";
+import {
+  useDataStreamRuntime,
+  UseDataStreamRuntimeOptions,
+} from "./useDataStreamRuntime";
+
+type UseCloudRuntimeOptions = Omit<UseDataStreamRuntimeOptions, "api"> & {
+  cloud: AssistantCloud;
+  assistantId: string;
+};
+
+/**
+ * @deprecated This is under active development and not yet ready for prod use.
+ */
+export const useCloudRuntime = (options: UseCloudRuntimeOptions) => {
+  const opts = options.cloud.runs.__internal_getAssistantOptions(
+    options.assistantId,
+  );
+
+  return useDataStreamRuntime({
+    ...options,
+    ...opts,
+  });
+};

package/src/useDataStreamRuntime.ts

@@ -0,0 +1,159 @@
+"use client";
+
+import { toLanguageModelMessages } from "./converters";
+import {
+  AssistantRuntime,
+  ChatModelAdapter,
+  ChatModelRunOptions,
+  INTERNAL,
+  LocalRuntimeOptions,
+  ThreadMessage,
+  Tool,
+  useLocalRuntime,
+} from "@assistant-ui/react";
+import { z } from "zod";
+import { JSONSchema7 } from "json-schema";
+import {
+  AssistantMessageAccumulator,
+  DataStreamDecoder,
+  unstable_toolResultStream,
+} from "assistant-stream";
+import { asAsyncIterableStream } from "assistant-stream/utils";
+
+const { splitLocalRuntimeOptions } = INTERNAL;
+
+type HeadersValue = Record<string, string> | Headers;
+
+export type UseDataStreamRuntimeOptions = {
+  api: string;
+  onResponse?: (response: Response) => void | Promise<void>;
+  onFinish?: (message: ThreadMessage) => void;
+  onError?: (error: Error) => void;
+  onCancel?: () => void;
+  credentials?: RequestCredentials;
+  headers?: HeadersValue | (() => Promise<HeadersValue>);
+  body?: object;
+  sendExtraMessageFields?: boolean;
+} & LocalRuntimeOptions;
+
+type DataStreamRuntimeRequestOptions = {
+  messages: any[];
+  tools: any;
+  system?: string | undefined;
+  runConfig?: any;
+  unstable_assistantMessageId?: string;
+  state?: any;
+};
+
+const toAISDKTools = (tools: Record<string, Tool>) => {
+  return Object.fromEntries(
+    Object.entries(tools).map(([name, tool]) => [
+      name,
+      {
+        ...(tool.description ? { description: tool.description } : undefined),
+        parameters: (tool.parameters instanceof z.ZodType
+          ? z.toJSONSchema(tool.parameters)
+          : tool.parameters) as JSONSchema7,
+      },
+    ]),
+  );
+};
+
+const getEnabledTools = (tools: Record<string, Tool>) => {
+  return Object.fromEntries(
+    Object.entries(tools).filter(
+      ([, tool]) => !tool.disabled && tool.type !== "backend",
+    ),
+  );
+};
+
+class DataStreamRuntimeAdapter implements ChatModelAdapter {
+  constructor(
+    private options: Omit<
+      UseDataStreamRuntimeOptions,
+      keyof LocalRuntimeOptions
+    >,
+  ) {}
+
+  async *run({
+    messages,
+    runConfig,
+    abortSignal,
+    context,
+    unstable_assistantMessageId,
+    unstable_getMessage,
+  }: ChatModelRunOptions) {
+    const headersValue =
+      typeof this.options.headers === "function"
+        ? await this.options.headers()
+        : this.options.headers;
+
+    abortSignal.addEventListener(
+      "abort",
+      () => {
+        if (!abortSignal.reason?.detach) this.options.onCancel?.();
+      },
+      { once: true },
+    );
+
+    const headers = new Headers(headersValue);
+    headers.set("Content-Type", "application/json");
+
+    const result = await fetch(this.options.api, {
+      method: "POST",
+      headers,
+      credentials: this.options.credentials ?? "same-origin",
+      body: JSON.stringify({
+        system: context.system,
+        messages: toLanguageModelMessages(messages, {
+          unstable_includeId: this.options.sendExtraMessageFields,
+        }) as DataStreamRuntimeRequestOptions["messages"],
+        tools: toAISDKTools(
+          getEnabledTools(context.tools ?? {}),
+        ) as unknown as DataStreamRuntimeRequestOptions["tools"],
+        ...(unstable_assistantMessageId ? { unstable_assistantMessageId } : {}),
+        runConfig,
+        state: unstable_getMessage().metadata.unstable_state || undefined,
+        ...context.callSettings,
+        ...context.config,
+        ...this.options.body,
+      } satisfies DataStreamRuntimeRequestOptions),
+      signal: abortSignal,
+    });
+
+    await this.options.onResponse?.(result);
+
+    try {
+      if (!result.ok) {
+        throw new Error(`Status ${result.status}: ${await result.text()}`);
+      }
+      if (!result.body) {
+        throw new Error("Response body is null");
+      }
+
+      const stream = result.body
+        .pipeThrough(new DataStreamDecoder())
+        .pipeThrough(unstable_toolResultStream(context.tools, abortSignal))
+        .pipeThrough(new AssistantMessageAccumulator());
+
+      yield* asAsyncIterableStream(stream);
+
+      this.options.onFinish?.(unstable_getMessage());
+    } catch (error: unknown) {
+      this.options.onError?.(error as Error);
+      throw error;
+    }
+  }
+}
+
+export const useDataStreamRuntime = (
+  options: UseDataStreamRuntimeOptions,
+): AssistantRuntime => {
+  const { localRuntimeOptions, otherOptions } =
+    splitLocalRuntimeOptions(options);
+
+  return useLocalRuntime(
+    new DataStreamRuntimeAdapter(otherOptions),
+    localRuntimeOptions,
+  );
+};