@assistant-ui/react 0.4.3 → 0.4.4

package/dist/index.js

@@ -43,6 +43,7 @@ __export(src_exports, {
   ComposerPrimitive: () => composer_exports,
   ContentPart: () => content_part_default,
   ContentPartPrimitive: () => contentPart_exports,
+  EdgeChatAdapter: () => EdgeChatAdapter,
   EditComposer: () => edit_composer_default,
   INTERNAL: () => internal_exports,
   MessagePrimitive: () => message_exports,
@@ -52,8 +53,11 @@ __export(src_exports, {
   ThreadWelcome: () => thread_welcome_default,
   UserActionBar: () => user_action_bar_default,
   UserMessage: () => user_message_default,
+  fromCoreMessages: () => fromCoreMessages,
+  fromLanguageModelMessages: () => fromLanguageModelMessages,
   makeAssistantTool: () => makeAssistantTool,
   makeAssistantToolUI: () => makeAssistantToolUI,
+  toLanguageModelMessages: () => toLanguageModelMessages,
   useActionBarCopy: () => useActionBarCopy,
   useActionBarEdit: () => useActionBarEdit,
   useActionBarReload: () => useActionBarReload,
@@ -113,7 +117,7 @@ var import_zustand = require("zustand");
 
 // src/types/ModelConfigTypes.ts
 var mergeModelConfigs = (configSet) => {
-  const configs = Array.from(configSet).map((c) => c()).sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0));
+  const configs = Array.from(configSet).map((c) => c.getModelConfig()).sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0));
   return configs.reduce((acc, config) => {
     if (config.system) {
       if (acc.system) {
@@ -302,7 +306,8 @@ var makeThreadActionStore = (runtimeRef) => {
       startRun: (parentId) => runtimeRef.current.startRun(parentId),
       append: (message) => runtimeRef.current.append(message),
       cancelRun: () => runtimeRef.current.cancelRun(),
-      addToolResult: (options) => runtimeRef.current.addToolResult(options)
+      addToolResult: (options) => runtimeRef.current.addToolResult(options),
+      getRuntime: () => runtimeRef.current
     })
   );
 };
@@ -319,10 +324,7 @@ var ThreadProvider = ({
   children,
   runtime
 }) => {
-  const runtimeRef = (0, import_react3.useRef)(runtime);
-  (0, import_react3.useInsertionEffect)(() => {
-    runtimeRef.current = runtime;
-  });
+  const runtimeRef = (0, import_react3.useRef)(runtime.thread);
   const [context] = (0, import_react3.useState)(() => {
     const useThread = makeThreadStore(runtimeRef);
     const useThreadMessages = makeThreadMessagesStore(runtimeRef);
@@ -337,8 +339,16 @@ var ThreadProvider = ({
       useViewport
     };
   });
+  const thread = (0, import_react3.useSyncExternalStore)(
+    (0, import_react3.useCallback)((c) => runtime.subscribe(c), [runtime]),
+    () => runtime.thread,
+    () => runtime.thread
+  );
+  (0, import_react3.useInsertionEffect)(() => {
+    runtimeRef.current = thread;
+  });
   (0, import_react3.useEffect)(() => {
-    const onRuntimeUpdate = () => {
+    const onThreadUpdate = () => {
       context.useThread.setState(
         Object.freeze({
           isRunning: runtimeRef.current.isRunning
@@ -347,20 +357,11 @@ var ThreadProvider = ({
       );
       context.useThreadMessages.setState(Object.freeze(runtimeRef.current.messages), true);
     };
-    onRuntimeUpdate();
-    return runtime.subscribe(onRuntimeUpdate);
-  }, [context, runtime]);
-  const subscribe = (0, import_react3.useCallback)(
-    (c) => runtime.subscribe(c),
-    [runtime]
-  );
-  const RuntimeSynchronizer = (0, import_react3.useSyncExternalStore)(
-    subscribe,
-    () => runtime.unstable_synchronizer,
-    () => void 0
-  );
+    onThreadUpdate();
+    return thread.subscribe(onThreadUpdate);
+  }, [context, thread]);
   return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(ThreadContext.Provider, { value: context, children: [
-    RuntimeSynchronizer && /* @__PURE__ */ (0, import_jsx_runtime.jsx)(RuntimeSynchronizer, {}),
+    thread.unstable_synchronizer && /* @__PURE__ */ (0, import_jsx_runtime.jsx)(thread.unstable_synchronizer, {}),
     children
   ] });
 };
@@ -369,7 +370,8 @@ var ThreadProvider = ({
 var import_zustand8 = require("zustand");
 var makeAssistantActionsStore = (runtimeRef) => (0, import_zustand8.create)(
   () => Object.freeze({
-    switchToThread: () => runtimeRef.current.switchToThread(null)
+    switchToThread: () => runtimeRef.current.switchToThread(null),
+    getRuntime: () => runtimeRef.current
   })
 );
 
@@ -386,10 +388,10 @@ var AssistantProvider = ({ children, runtime }) => {
     const useAssistantActions = makeAssistantActionsStore(runtimeRef);
     return { useModelConfig, useToolUIs, useAssistantActions };
   });
-  const getModelCOnfig = context.useModelConfig((c) => c.getModelConfig);
+  const getModelConfig = context.useModelConfig();
   (0, import_react4.useEffect)(() => {
-    return runtime.registerModelConfigProvider(getModelCOnfig);
-  }, [runtime, getModelCOnfig]);
+    return runtime.registerModelConfigProvider(getModelConfig);
+  }, [runtime, getModelConfig]);
   return /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(AssistantContext.Provider, { value: context, children: /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(ThreadProvider, { runtime, children }) });
 };
 
@@ -499,7 +501,9 @@ var useAssistantTool = (tool) => {
         [tool.toolName]: rest
       }
     };
-    const unsub1 = registerModelConfigProvider(() => config);
+    const unsub1 = registerModelConfigProvider({
+      getModelConfig: () => config
+    });
     const unsub2 = render ? setToolUI(toolName, render) : void 0;
     return () => {
       unsub1();
@@ -549,7 +553,7 @@ var useAssistantInstructions = (instruction) => {
     const config = {
       system: instruction
     };
-    return registerModelConfigProvider(() => config);
+    return registerModelConfigProvider({ getModelConfig: () => config });
   }, [registerModelConfigProvider, instruction]);
 };
 
@@ -1968,7 +1972,7 @@ var ThreadPrimitiveSuggestion = createActionButton(
 );
 
 // src/runtimes/local/useLocalRuntime.tsx
-var import_react50 = require("react");
+var import_react53 = require("react");
 
 // src/utils/idUtils.tsx
 var import_non_secure = require("nanoid/non-secure");
@@ -2146,45 +2150,14 @@ var BaseAssistantRuntime = class {
   constructor(_thread) {
     this._thread = _thread;
     this._thread = _thread;
-    this._unsubscribe = this._thread.subscribe(this.subscriptionHandler);
   }
-  get capabilities() {
-    return this._thread.capabilities;
-  }
-  _unsubscribe;
   get thread() {
     return this._thread;
   }
   set thread(thread) {
-    this._unsubscribe();
     this._thread = thread;
-    this._unsubscribe = this._thread.subscribe(this.subscriptionHandler);
     this.subscriptionHandler();
   }
-  get messages() {
-    return this.thread.messages;
-  }
-  get isRunning() {
-    return this.thread.isRunning;
-  }
-  getBranches(messageId) {
-    return this.thread.getBranches(messageId);
-  }
-  switchToBranch(branchId) {
-    return this.thread.switchToBranch(branchId);
-  }
-  append(message) {
-    return this.thread.append(message);
-  }
-  startRun(parentId) {
-    return this.thread.startRun(parentId);
-  }
-  cancelRun() {
-    return this.thread.cancelRun();
-  }
-  addToolResult(options) {
-    return this.thread.addToolResult(options);
-  }
   _subscriptions = /* @__PURE__ */ new Set();
   subscribe(callback) {
     this._subscriptions.add(callback);
@@ -2193,32 +2166,133 @@ var BaseAssistantRuntime = class {
   subscriptionHandler = () => {
     for (const callback of this._subscriptions) callback();
   };
-  get unstable_synchronizer() {
-    return this.thread.unstable_synchronizer;
-  }
 };
 
+// src/internal.ts
+var internal_exports = {};
+__export(internal_exports, {
+  BaseAssistantRuntime: () => BaseAssistantRuntime,
+  MessageRepository: () => MessageRepository,
+  ProxyConfigProvider: () => ProxyConfigProvider,
+  TooltipIconButton: () => TooltipIconButton,
+  generateId: () => generateId,
+  useSmooth: () => useSmooth
+});
+
+// src/ui/base/tooltip-icon-button.tsx
+var import_react52 = require("react");
+
+// src/ui/base/tooltip.tsx
+var TooltipPrimitive = __toESM(require("@radix-ui/react-tooltip"));
+
+// src/ui/utils/withDefaults.tsx
+var import_react50 = require("react");
+var import_classnames = __toESM(require("classnames"));
+var import_jsx_runtime25 = require("react/jsx-runtime");
+var withDefaultProps = ({
+  className,
+  ...defaultProps
+}) => ({ className: classNameProp, ...props }) => {
+  return {
+    className: (0, import_classnames.default)(className, classNameProp),
+    ...defaultProps,
+    ...props
+  };
+};
+var withDefaults = (Component, defaultProps) => {
+  const getProps = withDefaultProps(defaultProps);
+  const WithDefaults = (0, import_react50.forwardRef)(
+    (props, ref) => {
+      const ComponentAsAny = Component;
+      return /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(ComponentAsAny, { ...getProps(props), ref });
+    }
+  );
+  WithDefaults.displayName = "withDefaults(" + (typeof Component === "string" ? Component : Component.displayName) + ")";
+  return WithDefaults;
+};
+
+// src/ui/base/tooltip.tsx
+var import_jsx_runtime26 = require("react/jsx-runtime");
+var Tooltip = (props) => {
+  return /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(TooltipPrimitive.Provider, { children: /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(TooltipPrimitive.Root, { ...props }) });
+};
+Tooltip.displayName = "Tooltip";
+var TooltipTrigger = TooltipPrimitive.Trigger;
+var TooltipContent = withDefaults(TooltipPrimitive.Content, {
+  sideOffset: 4,
+  className: "aui-tooltip-content"
+});
+TooltipContent.displayName = "TooltipContent";
+
+// src/ui/base/button.tsx
+var import_class_variance_authority = require("class-variance-authority");
+var import_react_primitive11 = require("@radix-ui/react-primitive");
+var import_react51 = require("react");
+var import_jsx_runtime27 = require("react/jsx-runtime");
+var buttonVariants = (0, import_class_variance_authority.cva)("aui-button", {
+  variants: {
+    variant: {
+      default: "aui-button-primary",
+      outline: "aui-button-outline",
+      ghost: "aui-button-ghost"
+    },
+    size: {
+      default: "aui-button-medium",
+      icon: "aui-button-icon"
+    }
+  },
+  defaultVariants: {
+    variant: "default",
+    size: "default"
+  }
+});
+var Button = (0, import_react51.forwardRef)(
+  ({ className, variant, size, ...props }, ref) => {
+    return /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(
+      import_react_primitive11.Primitive.button,
+      {
+        className: buttonVariants({ variant, size, className }),
+        ...props,
+        ref
+      }
+    );
+  }
+);
+Button.displayName = "Button";
+
+// src/ui/base/tooltip-icon-button.tsx
+var import_jsx_runtime28 = require("react/jsx-runtime");
+var TooltipIconButton = (0, import_react52.forwardRef)(({ children, tooltip, side = "bottom", ...rest }, ref) => {
+  return /* @__PURE__ */ (0, import_jsx_runtime28.jsxs)(Tooltip, { children: [
+    /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime28.jsxs)(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
+      children,
+      /* @__PURE__ */ (0, import_jsx_runtime28.jsx)("span", { className: "aui-sr-only", children: tooltip })
+    ] }) }),
+    /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(TooltipContent, { side, children: tooltip })
+  ] });
+});
+TooltipIconButton.displayName = "TooltipIconButton";
+
 // src/runtimes/local/LocalRuntime.tsx
 var LocalRuntime = class extends BaseAssistantRuntime {
-  _configProviders;
+  _proxyConfigProvider;
   constructor(adapter) {
-    const configProviders = /* @__PURE__ */ new Set();
-    super(new LocalThreadRuntime(configProviders, adapter));
-    this._configProviders = configProviders;
+    const proxyConfigProvider = new ProxyConfigProvider();
+    super(new LocalThreadRuntime(proxyConfigProvider, adapter));
+    this._proxyConfigProvider = proxyConfigProvider;
   }
   set adapter(adapter) {
     this.thread.adapter = adapter;
   }
   registerModelConfigProvider(provider) {
-    this._configProviders.add(provider);
-    return () => this._configProviders.delete(provider);
+    return this._proxyConfigProvider.registerModelConfigProvider(provider);
   }
   switchToThread(threadId) {
     if (threadId) {
       throw new Error("LocalRuntime does not yet support switching threads");
     }
     return this.thread = new LocalThreadRuntime(
-      this._configProviders,
+      this._proxyConfigProvider,
       this.thread.adapter
     );
   }
@@ -2230,8 +2304,8 @@ var CAPABILITIES = Object.freeze({
   copy: true
 });
 var LocalThreadRuntime = class {
-  constructor(_configProviders, adapter) {
-    this._configProviders = _configProviders;
+  constructor(configProvider, adapter) {
+    this.configProvider = configProvider;
     this.adapter = adapter;
   }
   _subscriptions = /* @__PURE__ */ new Set();
@@ -2252,6 +2326,10 @@ var LocalThreadRuntime = class {
     this.notifySubscribers();
   }
   async append(message) {
+    if (message.role !== "user")
+      throw new Error(
+        "Only appending user messages are supported in LocalRuntime. This is likely an internal bug in assistant-ui."
+      );
     const userMessageId = generateId();
     const userMessage = {
       id: userMessageId,
@@ -2263,11 +2341,10 @@ var LocalThreadRuntime = class {
     await this.startRun(userMessageId);
   }
   async startRun(parentId) {
-    const id = generateId();
     this.repository.resetHead(parentId);
     const messages = this.repository.getMessages();
     const message = {
-      id,
+      id: generateId(),
       role: "assistant",
       status: { type: "in_progress" },
       content: [{ type: "text", text: "" }],
@@ -2286,7 +2363,7 @@ var LocalThreadRuntime = class {
       const result = await this.adapter.run({
         messages,
         abortSignal: this.abortController.signal,
-        config: mergeModelConfigs(this._configProviders),
+        config: this.configProvider.getModelConfig(),
         onUpdate: updateHandler
       });
       if (result !== void 0) {
@@ -2301,7 +2378,7 @@ var LocalThreadRuntime = class {
     } catch (e) {
       message.status = { type: "error", error: e };
       this.repository.addOrUpdateMessage(parentId, { ...message });
-      console.error(e);
+      throw e;
     } finally {
       this.abortController = null;
       this.notifySubscribers();
@@ -2311,7 +2388,6 @@ var LocalThreadRuntime = class {
     if (!this.abortController) return;
     this.abortController.abort();
     this.abortController = null;
-    this.notifySubscribers();
   }
   notifySubscribers() {
     for (const callback of this._subscriptions) callback();
@@ -2323,7 +2399,7 @@ var LocalThreadRuntime = class {
   addToolResult({ messageId, toolCallId, result }) {
     const { parentId, message } = this.repository.getMessage(messageId);
     if (message.role !== "assistant")
-      throw new Error("Tried to add tool re^sult to non-assistant message");
+      throw new Error("Tried to add tool result to non-assistant message");
     let found = false;
     const newContent = message.content.map((c) => {
       if (c.type !== "tool-call") return c;
@@ -2345,27 +2421,55 @@ var LocalThreadRuntime = class {
 
 // src/runtimes/local/useLocalRuntime.tsx
 var useLocalRuntime = (adapter) => {
-  const [runtime] = (0, import_react50.useState)(() => new LocalRuntime(adapter));
-  (0, import_react50.useInsertionEffect)(() => {
+  const [runtime] = (0, import_react53.useState)(() => new LocalRuntime(adapter));
+  (0, import_react53.useInsertionEffect)(() => {
     runtime.adapter = adapter;
   });
   return runtime;
 };
 
+// src/runtimes/edge/useEdgeRuntime.ts
+var import_react54 = require("react");
+
+// src/runtimes/edge/converters/toCoreMessages.ts
+var toCoreMessages = (message) => {
+  return message.map((message2) => {
+    return {
+      role: message2.role,
+      content: message2.content.map((part) => {
+        if (part.type === "ui") throw new Error("UI parts are not supported");
+        return part;
+      })
+    };
+  });
+};
+
+// src/runtimes/edge/converters/toLanguageModelTools.ts
+var import_zod = require("zod");
+var import_zod_to_json_schema = __toESM(require("zod-to-json-schema"));
+var toLanguageModelTools = (tools) => {
+  if (!tools) return [];
+  return Object.entries(tools).map(([name, tool]) => ({
+    type: "function",
+    name,
+    ...tool.description ? { description: tool.description } : void 0,
+    parameters: tool.parameters instanceof import_zod.z.ZodType ? (0, import_zod_to_json_schema.default)(tool.parameters) : tool.parameters
+  }));
+};
+
 // src/runtimes/edge/streams/assistantDecoderStream.ts
 function assistantDecoderStream() {
   let currentToolCall;
   return new TransformStream({
     transform(chunk, controller) {
-      const [code, valueJson] = parseStreamPart(chunk);
-      const value = JSON.parse(valueJson);
+      const [code, value] = parseStreamPart(chunk);
       if (currentToolCall && code !== "2" /* ToolCallArgsTextDelta */ && code !== "E" /* Error */) {
         controller.enqueue({
           type: "tool-call",
           toolCallType: "function",
           toolCallId: currentToolCall.id,
           toolName: currentToolCall.name,
-          args: JSON.parse(currentToolCall.argsText)
+          args: currentToolCall.argsText
         });
         currentToolCall = void 0;
       }
@@ -2378,12 +2482,12 @@ function assistantDecoderStream() {
           break;
         }
         case "1" /* ToolCallBegin */: {
-          const { id, name } = JSON.parse(value);
+          const { id, name } = value;
           currentToolCall = { id, name, argsText: "" };
           break;
         }
         case "2" /* ToolCallArgsTextDelta */: {
-          const delta = JSON.parse(value);
+          const delta = value;
           currentToolCall.argsText += delta;
           controller.enqueue({
             type: "tool-call-delta",
@@ -2397,14 +2501,14 @@ function assistantDecoderStream() {
         case "F" /* Finish */: {
           controller.enqueue({
             type: "finish",
-            ...JSON.parse(value)
+            ...value
           });
           break;
         }
         case "E" /* Error */: {
           controller.enqueue({
             type: "error",
-            error: JSON.parse(value)
+            error: value
           });
           break;
         }
@@ -2421,7 +2525,7 @@ var parseStreamPart = (part) => {
   if (index === -1) throw new Error("Invalid stream part");
   return [
     part.slice(0, index),
-    part.slice(index + 1)
+    JSON.parse(part.slice(index + 1))
   ];
 };
 
@@ -2806,7 +2910,7 @@ function runResultStream() {
             message,
             toolCallId,
             toolName,
-            parsePartialJson(currentToolCall.argsText)
+            currentToolCall.argsText
           );
           controller.enqueue(message);
           break;
@@ -2854,7 +2958,7 @@ var appendOrUpdateText = (message, textDelta) => {
     content: contentParts.concat([contentPart])
   };
 };
-var appendOrUpdateToolCall = (message, toolCallId, toolName, args) => {
+var appendOrUpdateToolCall = (message, toolCallId, toolName, argsText) => {
   let contentParts = message.content;
   let contentPart = message.content.at(-1);
   if (contentPart?.type !== "tool-call" || contentPart.toolCallId !== toolCallId) {
@@ -2862,13 +2966,15 @@ var appendOrUpdateToolCall = (message, toolCallId, toolName, args) => {
       type: "tool-call",
       toolCallId,
       toolName,
-      args
+      argsText,
+      args: parsePartialJson(argsText)
     };
   } else {
     contentParts = contentParts.slice(0, -1);
     contentPart = {
       ...contentPart,
-      args
+      argsText,
+      args: parsePartialJson(argsText)
     };
   }
   return {
@@ -2911,11 +3017,8 @@ var appendOrUpdateFinish = (message, chunk) => {
   };
 };
 
-// src/runtimes/edge/useEdgeRuntime.ts
-var import_react51 = require("react");
-
 // src/runtimes/edge/streams/toolResultStream.ts
-var import_zod = require("zod");
+var import_zod2 = require("zod");
 var import_secure_json_parse2 = __toESM(require("secure-json-parse"));
 function toolResultStream(tools) {
   const toolCallExecutions = /* @__PURE__ */ new Map();
@@ -2927,9 +3030,9 @@ function toolResultStream(tools) {
         case "tool-call": {
           const { toolCallId, toolCallType, toolName, args: argsText } = chunk;
           const tool = tools?.[toolName];
-          if (!tool) return;
+          if (!tool || !tool.execute) return;
           const args = import_secure_json_parse2.default.parse(argsText);
-          if (tool.parameters instanceof import_zod.z.ZodType) {
+          if (tool.parameters instanceof import_zod2.z.ZodType) {
             const result = tool.parameters.safeParse(args);
             if (!result.success) {
               controller.enqueue({
@@ -2981,9 +3084,7 @@ function toolResultStream(tools) {
   });
 }
 
-// src/runtimes/edge/useEdgeRuntime.ts
-var import_zod2 = require("zod");
-var import_zod_to_json_schema = require("zod-to-json-schema");
+// src/runtimes/edge/EdgeChatAdapter.ts
 function asAsyncIterable(source) {
   return {
     [Symbol.asyncIterator]: () => {
@@ -2997,28 +3098,20 @@ function asAsyncIterable(source) {
     }
   };
 }
-var toSerializableTools = (tools) => {
-  if (!tools) return [];
-  return Object.entries(tools).map(([name, tool]) => ({
-    type: "function",
-    name,
-    ...tool.description ? { description: tool.description } : void 0,
-    parameters: tool.parameters instanceof import_zod2.z.ZodType ? (0, import_zod_to_json_schema.zodToJsonSchema)(tool.parameters) : tool.parameters
-  }));
-};
-var createEdgeChatAdapter = ({
-  api
-}) => ({
-  run: async ({ messages, abortSignal, config, onUpdate }) => {
-    const result = await fetch(api, {
+var EdgeChatAdapter = class {
+  constructor(options) {
+    this.options = options;
+  }
+  async run({ messages, abortSignal, config, onUpdate }) {
+    const result = await fetch(this.options.api, {
       method: "POST",
       headers: {
         "Content-Type": "application/json"
       },
       body: JSON.stringify({
         system: config.system,
-        messages,
-        tools: toSerializableTools(
+        messages: toCoreMessages(messages),
+        tools: toLanguageModelTools(
           config.tools
         )
       }),
@@ -3033,134 +3126,274 @@ var createEdgeChatAdapter = ({
       throw new Error("No data received from Edge Runtime");
     return update;
   }
-});
+};
+
+// src/runtimes/edge/useEdgeRuntime.ts
 var useEdgeRuntime = (options) => {
-  const adapter = (0, import_react51.useMemo)(() => createEdgeChatAdapter(options), [options]);
+  const [adapter] = (0, import_react54.useState)(() => new EdgeChatAdapter(options));
   return useLocalRuntime(adapter);
 };
 
+// src/runtimes/edge/converters/toLanguageModelMessages.ts
+var assistantMessageSplitter = () => {
+  const stash = [];
+  let assistantMessage = {
+    role: "assistant",
+    content: []
+  };
+  let toolMessage = {
+    role: "tool",
+    content: []
+  };
+  return {
+    addTextContentPart: (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
+      });
+      if (part.result) {
+        toolMessage.content.push({
+          type: "tool-result",
+          toolCallId: part.toolCallId,
+          toolName: part.toolName,
+          result: part.result
+          // isError
+        });
+      }
+    },
+    getMessages: () => {
+      if (toolMessage.content.length > 0) {
+        return [...stash, assistantMessage, toolMessage];
+      }
+      return [...stash, assistantMessage];
+    }
+  };
+};
+function toLanguageModelMessages(message) {
+  return message.flatMap((message2) => {
+    const role = message2.role;
+    switch (role) {
+      case "system": {
+        return [{ role: "system", content: message2.content[0].text }];
+      }
+      case "user": {
+        const msg = {
+          role: "user",
+          content: message2.content.map(
+            (part) => {
+              const type = part.type;
+              switch (type) {
+                case "text": {
+                  return part;
+                }
+                case "image": {
+                  return {
+                    type: "image",
+                    image: new URL(part.image)
+                  };
+                }
+                default: {
+                  const unhandledType = type;
+                  throw new Error(
+                    `Unspported content part type: ${unhandledType}`
+                  );
+                }
+              }
+            }
+          )
+        };
+        return [msg];
+      }
+      case "assistant": {
+        const splitter = assistantMessageSplitter();
+        for (const part of message2.content) {
+          const type = part.type;
+          switch (type) {
+            case "text": {
+              splitter.addTextContentPart(part);
+              break;
+            }
+            case "tool-call": {
+              splitter.addToolCallPart(part);
+              break;
+            }
+            default: {
+              const unhandledType = type;
+              throw new Error(`Unhandled content part type: ${unhandledType}`);
+            }
+          }
+        }
+        return splitter.getMessages();
+      }
+      default: {
+        const unhandledRole = role;
+        throw new Error(`Unknown message role: ${unhandledRole}`);
+      }
+    }
+  });
+}
+
+// src/runtimes/edge/converters/fromLanguageModelMessages.ts
+var fromLanguageModelMessages = (lm, mergeRoundtrips) => {
+  const messages = [];
+  for (const lmMessage of lm) {
+    const role = lmMessage.role;
+    switch (role) {
+      case "system": {
+        messages.push({
+          role: "system",
+          content: [
+            {
+              type: "text",
+              text: lmMessage.content
+            }
+          ]
+        });
+        break;
+      }
+      case "user": {
+        messages.push({
+          role: "user",
+          content: lmMessage.content.map((part) => {
+            const type = part.type;
+            switch (type) {
+              case "text": {
+                return {
+                  type: "text",
+                  text: part.text
+                };
+              }
+              case "image": {
+                if (part.image instanceof URL) {
+                  return {
+                    type: "image",
+                    image: part.image.href
+                  };
+                }
+                throw new Error("Only images with URL data are supported");
+              }
+              default: {
+                const unhandledType = type;
+                throw new Error(`Unknown content part type: ${unhandledType}`);
+              }
+            }
+          })
+        });
+        break;
+      }
+      case "assistant": {
+        const newContent = lmMessage.content.map((part) => {
+          if (part.type === "tool-call") {
+            return {
+              type: "tool-call",
+              toolCallId: part.toolCallId,
+              toolName: part.toolName,
+              argsText: JSON.stringify(part.args),
+              args: typeof part.args === "string" ? part.args : part.args
+            };
+          }
+          return part;
+        });
+        if (mergeRoundtrips) {
+          const previousMessage = messages[messages.length - 1];
+          if (previousMessage?.role === "assistant") {
+            previousMessage.content.push(...newContent);
+            break;
+          }
+        }
+        messages.push({
+          role: "assistant",
+          content: newContent
+        });
+        break;
+      }
+      case "tool": {
+        const previousMessage = messages[messages.length - 1];
+        if (previousMessage?.role !== "assistant")
+          throw new Error(
+            "A tool message must be preceded by an assistant message."
+          );
+        for (const tool of lmMessage.content) {
+          const toolCall = previousMessage.content.find(
+            (c) => c.type === "tool-call" && c.toolCallId === tool.toolCallId
+          );
+          if (!toolCall)
+            throw new Error("Received tool result for an unknown tool call.");
+          if (toolCall.toolName !== tool.toolName)
+            throw new Error("Tool call name mismatch.");
+          toolCall.result = tool.result;
+          if (tool.isError) {
+            toolCall.isError = true;
+          }
+        }
+        break;
+      }
+      default: {
+        const unhandledRole = role;
+        throw new Error(`Unknown message role: ${unhandledRole}`);
+      }
+    }
+  }
+  return messages;
+};
+
+// src/runtimes/edge/converters/fromCoreMessage.ts
+var fromCoreMessages = (message) => {
+  return message.map((message2) => {
+    return {
+      ...message2,
+      id: generateId(),
+      createdAt: /* @__PURE__ */ new Date(),
+      ...message2.role === "assistant" ? {
+        status: { type: "done" }
+      } : void 0
+    };
+  });
+};
+
 // src/ui/thread-config.tsx
-var import_react52 = require("react");
-var import_jsx_runtime25 = require("react/jsx-runtime");
-var ThreadConfigContext = (0, import_react52.createContext)({});
+var import_react55 = require("react");
+var import_jsx_runtime29 = require("react/jsx-runtime");
+var ThreadConfigContext = (0, import_react55.createContext)({});
 var useThreadConfig = () => {
-  return (0, import_react52.useContext)(ThreadConfigContext);
+  return (0, import_react55.useContext)(ThreadConfigContext);
 };
 var ThreadConfigProvider = ({
   children,
   config
 }) => {
   const assistant = useAssistantContext({ optional: true });
-  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(import_jsx_runtime25.Fragment, { children });
+  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ (0, import_jsx_runtime29.jsx)(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ (0, import_jsx_runtime29.jsx)(import_jsx_runtime29.Fragment, { children });
   if (!config?.runtime) return configProvider;
   if (assistant) {
     throw new Error(
       "You provided a runtime to <Thread> while simulataneously using <AssistantRuntimeProvider>. This is not allowed."
     );
   }
-  return /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
+  return /* @__PURE__ */ (0, import_jsx_runtime29.jsx)(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
 };
 ThreadConfigProvider.displayName = "ThreadConfigProvider";
 
 // src/ui/assistant-action-bar.tsx
 var import_react56 = require("react");
 var import_lucide_react = require("lucide-react");
-
-// src/ui/base/tooltip-icon-button.tsx
-var import_react55 = require("react");
-
-// src/ui/base/tooltip.tsx
-var TooltipPrimitive = __toESM(require("@radix-ui/react-tooltip"));
-
-// src/ui/utils/withDefaults.tsx
-var import_react53 = require("react");
-var import_classnames = __toESM(require("classnames"));
-var import_jsx_runtime26 = require("react/jsx-runtime");
-var withDefaultProps = ({
-  className,
-  ...defaultProps
-}) => ({ className: classNameProp, ...props }) => {
-  return {
-    className: (0, import_classnames.default)(className, classNameProp),
-    ...defaultProps,
-    ...props
-  };
-};
-var withDefaults = (Component, defaultProps) => {
-  const getProps = withDefaultProps(defaultProps);
-  const WithDefaults = (0, import_react53.forwardRef)(
-    (props, ref) => {
-      const ComponentAsAny = Component;
-      return /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(ComponentAsAny, { ...getProps(props), ref });
-    }
-  );
-  WithDefaults.displayName = "withDefaults(" + (typeof Component === "string" ? Component : Component.displayName) + ")";
-  return WithDefaults;
-};
-
-// src/ui/base/tooltip.tsx
-var import_jsx_runtime27 = require("react/jsx-runtime");
-var Tooltip = (props) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(TooltipPrimitive.Provider, { children: /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(TooltipPrimitive.Root, { ...props }) });
-};
-Tooltip.displayName = "Tooltip";
-var TooltipTrigger = TooltipPrimitive.Trigger;
-var TooltipContent = withDefaults(TooltipPrimitive.Content, {
-  sideOffset: 4,
-  className: "aui-tooltip-content"
-});
-TooltipContent.displayName = "TooltipContent";
-
-// src/ui/base/button.tsx
-var import_class_variance_authority = require("class-variance-authority");
-var import_react_primitive11 = require("@radix-ui/react-primitive");
-var import_react54 = require("react");
-var import_jsx_runtime28 = require("react/jsx-runtime");
-var buttonVariants = (0, import_class_variance_authority.cva)("aui-button", {
-  variants: {
-    variant: {
-      default: "aui-button-primary",
-      outline: "aui-button-outline",
-      ghost: "aui-button-ghost"
-    },
-    size: {
-      default: "aui-button-medium",
-      icon: "aui-button-icon"
-    }
-  },
-  defaultVariants: {
-    variant: "default",
-    size: "default"
-  }
-});
-var Button = (0, import_react54.forwardRef)(
-  ({ className, variant, size, ...props }, ref) => {
-    return /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(
-      import_react_primitive11.Primitive.button,
-      {
-        className: buttonVariants({ variant, size, className }),
-        ...props,
-        ref
-      }
-    );
-  }
-);
-Button.displayName = "Button";
-
-// src/ui/base/tooltip-icon-button.tsx
-var import_jsx_runtime29 = require("react/jsx-runtime");
-var TooltipIconButton = (0, import_react55.forwardRef)(({ children, tooltip, side = "bottom", ...rest }, ref) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime29.jsxs)(Tooltip, { children: [
-    /* @__PURE__ */ (0, import_jsx_runtime29.jsx)(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime29.jsxs)(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
-      children,
-      /* @__PURE__ */ (0, import_jsx_runtime29.jsx)("span", { className: "aui-sr-only", children: tooltip })
-    ] }) }),
-    /* @__PURE__ */ (0, import_jsx_runtime29.jsx)(TooltipContent, { side, children: tooltip })
-  ] });
-});
-TooltipIconButton.displayName = "TooltipIconButton";
-
-// src/ui/assistant-action-bar.tsx
 var import_jsx_runtime30 = require("react/jsx-runtime");
 var useAllowCopy = () => {
   const { assistantMessage: { allowCopy = true } = {} } = useThreadConfig();
@@ -3204,10 +3437,10 @@ var AssistantActionBarCopy = (0, import_react56.forwardRef)((props, ref) => {
   } = useThreadConfig();
   const allowCopy = useAllowCopy();
   if (!allowCopy) return null;
-  return /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime30.jsxs)(TooltipIconButton, { tooltip, ...props, ref, children: [
+  return /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ (0, import_jsx_runtime30.jsxs)(import_jsx_runtime30.Fragment, { children: [
     /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(message_exports.If, { copied: true, children: /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(import_lucide_react.CheckIcon, {}) }),
     /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(message_exports.If, { copied: false, children: /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(import_lucide_react.CopyIcon, {}) })
-  ] }) });
+  ] }) }) });
 });
 AssistantActionBarCopy.displayName = "AssistantActionBarCopy";
 var AssistantActionBarReload = (0, import_react56.forwardRef)((props, ref) => {
@@ -3256,7 +3489,7 @@ var BranchPickerPrevious2 = (0, import_react57.forwardRef)((props, ref) => {
       branchPicker: { previous: { tooltip = "Previous" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(branchPicker_exports.Previous, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(import_lucide_react2.ChevronLeftIcon, {}) }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(branchPicker_exports.Previous, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(import_lucide_react2.ChevronLeftIcon, {}) }) });
 });
 BranchPickerPrevious2.displayName = "BranchPickerPrevious";
 var BranchPickerStateWrapper = withDefaults("span", {
@@ -3274,7 +3507,7 @@ var BranchPickerNext = (0, import_react57.forwardRef)((props, ref) => {
   const {
     strings: { branchPicker: { next: { tooltip = "Next" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(branchPicker_exports.Next, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(import_lucide_react2.ChevronRightIcon, {}) }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(branchPicker_exports.Next, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(import_lucide_react2.ChevronRightIcon, {}) }) });
 });
 BranchPickerNext.displayName = "BranchPickerNext";
 var exports3 = {
@@ -3449,7 +3682,7 @@ var ComposerSend = (0, import_react59.forwardRef)((props, ref) => {
   const {
     strings: { composer: { send: { tooltip = "Send" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(ComposerSendButton, { tooltip, ...props, ref, children: /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(import_lucide_react3.SendHorizonalIcon, {}) }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(ComposerSendButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(import_lucide_react3.SendHorizontalIcon, {}) }) });
 });
 ComposerSend.displayName = "ComposerSend";
 var ComposerCancelButton = withDefaults(TooltipIconButton, {
@@ -3460,7 +3693,7 @@ var ComposerCancel = (0, import_react59.forwardRef)((props, ref) => {
   const {
     strings: { composer: { cancel: { tooltip = "Cancel" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(ComposerCancelButton, { tooltip, ...props, ref, children: /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(CircleStopIcon, {}) }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(ComposerCancelButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(CircleStopIcon, {}) }) });
 });
 ComposerCancel.displayName = "ComposerCancel";
 var exports6 = {
@@ -3575,7 +3808,7 @@ var UserActionBarEdit = (0, import_react61.forwardRef)((props, ref) => {
   } = useThreadConfig();
   const allowEdit = useAllowEdit();
   if (!allowEdit) return null;
-  return /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(actionBar_exports.Edit, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(import_lucide_react4.PencilIcon, {}) }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(actionBar_exports.Edit, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(import_lucide_react4.PencilIcon, {}) }) });
 });
 UserActionBarEdit.displayName = "UserActionBarEdit";
 var exports8 = {
@@ -3653,7 +3886,7 @@ var EditComposerCancel = (0, import_react63.forwardRef)(
         editComposer: { cancel: { label = "Cancel" } = {} } = {}
       } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ (0, import_jsx_runtime40.jsx)(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime40.jsx)(Button, { variant: "ghost", ...props, ref, children: label }) });
+    return /* @__PURE__ */ (0, import_jsx_runtime40.jsx)(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime40.jsx)(Button, { variant: "ghost", ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerCancel.displayName = "EditComposerCancel";
@@ -3662,7 +3895,7 @@ var EditComposerSend = (0, import_react63.forwardRef)(
     const {
       strings: { editComposer: { send: { label = "Send" } = {} } = {} } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ (0, import_jsx_runtime40.jsx)(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime40.jsx)(Button, { ...props, ref, children: label }) });
+    return /* @__PURE__ */ (0, import_jsx_runtime40.jsx)(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime40.jsx)(Button, { ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerSend.displayName = "EditComposerSend";
@@ -3730,7 +3963,10 @@ var ThreadScrollToBottom = (0, import_react64.forwardRef)((props, ref) => {
       thread: { scrollToBottom: { tooltip = "Scroll to bottom" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ (0, import_jsx_runtime41.jsx)(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime41.jsx)(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ (0, import_jsx_runtime41.jsx)(import_lucide_react5.ArrowDownIcon, {}) }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime41.jsx)(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime41.jsxs)(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: [
+    "|",
+    props.children ?? /* @__PURE__ */ (0, import_jsx_runtime41.jsx)(import_lucide_react5.ArrowDownIcon, {})
+  ] }) });
 });
 ThreadScrollToBottom.displayName = "ThreadScrollToBottom";
 var exports11 = {
@@ -3781,7 +4017,7 @@ var AssistantModalButton = (0, import_react65.forwardRef)(({ "data-state": state
     } = {}
   } = useThreadConfig();
   const tooltip = state === "open" ? openTooltip : closedTooltip;
-  return /* @__PURE__ */ (0, import_jsx_runtime42.jsxs)(
+  return /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(
     ModalButtonStyled,
     {
       side: "left",
@@ -3789,17 +4025,22 @@ var AssistantModalButton = (0, import_react65.forwardRef)(({ "data-state": state
       "data-state": state,
       ...rest,
       ref,
-      children: [
-        /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_lucide_react6.BotIcon, { "data-state": state, className: "aui-modal-button-closed-icon" }),
+      children: rest.children ?? /* @__PURE__ */ (0, import_jsx_runtime42.jsxs)(import_jsx_runtime42.Fragment, { children: [
+        /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(
+          import_lucide_react6.BotIcon,
+          {
+            "data-state": state,
+            className: "aui-modal-button-closed-icon"
+          }
+        ),
         /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(
           import_lucide_react6.ChevronDownIcon,
           {
             "data-state": state,
             className: "aui-modal-button-open-icon"
           }
-        ),
-        /* @__PURE__ */ (0, import_jsx_runtime42.jsx)("span", { className: "aui-sr-only", children: tooltip })
-      ]
+        )
+      ] })
     }
   );
 });
@@ -3815,16 +4056,6 @@ var exports12 = {
   Content: AssistantModalContent
 };
 var assistant_modal_default = Object.assign(AssistantModal, exports12);
-
-// src/internal.ts
-var internal_exports = {};
-__export(internal_exports, {
-  BaseAssistantRuntime: () => BaseAssistantRuntime,
-  MessageRepository: () => MessageRepository,
-  ProxyConfigProvider: () => ProxyConfigProvider,
-  TooltipIconButton: () => TooltipIconButton,
-  useSmooth: () => useSmooth
-});
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ActionBarPrimitive,
@@ -3839,6 +4070,7 @@ __export(internal_exports, {
   ComposerPrimitive,
   ContentPart,
   ContentPartPrimitive,
+  EdgeChatAdapter,
   EditComposer,
   INTERNAL,
   MessagePrimitive,
@@ -3848,8 +4080,11 @@ __export(internal_exports, {
   ThreadWelcome,
   UserActionBar,
   UserMessage,
+  fromCoreMessages,
+  fromLanguageModelMessages,
   makeAssistantTool,
   makeAssistantToolUI,
+  toLanguageModelMessages,
   useActionBarCopy,
   useActionBarEdit,
   useActionBarReload,