@assistant-ui/react-ai-sdk 0.10.14 → 0.10.16

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 ReasoningContentPart,\n  type ToolCallContentPart,\n  type TextContentPart,\n  type CompleteAttachment,\n  type SourceContentPart,\n  type FileContentPart,\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 TextContentPart;\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 ToolCallContentPart;\n          case \"reasoning\":\n            return {\n              type: \"reasoning\",\n              text: part.reasoning,\n            } satisfies ReasoningContentPart;\n          case \"source\":\n            return {\n              type: \"source\",\n              ...part.source,\n            } satisfies SourceContentPart;\n          case \"file\":\n            return {\n              type: \"file\",\n              data: part.data,\n              mimeType: part.mimeType,\n            } satisfies FileContentPart;\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 TextContentPart,\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          | ToolCallContentPart\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 { 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":[]}

package/dist/ui/utils/sliceMessagesUntil.js

@@ -4,7 +4,7 @@ var sliceMessagesUntil = (messages, messageId) => {
   let messageIdx = messages.findIndex((m) => m.id === messageId);
   if (messageIdx === -1)
     throw new Error(
-      "useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui."
+      "useVercelAIThreadState: Message not found. This is likely an internal bug in assistant-ui."
     );
   while (messages[messageIdx + 1]?.role === "assistant") {
     messageIdx++;

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 liekly 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 { 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":[]}

package/dist/useChatRuntime.d.ts

@@ -1,4 +1,16 @@
-import { EdgeRuntimeOptions } from "@assistant-ui/react-edge";
-export type UseChatRuntimeOptions = Omit<EdgeRuntimeOptions, "unstable_AISDKInterop">;
-export declare const useChatRuntime: (options: UseChatRuntimeOptions) => import("@assistant-ui/react").AssistantRuntime;
+import { AssistantRuntime, LocalRuntimeOptions, ThreadMessage } from "@assistant-ui/react";
+type HeadersValue = Record<string, string> | Headers;
+export type UseChatRuntimeOptions = {
+    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 useChatRuntime: (options: UseChatRuntimeOptions) => AssistantRuntime;
+export {};
 //# sourceMappingURL=useChatRuntime.d.ts.map

package/dist/useChatRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useChatRuntime.d.ts","sourceRoot":"","sources":["../src/useChatRuntime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAkB,MAAM,0BAA0B,CAAC;AAE9E,MAAM,MAAM,qBAAqB,GAAG,IAAI,CACtC,kBAAkB,EAClB,uBAAuB,CACxB,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,SAAS,qBAAqB,mDAK5D,CAAC"}
+{"version":3,"file":"useChatRuntime.d.ts","sourceRoot":"","sources":["../src/useChatRuntime.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,gBAAgB,EAIhB,mBAAmB,EACnB,aAAa,EAGd,MAAM,qBAAqB,CAAC;AAa7B,KAAK,YAAY,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC;AAErD,MAAM,MAAM,qBAAqB,GAAG;IAClC,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;AA6GxB,eAAO,MAAM,cAAc,GACzB,SAAS,qBAAqB,KAC7B,gBAQF,CAAC"}

package/dist/useChatRuntime.js

@@ -1,10 +1,104 @@
+"use client";
+
 // src/useChatRuntime.ts
-import { useEdgeRuntime } from "@assistant-ui/react-edge";
+import { toLanguageModelMessages } from "./converters/index.js";
+import {
+  INTERNAL,
+  useLocalRuntime
+} from "@assistant-ui/react";
+import { z } from "zod";
+import zodToJsonSchema from "zod-to-json-schema";
+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 ? zodToJsonSchema(tool.parameters) : tool.parameters
+      }
+    ])
+  );
+};
+var getEnabledTools = (tools) => {
+  return Object.fromEntries(
+    Object.entries(tools).filter(
+      ([, tool]) => !tool.disabled && tool.type !== "backend"
+    )
+  );
+};
+var ChatRuntimeAdapter = 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 useChatRuntime = (options) => {
-  return useEdgeRuntime({
-    ...options,
-    unstable_AISDKInterop: "v2"
-  });
+  const { localRuntimeOptions, otherOptions } = splitLocalRuntimeOptions(options);
+  return useLocalRuntime(
+    new ChatRuntimeAdapter(otherOptions),
+    localRuntimeOptions
+  );
 };
 export {
   useChatRuntime

package/dist/useChatRuntime.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/useChatRuntime.ts"],"sourcesContent":["import { EdgeRuntimeOptions, useEdgeRuntime } from \"@assistant-ui/react-edge\";\n\nexport type UseChatRuntimeOptions = Omit<\n  EdgeRuntimeOptions,\n  \"unstable_AISDKInterop\"\n>;\n\nexport const useChatRuntime = (options: UseChatRuntimeOptions) => {\n  return useEdgeRuntime({\n    ...options,\n    unstable_AISDKInterop: \"v2\",\n  });\n};\n"],"mappings":";AAAA,SAA6B,sBAAsB;AAO5C,IAAM,iBAAiB,CAAC,YAAmC;AAChE,SAAO,eAAe;AAAA,IACpB,GAAG;AAAA,IACH,uBAAuB;AAAA,EACzB,CAAC;AACH;","names":[]}
+{"version":3,"sources":["../src/useChatRuntime.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 zodToJsonSchema from \"zod-to-json-schema\";\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 UseChatRuntimeOptions = {\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 ChatRuntimeRequestOptions = {\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          ? zodToJsonSchema(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 ChatRuntimeAdapter implements ChatModelAdapter {\n  constructor(\n    private options: Omit<UseChatRuntimeOptions, keyof LocalRuntimeOptions>,\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 ChatRuntimeRequestOptions[\"messages\"],\n        tools: toAISDKTools(\n          getEnabledTools(context.tools ?? {}),\n        ) as unknown as ChatRuntimeRequestOptions[\"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 ChatRuntimeRequestOptions),\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 useChatRuntime = (\n  options: UseChatRuntimeOptions,\n): AssistantRuntime => {\n  const { localRuntimeOptions, otherOptions } =\n    splitLocalRuntimeOptions(options);\n\n  return useLocalRuntime(\n    new ChatRuntimeAdapter(otherOptions),\n    localRuntimeOptions,\n  );\n};\n"],"mappings":";;;AAEA,SAAS,+BAA+B;AACxC;AAAA,EAIE;AAAA,EAIA;AAAA,OACK;AACP,SAAS,SAAS;AAClB,OAAO,qBAAqB;AAE5B;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,gBAAgB,KAAK,UAAU,IAC/B,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,qBAAN,MAAqD;AAAA,EACnD,YACU,SACR;AADQ;AAAA,EACP;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,CAAqC;AAAA,MACrC,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,iBAAiB,CAC5B,YACqB;AACrB,QAAM,EAAE,qBAAqB,aAAa,IACxC,yBAAyB,OAAO;AAElC,SAAO;AAAA,IACL,IAAI,mBAAmB,YAAY;AAAA,IACnC;AAAA,EACF;AACF;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ai-sdk",
-  "version": "0.10.14",
+  "version": "0.10.16",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -18,16 +18,19 @@
   ],
   "sideEffects": false,
   "dependencies": {
+    "@ai-sdk/provider": "^1.1.3",
     "@ai-sdk/react": "*",
     "@ai-sdk/ui-utils": "*",
     "@radix-ui/react-use-callback-ref": "^1.1.1",
     "@types/json-schema": "^7.0.15",
-    "zod": "^3.25.64",
-    "zustand": "^5.0.5",
-    "@assistant-ui/react-edge": "0.2.12"
+    "assistant-stream": "^0.2.19",
+    "json-schema": "^0.4.0",
+    "zod": "^3.25.67",
+    "zod-to-json-schema": "^3.24.6",
+    "zustand": "^5.0.6"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.10.24",
+    "@assistant-ui/react": "^0.10.26",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc"
   },
@@ -37,12 +40,13 @@
     }
   },
   "devDependencies": {
-    "@types/node": "^24.0.1",
+    "@types/node": "^24.0.4",
     "@types/react": "^19.1.8",
     "eslint": "^9",
-    "eslint-config-next": "15.3.3",
+    "eslint-config-next": "15.3.4",
     "react": "^19.1.0",
     "tsx": "^4.20.3",
+    "@assistant-ui/react": "0.10.26",
     "@assistant-ui/x-buildutils": "0.0.1"
   },
   "publishConfig": {

package/src/converters/fromLanguageModelTools.ts

@@ -0,0 +1,19 @@
+import { LanguageModelV1FunctionTool } from "@ai-sdk/provider";
+import { Tool } from "@assistant-ui/react";
+
+/**
+ * @deprecated This is an internal API and may change without notice.
+ */
+export const fromLanguageModelTools = (
+  tools: LanguageModelV1FunctionTool[],
+): Record<string, Tool<any, any>> => {
+  return Object.fromEntries(
+    tools.map((tool) => [
+      tool.name,
+      {
+        description: tool.description,
+        parameters: tool.parameters,
+      },
+    ]),
+  );
+};

package/src/converters/index.ts

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

package/src/converters/toLanguageModelMessages.ts

@@ -0,0 +1,185 @@
+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": {
+              break; // reasoning, source, and file 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/converters/toLanguageModelTools.ts

@@ -0,0 +1,21 @@
+import { LanguageModelV1FunctionTool } from "@ai-sdk/provider";
+import { JSONSchema7 } from "json-schema";
+import { z } from "zod";
+import zodToJsonSchema from "zod-to-json-schema";
+import { Tool } from "@assistant-ui/react";
+
+/**
+ * @deprecated This is an internal API and may change without notice.
+ */
+export const toLanguageModelTools = (
+  tools: Record<string, Tool<any, any>>,
+): LanguageModelV1FunctionTool[] => {
+  return Object.entries(tools).map(([name, tool]) => ({
+    type: "function",
+    name,
+    ...(tool.description ? { description: tool.description } : undefined),
+    parameters: (tool.parameters instanceof z.ZodType
+      ? zodToJsonSchema(tool.parameters)
+      : tool.parameters) as JSONSchema7,
+  }));
+};

package/src/dangerous-in-browser/DangerousInBrowserAdapter.ts

@@ -0,0 +1,41 @@
+import { asAsyncIterableStream } from "assistant-stream/utils";
+import {
+  AssistantMessageAccumulator,
+  unstable_toolResultStream,
+} from "assistant-stream";
+import { ChatModelAdapter, ChatModelRunOptions } from "@assistant-ui/react";
+import {
+  CreateEdgeRuntimeAPIOptions,
+  getEdgeRuntimeStream,
+} from "./createEdgeRuntimeAPI";
+
+export type DangerousInBrowserAdapterOptions = CreateEdgeRuntimeAPIOptions;
+
+export class DangerousInBrowserAdapter implements ChatModelAdapter {
+  constructor(private options: DangerousInBrowserAdapterOptions) {}
+
+  async *run({ messages, abortSignal, context }: ChatModelRunOptions) {
+    const res = await getEdgeRuntimeStream({
+      options: {
+        ...this.options,
+        ...(context.tools && { tools: context.tools }),
+      },
+      abortSignal,
+      requestData: {
+        system: context.system,
+        messages: messages,
+        ...Object.fromEntries(
+          Object.entries({ ...context.callSettings, ...context.config }).filter(
+            ([_, v]) => v !== undefined,
+          ),
+        ),
+      },
+    });
+
+    const stream = res
+      .pipeThrough(unstable_toolResultStream(context.tools, abortSignal))
+      .pipeThrough(new AssistantMessageAccumulator());
+
+    yield* asAsyncIterableStream(stream);
+  }
+}