@assistant-ui/react 0.4.3 → 0.4.5

package/dist/index.mjs

@@ -30,7 +30,7 @@ import { create } from "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) {
@@ -226,7 +226,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
     })
   );
 };
@@ -243,10 +244,7 @@ var ThreadProvider = ({
   children,
   runtime
 }) => {
-  const runtimeRef = useRef(runtime);
-  useInsertionEffect(() => {
-    runtimeRef.current = runtime;
-  });
+  const runtimeRef = useRef(runtime.thread);
   const [context] = useState(() => {
     const useThread = makeThreadStore(runtimeRef);
     const useThreadMessages = makeThreadMessagesStore(runtimeRef);
@@ -261,8 +259,16 @@ var ThreadProvider = ({
       useViewport
     };
   });
+  const thread = useSyncExternalStore(
+    useCallback((c) => runtime.subscribe(c), [runtime]),
+    () => runtime.thread,
+    () => runtime.thread
+  );
+  useInsertionEffect(() => {
+    runtimeRef.current = thread;
+  });
   useEffect(() => {
-    const onRuntimeUpdate = () => {
+    const onThreadUpdate = () => {
       context.useThread.setState(
         Object.freeze({
           isRunning: runtimeRef.current.isRunning
@@ -271,20 +277,11 @@ var ThreadProvider = ({
       );
       context.useThreadMessages.setState(Object.freeze(runtimeRef.current.messages), true);
     };
-    onRuntimeUpdate();
-    return runtime.subscribe(onRuntimeUpdate);
-  }, [context, runtime]);
-  const subscribe = useCallback(
-    (c) => runtime.subscribe(c),
-    [runtime]
-  );
-  const RuntimeSynchronizer = useSyncExternalStore(
-    subscribe,
-    () => runtime.unstable_synchronizer,
-    () => void 0
-  );
+    onThreadUpdate();
+    return thread.subscribe(onThreadUpdate);
+  }, [context, thread]);
   return /* @__PURE__ */ jsxs(ThreadContext.Provider, { value: context, children: [
-    RuntimeSynchronizer && /* @__PURE__ */ jsx(RuntimeSynchronizer, {}),
+    thread.unstable_synchronizer && /* @__PURE__ */ jsx(thread.unstable_synchronizer, {}),
     children
   ] });
 };
@@ -293,7 +290,8 @@ var ThreadProvider = ({
 import { create as create8 } from "zustand";
 var makeAssistantActionsStore = (runtimeRef) => create8(
   () => Object.freeze({
-    switchToThread: () => runtimeRef.current.switchToThread(null)
+    switchToThread: () => runtimeRef.current.switchToThread(null),
+    getRuntime: () => runtimeRef.current
   })
 );
 
@@ -310,10 +308,10 @@ var AssistantProvider = ({ children, runtime }) => {
     const useAssistantActions = makeAssistantActionsStore(runtimeRef);
     return { useModelConfig, useToolUIs, useAssistantActions };
   });
-  const getModelCOnfig = context.useModelConfig((c) => c.getModelConfig);
+  const getModelConfig = context.useModelConfig();
   useEffect2(() => {
-    return runtime.registerModelConfigProvider(getModelCOnfig);
-  }, [runtime, getModelCOnfig]);
+    return runtime.registerModelConfigProvider(getModelConfig);
+  }, [runtime, getModelConfig]);
   return /* @__PURE__ */ jsx2(AssistantContext.Provider, { value: context, children: /* @__PURE__ */ jsx2(ThreadProvider, { runtime, children }) });
 };
 
@@ -423,7 +421,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();
@@ -473,7 +473,7 @@ var useAssistantInstructions = (instruction) => {
     const config = {
       system: instruction
     };
-    return registerModelConfigProvider(() => config);
+    return registerModelConfigProvider({ getModelConfig: () => config });
   }, [registerModelConfigProvider, instruction]);
 };
 
@@ -1902,7 +1902,7 @@ var ThreadPrimitiveSuggestion = createActionButton(
 );
 
 // src/runtimes/local/useLocalRuntime.tsx
-import { useInsertionEffect as useInsertionEffect3, useState as useState7 } from "react";
+import { useInsertionEffect as useInsertionEffect3, useState as useState8 } from "react";
 
 // src/utils/idUtils.tsx
 import { customAlphabet } from "nanoid/non-secure";
@@ -2080,45 +2080,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);
@@ -2127,179 +2096,158 @@ var BaseAssistantRuntime = class {
   subscriptionHandler = () => {
     for (const callback of this._subscriptions) callback();
   };
-  get unstable_synchronizer() {
-    return this.thread.unstable_synchronizer;
-  }
 };
 
-// src/runtimes/local/LocalRuntime.tsx
-var LocalRuntime = class extends BaseAssistantRuntime {
-  _configProviders;
-  constructor(adapter) {
-    const configProviders = /* @__PURE__ */ new Set();
-    super(new LocalThreadRuntime(configProviders, adapter));
-    this._configProviders = configProviders;
-  }
-  set adapter(adapter) {
-    this.thread.adapter = adapter;
-  }
-  registerModelConfigProvider(provider) {
-    this._configProviders.add(provider);
-    return () => this._configProviders.delete(provider);
-  }
-  switchToThread(threadId) {
-    if (threadId) {
-      throw new Error("LocalRuntime does not yet support switching threads");
+// 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
+import { forwardRef as forwardRef17 } from "react";
+
+// src/ui/base/tooltip.tsx
+import * as TooltipPrimitive from "@radix-ui/react-tooltip";
+
+// src/ui/utils/withDefaults.tsx
+import {
+  forwardRef as forwardRef15
+} from "react";
+import classNames from "classnames";
+import { jsx as jsx25 } from "react/jsx-runtime";
+var withDefaultProps = ({
+  className,
+  ...defaultProps
+}) => ({ className: classNameProp, ...props }) => {
+  return {
+    className: classNames(className, classNameProp),
+    ...defaultProps,
+    ...props
+  };
+};
+var withDefaults = (Component, defaultProps) => {
+  const getProps = withDefaultProps(defaultProps);
+  const WithDefaults = forwardRef15(
+    (props, ref) => {
+      const ComponentAsAny = Component;
+      return /* @__PURE__ */ jsx25(ComponentAsAny, { ...getProps(props), ref });
     }
-    return this.thread = new LocalThreadRuntime(
-      this._configProviders,
-      this.thread.adapter
-    );
-  }
+  );
+  WithDefaults.displayName = "withDefaults(" + (typeof Component === "string" ? Component : Component.displayName) + ")";
+  return WithDefaults;
 };
-var CAPABILITIES = Object.freeze({
-  edit: true,
-  reload: true,
-  cancel: true,
-  copy: true
+
+// src/ui/base/tooltip.tsx
+import { jsx as jsx26 } from "react/jsx-runtime";
+var Tooltip = (props) => {
+  return /* @__PURE__ */ jsx26(TooltipPrimitive.Provider, { children: /* @__PURE__ */ jsx26(TooltipPrimitive.Root, { ...props }) });
+};
+Tooltip.displayName = "Tooltip";
+var TooltipTrigger = TooltipPrimitive.Trigger;
+var TooltipContent = withDefaults(TooltipPrimitive.Content, {
+  sideOffset: 4,
+  className: "aui-tooltip-content"
 });
-var LocalThreadRuntime = class {
-  constructor(_configProviders, adapter) {
-    this._configProviders = _configProviders;
-    this.adapter = adapter;
-  }
-  _subscriptions = /* @__PURE__ */ new Set();
-  abortController = null;
-  repository = new MessageRepository();
-  capabilities = CAPABILITIES;
-  get messages() {
-    return this.repository.getMessages();
-  }
-  get isRunning() {
-    return this.abortController != null;
-  }
-  getBranches(messageId) {
-    return this.repository.getBranches(messageId);
-  }
-  switchToBranch(branchId) {
-    this.repository.switchToBranch(branchId);
-    this.notifySubscribers();
-  }
-  async append(message) {
-    const userMessageId = generateId();
-    const userMessage = {
-      id: userMessageId,
-      role: "user",
-      content: message.content,
-      createdAt: /* @__PURE__ */ new Date()
-    };
-    this.repository.addOrUpdateMessage(message.parentId, userMessage);
-    await this.startRun(userMessageId);
-  }
-  async startRun(parentId) {
-    const id = generateId();
-    this.repository.resetHead(parentId);
-    const messages = this.repository.getMessages();
-    const message = {
-      id,
-      role: "assistant",
-      status: { type: "in_progress" },
-      content: [{ type: "text", text: "" }],
-      createdAt: /* @__PURE__ */ new Date()
-    };
-    this.repository.addOrUpdateMessage(parentId, { ...message });
-    this.abortController?.abort();
-    this.abortController = new AbortController();
-    this.notifySubscribers();
-    try {
-      const updateHandler = ({ content }) => {
-        message.content = content;
-        this.repository.addOrUpdateMessage(parentId, { ...message });
-        this.notifySubscribers();
-      };
-      const result = await this.adapter.run({
-        messages,
-        abortSignal: this.abortController.signal,
-        config: mergeModelConfigs(this._configProviders),
-        onUpdate: updateHandler
-      });
-      if (result !== void 0) {
-        updateHandler(result);
-      }
-      if (result.status?.type === "in_progress")
-        throw new Error(
-          "Unexpected in_progress status returned from ChatModelAdapter"
-        );
-      message.status = result.status ?? { type: "done" };
-      this.repository.addOrUpdateMessage(parentId, { ...message });
-    } catch (e) {
-      message.status = { type: "error", error: e };
-      this.repository.addOrUpdateMessage(parentId, { ...message });
-      console.error(e);
-    } finally {
-      this.abortController = null;
-      this.notifySubscribers();
+TooltipContent.displayName = "TooltipContent";
+
+// src/ui/base/button.tsx
+import { cva } from "class-variance-authority";
+import { Primitive as Primitive11 } from "@radix-ui/react-primitive";
+import { forwardRef as forwardRef16 } from "react";
+import { jsx as jsx27 } from "react/jsx-runtime";
+var buttonVariants = 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"
   }
-  cancelRun() {
-    if (!this.abortController) return;
-    this.abortController.abort();
-    this.abortController = null;
-    this.notifySubscribers();
-  }
-  notifySubscribers() {
-    for (const callback of this._subscriptions) callback();
-  }
-  subscribe(callback) {
-    this._subscriptions.add(callback);
-    return () => this._subscriptions.delete(callback);
-  }
-  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");
-    let found = false;
-    const newContent = message.content.map((c) => {
-      if (c.type !== "tool-call") return c;
-      if (c.toolCallId !== toolCallId) return c;
-      found = true;
-      return {
-        ...c,
-        result
-      };
-    });
-    if (!found)
-      throw new Error("Tried to add tool result to non-existing tool call");
-    this.repository.addOrUpdateMessage(parentId, {
-      ...message,
-      content: newContent
-    });
+});
+var Button = forwardRef16(
+  ({ className, variant, size, ...props }, ref) => {
+    return /* @__PURE__ */ jsx27(
+      Primitive11.button,
+      {
+        className: buttonVariants({ variant, size, className }),
+        ...props,
+        ref
+      }
+    );
   }
-};
+);
+Button.displayName = "Button";
 
-// src/runtimes/local/useLocalRuntime.tsx
-var useLocalRuntime = (adapter) => {
-  const [runtime] = useState7(() => new LocalRuntime(adapter));
-  useInsertionEffect3(() => {
-    runtime.adapter = adapter;
+// src/ui/base/tooltip-icon-button.tsx
+import { jsx as jsx28, jsxs as jsxs4 } from "react/jsx-runtime";
+var TooltipIconButton = forwardRef17(({ children, tooltip, side = "bottom", ...rest }, ref) => {
+  return /* @__PURE__ */ jsxs4(Tooltip, { children: [
+    /* @__PURE__ */ jsx28(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ jsxs4(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
+      children,
+      /* @__PURE__ */ jsx28("span", { className: "aui-sr-only", children: tooltip })
+    ] }) }),
+    /* @__PURE__ */ jsx28(TooltipContent, { side, children: tooltip })
+  ] });
+});
+TooltipIconButton.displayName = "TooltipIconButton";
+
+// src/runtimes/edge/useEdgeRuntime.ts
+import { useState as useState7 } from "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;
+      })
+    };
   });
-  return runtime;
+};
+
+// src/runtimes/edge/converters/toLanguageModelTools.ts
+import { z } from "zod";
+import zodToJsonSchema from "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 z.ZodType ? zodToJsonSchema(tool.parameters) : tool.parameters
+  }));
 };
 
 // src/runtimes/edge/streams/assistantDecoderStream.ts
 function assistantDecoderStream() {
+  const toolCallNames = /* @__PURE__ */ new Map();
   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;
       }
@@ -2312,12 +2260,13 @@ function assistantDecoderStream() {
           break;
         }
         case "1" /* ToolCallBegin */: {
-          const { id, name } = JSON.parse(value);
+          const { id, name } = value;
+          toolCallNames.set(id, name);
           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",
@@ -2328,17 +2277,27 @@ function assistantDecoderStream() {
           });
           break;
         }
-        case "F" /* Finish */: {
+        case "3" /* ToolCallResult */: {
           controller.enqueue({
-            type: "finish",
-            ...JSON.parse(value)
+            type: "tool-result",
+            toolCallType: "function",
+            toolCallId: value.id,
+            toolName: toolCallNames.get(value.id),
+            result: value.result
+          });
+          break;
+        }
+        case "F" /* Finish */: {
+          controller.enqueue({
+            type: "finish",
+            ...value
           });
           break;
         }
         case "E" /* Error */: {
           controller.enqueue({
             type: "error",
-            error: JSON.parse(value)
+            error: value
           });
           break;
         }
@@ -2355,7 +2314,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))
   ];
 };
 
@@ -2714,9 +2673,9 @@ var parsePartialJson = (json) => {
 };
 
 // src/runtimes/edge/streams/runResultStream.ts
-function runResultStream() {
+function runResultStream(initialContent) {
   let message = {
-    content: []
+    content: initialContent
   };
   const currentToolCall = { toolCallId: "", argsText: "" };
   return new TransformStream({
@@ -2740,7 +2699,7 @@ function runResultStream() {
             message,
             toolCallId,
             toolName,
-            parsePartialJson(currentToolCall.argsText)
+            currentToolCall.argsText
           );
           controller.enqueue(message);
           break;
@@ -2788,7 +2747,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) {
@@ -2796,13 +2755,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 {
@@ -2845,11 +2806,8 @@ var appendOrUpdateFinish = (message, chunk) => {
   };
 };
 
-// src/runtimes/edge/useEdgeRuntime.ts
-import { useMemo as useMemo3 } from "react";
-
 // src/runtimes/edge/streams/toolResultStream.ts
-import { z } from "zod";
+import { z as z2 } from "zod";
 import sjson2 from "secure-json-parse";
 function toolResultStream(tools) {
   const toolCallExecutions = /* @__PURE__ */ new Map();
@@ -2861,9 +2819,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 = sjson2.parse(argsText);
-          if (tool.parameters instanceof z.ZodType) {
+          if (tool.parameters instanceof z2.ZodType) {
             const result = tool.parameters.safeParse(args);
             if (!result.success) {
               controller.enqueue({
@@ -2900,6 +2858,7 @@ function toolResultStream(tools) {
         }
         case "text-delta":
         case "tool-call-delta":
+        case "tool-result":
         case "finish":
         case "error":
           break;
@@ -2915,9 +2874,7 @@ function toolResultStream(tools) {
   });
 }
 
-// src/runtimes/edge/useEdgeRuntime.ts
-import { z as z2 } from "zod";
-import { zodToJsonSchema } from "zod-to-json-schema";
+// src/runtimes/edge/EdgeChatAdapter.ts
 function asAsyncIterable(source) {
   return {
     [Symbol.asyncIterator]: () => {
@@ -2931,51 +2888,479 @@ 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 z2.ZodType ? 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 roundtrip(initialContent, { 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
         )
       }),
       signal: abortSignal
     });
-    const stream = result.body.pipeThrough(new TextDecoderStream()).pipeThrough(chunkByLineStream()).pipeThrough(assistantDecoderStream()).pipeThrough(toolResultStream(config.tools)).pipeThrough(runResultStream());
+    const stream = result.body.pipeThrough(new TextDecoderStream()).pipeThrough(chunkByLineStream()).pipeThrough(assistantDecoderStream()).pipeThrough(toolResultStream(config.tools)).pipeThrough(runResultStream(initialContent));
+    let message;
     let update;
     for await (update of asAsyncIterable(stream)) {
-      onUpdate(update);
+      message = onUpdate(update);
     }
     if (update === void 0)
       throw new Error("No data received from Edge Runtime");
-    return update;
+    return [message, update];
+  }
+  async run({ messages, abortSignal, config, onUpdate }) {
+    let roundtripAllowance = this.options.maxToolRoundtrips ?? 1;
+    let usage = {
+      promptTokens: 0,
+      completionTokens: 0
+    };
+    let result;
+    let assistantMessage;
+    do {
+      [assistantMessage, result] = await this.roundtrip(result?.content ?? [], {
+        messages: assistantMessage ? [...messages, assistantMessage] : messages,
+        abortSignal,
+        config,
+        onUpdate
+      });
+      if (result.status?.type === "done") {
+        usage.promptTokens += result.status.usage?.promptTokens ?? 0;
+        usage.completionTokens += result.status.usage?.completionTokens ?? 0;
+      }
+    } while (result.status?.type === "done" && result.status.finishReason === "tool-calls" && result.content.every((c) => c.type !== "tool-call" || !!c.result) && roundtripAllowance-- > 0);
+    if (result.status?.type === "done" && usage.promptTokens > 0) {
+      result = {
+        ...result,
+        status: {
+          ...result.status,
+          usage
+        }
+      };
+    }
+    return result;
+  }
+};
+
+// src/runtimes/edge/useEdgeRuntime.ts
+var useEdgeRuntime = ({
+  initialMessages,
+  ...options
+}) => {
+  const [adapter] = useState7(() => new EdgeChatAdapter(options));
+  return useLocalRuntime(adapter, { initialMessages });
+};
+
+// 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 {
+      id: generateId(),
+      createdAt: /* @__PURE__ */ new Date(),
+      ...message2.role === "assistant" ? {
+        status: { type: "done" }
+      } : void 0,
+      ...message2
+    };
+  });
+};
+
+// src/runtimes/local/LocalRuntime.tsx
+var LocalRuntime = class extends BaseAssistantRuntime {
+  _proxyConfigProvider;
+  constructor(adapter, options) {
+    const proxyConfigProvider = new ProxyConfigProvider();
+    super(new LocalThreadRuntime(proxyConfigProvider, adapter, options));
+    this._proxyConfigProvider = proxyConfigProvider;
+  }
+  set adapter(adapter) {
+    this.thread.adapter = adapter;
+  }
+  registerModelConfigProvider(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._proxyConfigProvider,
+      this.thread.adapter
+    );
   }
+};
+var CAPABILITIES = Object.freeze({
+  edit: true,
+  reload: true,
+  cancel: true,
+  copy: true
 });
-var useEdgeRuntime = (options) => {
-  const adapter = useMemo3(() => createEdgeChatAdapter(options), [options]);
-  return useLocalRuntime(adapter);
+var LocalThreadRuntime = class {
+  constructor(configProvider, adapter, options) {
+    this.configProvider = configProvider;
+    this.adapter = adapter;
+    if (options?.initialMessages) {
+      let parentId = null;
+      const messages = fromCoreMessages(options.initialMessages);
+      for (const message of messages) {
+        this.repository.addOrUpdateMessage(parentId, message);
+        parentId = message.id;
+      }
+    }
+  }
+  _subscriptions = /* @__PURE__ */ new Set();
+  abortController = null;
+  repository = new MessageRepository();
+  capabilities = CAPABILITIES;
+  get messages() {
+    return this.repository.getMessages();
+  }
+  get isRunning() {
+    return this.abortController != null;
+  }
+  getBranches(messageId) {
+    return this.repository.getBranches(messageId);
+  }
+  switchToBranch(branchId) {
+    this.repository.switchToBranch(branchId);
+    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,
+      role: "user",
+      content: message.content,
+      createdAt: /* @__PURE__ */ new Date()
+    };
+    this.repository.addOrUpdateMessage(message.parentId, userMessage);
+    await this.startRun(userMessageId);
+  }
+  async startRun(parentId) {
+    this.repository.resetHead(parentId);
+    const messages = this.repository.getMessages();
+    const message = {
+      id: generateId(),
+      role: "assistant",
+      status: { type: "in_progress" },
+      content: [{ type: "text", text: "" }],
+      createdAt: /* @__PURE__ */ new Date()
+    };
+    this.repository.addOrUpdateMessage(parentId, { ...message });
+    this.abortController?.abort();
+    this.abortController = new AbortController();
+    this.notifySubscribers();
+    try {
+      const updateHandler = ({ content }) => {
+        message.content = content;
+        const newMessage = { ...message };
+        this.repository.addOrUpdateMessage(parentId, newMessage);
+        this.notifySubscribers();
+        return newMessage;
+      };
+      const result = await this.adapter.run({
+        messages,
+        abortSignal: this.abortController.signal,
+        config: this.configProvider.getModelConfig(),
+        onUpdate: updateHandler
+      });
+      if (result !== void 0) {
+        updateHandler(result);
+      }
+      if (result.status?.type === "in_progress")
+        throw new Error(
+          "Unexpected in_progress status returned from ChatModelAdapter"
+        );
+      message.status = result.status ?? { type: "done" };
+      this.repository.addOrUpdateMessage(parentId, { ...message });
+    } catch (e) {
+      message.status = { type: "error", error: e };
+      this.repository.addOrUpdateMessage(parentId, { ...message });
+      throw e;
+    } finally {
+      this.abortController = null;
+      this.notifySubscribers();
+    }
+  }
+  cancelRun() {
+    if (!this.abortController) return;
+    this.abortController.abort();
+    this.abortController = null;
+  }
+  notifySubscribers() {
+    for (const callback of this._subscriptions) callback();
+  }
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    return () => this._subscriptions.delete(callback);
+  }
+  addToolResult({ messageId, toolCallId, result }) {
+    const { parentId, message } = this.repository.getMessage(messageId);
+    if (message.role !== "assistant")
+      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;
+      if (c.toolCallId !== toolCallId) return c;
+      found = true;
+      return {
+        ...c,
+        result
+      };
+    });
+    if (!found)
+      throw new Error("Tried to add tool result to non-existing tool call");
+    this.repository.addOrUpdateMessage(parentId, {
+      ...message,
+      content: newContent
+    });
+  }
+};
+
+// src/runtimes/local/useLocalRuntime.tsx
+var useLocalRuntime = (adapter, options) => {
+  const [runtime] = useState8(() => new LocalRuntime(adapter, options));
+  useInsertionEffect3(() => {
+    runtime.adapter = adapter;
+  });
+  return runtime;
 };
 
 // src/ui/thread-config.tsx
 import { createContext as createContext5, useContext as useContext5 } from "react";
-import { Fragment as Fragment3, jsx as jsx25 } from "react/jsx-runtime";
+import { Fragment as Fragment3, jsx as jsx29 } from "react/jsx-runtime";
 var ThreadConfigContext = createContext5({});
 var useThreadConfig = () => {
   return useContext5(ThreadConfigContext);
@@ -2985,119 +3370,21 @@ var ThreadConfigProvider = ({
   config
 }) => {
   const assistant = useAssistantContext({ optional: true });
-  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ jsx25(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ jsx25(Fragment3, { children });
+  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ jsx29(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ jsx29(Fragment3, { 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__ */ jsx25(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
+  return /* @__PURE__ */ jsx29(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
 };
 ThreadConfigProvider.displayName = "ThreadConfigProvider";
 
 // src/ui/assistant-action-bar.tsx
 import { forwardRef as forwardRef18 } from "react";
 import { CheckIcon, CopyIcon, RefreshCwIcon } from "lucide-react";
-
-// src/ui/base/tooltip-icon-button.tsx
-import { forwardRef as forwardRef17 } from "react";
-
-// src/ui/base/tooltip.tsx
-import * as TooltipPrimitive from "@radix-ui/react-tooltip";
-
-// src/ui/utils/withDefaults.tsx
-import {
-  forwardRef as forwardRef15
-} from "react";
-import classNames from "classnames";
-import { jsx as jsx26 } from "react/jsx-runtime";
-var withDefaultProps = ({
-  className,
-  ...defaultProps
-}) => ({ className: classNameProp, ...props }) => {
-  return {
-    className: classNames(className, classNameProp),
-    ...defaultProps,
-    ...props
-  };
-};
-var withDefaults = (Component, defaultProps) => {
-  const getProps = withDefaultProps(defaultProps);
-  const WithDefaults = forwardRef15(
-    (props, ref) => {
-      const ComponentAsAny = Component;
-      return /* @__PURE__ */ jsx26(ComponentAsAny, { ...getProps(props), ref });
-    }
-  );
-  WithDefaults.displayName = "withDefaults(" + (typeof Component === "string" ? Component : Component.displayName) + ")";
-  return WithDefaults;
-};
-
-// src/ui/base/tooltip.tsx
-import { jsx as jsx27 } from "react/jsx-runtime";
-var Tooltip = (props) => {
-  return /* @__PURE__ */ jsx27(TooltipPrimitive.Provider, { children: /* @__PURE__ */ jsx27(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
-import { cva } from "class-variance-authority";
-import { Primitive as Primitive11 } from "@radix-ui/react-primitive";
-import { forwardRef as forwardRef16 } from "react";
-import { jsx as jsx28 } from "react/jsx-runtime";
-var buttonVariants = 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 = forwardRef16(
-  ({ className, variant, size, ...props }, ref) => {
-    return /* @__PURE__ */ jsx28(
-      Primitive11.button,
-      {
-        className: buttonVariants({ variant, size, className }),
-        ...props,
-        ref
-      }
-    );
-  }
-);
-Button.displayName = "Button";
-
-// src/ui/base/tooltip-icon-button.tsx
-import { jsx as jsx29, jsxs as jsxs4 } from "react/jsx-runtime";
-var TooltipIconButton = forwardRef17(({ children, tooltip, side = "bottom", ...rest }, ref) => {
-  return /* @__PURE__ */ jsxs4(Tooltip, { children: [
-    /* @__PURE__ */ jsx29(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ jsxs4(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
-      children,
-      /* @__PURE__ */ jsx29("span", { className: "aui-sr-only", children: tooltip })
-    ] }) }),
-    /* @__PURE__ */ jsx29(TooltipContent, { side, children: tooltip })
-  ] });
-});
-TooltipIconButton.displayName = "TooltipIconButton";
-
-// src/ui/assistant-action-bar.tsx
-import { jsx as jsx30, jsxs as jsxs5 } from "react/jsx-runtime";
+import { Fragment as Fragment4, jsx as jsx30, jsxs as jsxs5 } from "react/jsx-runtime";
 var useAllowCopy = () => {
   const { assistantMessage: { allowCopy = true } = {} } = useThreadConfig();
   const { useThreadActions } = useThreadContext();
@@ -3140,10 +3427,10 @@ var AssistantActionBarCopy = forwardRef18((props, ref) => {
   } = useThreadConfig();
   const allowCopy = useAllowCopy();
   if (!allowCopy) return null;
-  return /* @__PURE__ */ jsx30(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ jsxs5(TooltipIconButton, { tooltip, ...props, ref, children: [
+  return /* @__PURE__ */ jsx30(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ jsx30(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsxs5(Fragment4, { children: [
     /* @__PURE__ */ jsx30(message_exports.If, { copied: true, children: /* @__PURE__ */ jsx30(CheckIcon, {}) }),
     /* @__PURE__ */ jsx30(message_exports.If, { copied: false, children: /* @__PURE__ */ jsx30(CopyIcon, {}) })
-  ] }) });
+  ] }) }) });
 });
 AssistantActionBarCopy.displayName = "AssistantActionBarCopy";
 var AssistantActionBarReload = forwardRef18((props, ref) => {
@@ -3192,7 +3479,7 @@ var BranchPickerPrevious2 = forwardRef19((props, ref) => {
       branchPicker: { previous: { tooltip = "Previous" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx31(branchPicker_exports.Previous, { asChild: true, children: /* @__PURE__ */ jsx31(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx31(ChevronLeftIcon, {}) }) });
+  return /* @__PURE__ */ jsx31(branchPicker_exports.Previous, { asChild: true, children: /* @__PURE__ */ jsx31(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx31(ChevronLeftIcon, {}) }) });
 });
 BranchPickerPrevious2.displayName = "BranchPickerPrevious";
 var BranchPickerStateWrapper = withDefaults("span", {
@@ -3210,7 +3497,7 @@ var BranchPickerNext = forwardRef19((props, ref) => {
   const {
     strings: { branchPicker: { next: { tooltip = "Next" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx31(branchPicker_exports.Next, { asChild: true, children: /* @__PURE__ */ jsx31(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx31(ChevronRightIcon, {}) }) });
+  return /* @__PURE__ */ jsx31(branchPicker_exports.Next, { asChild: true, children: /* @__PURE__ */ jsx31(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx31(ChevronRightIcon, {}) }) });
 });
 BranchPickerNext.displayName = "BranchPickerNext";
 var exports2 = {
@@ -3317,7 +3604,7 @@ import { ArrowDownIcon } from "lucide-react";
 
 // src/ui/composer.tsx
 import { forwardRef as forwardRef21 } from "react";
-import { SendHorizonalIcon } from "lucide-react";
+import { SendHorizontalIcon } from "lucide-react";
 
 // src/ui/base/CircleStopIcon.tsx
 import { jsx as jsx35 } from "react/jsx-runtime";
@@ -3335,7 +3622,7 @@ var CircleStopIcon = () => {
 CircleStopIcon.displayName = "CircleStopIcon";
 
 // src/ui/composer.tsx
-import { Fragment as Fragment4, jsx as jsx36, jsxs as jsxs9 } from "react/jsx-runtime";
+import { Fragment as Fragment5, jsx as jsx36, jsxs as jsxs9 } from "react/jsx-runtime";
 var Composer = () => {
   return /* @__PURE__ */ jsxs9(ComposerRoot, { children: [
     /* @__PURE__ */ jsx36(ComposerInput, { autoFocus: true }),
@@ -3371,7 +3658,7 @@ var useAllowCancel = () => {
 var ComposerAction = () => {
   const allowCancel = useAllowCancel();
   if (!allowCancel) return /* @__PURE__ */ jsx36(ComposerSend, {});
-  return /* @__PURE__ */ jsxs9(Fragment4, { children: [
+  return /* @__PURE__ */ jsxs9(Fragment5, { children: [
     /* @__PURE__ */ jsx36(thread_exports.If, { running: false, children: /* @__PURE__ */ jsx36(ComposerSend, {}) }),
     /* @__PURE__ */ jsx36(thread_exports.If, { running: true, children: /* @__PURE__ */ jsx36(ComposerCancel, {}) })
   ] });
@@ -3385,7 +3672,7 @@ var ComposerSend = forwardRef21((props, ref) => {
   const {
     strings: { composer: { send: { tooltip = "Send" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx36(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx36(ComposerSendButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx36(SendHorizonalIcon, {}) }) });
+  return /* @__PURE__ */ jsx36(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx36(ComposerSendButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx36(SendHorizontalIcon, {}) }) });
 });
 ComposerSend.displayName = "ComposerSend";
 var ComposerCancelButton = withDefaults(TooltipIconButton, {
@@ -3396,7 +3683,7 @@ var ComposerCancel = forwardRef21((props, ref) => {
   const {
     strings: { composer: { cancel: { tooltip = "Cancel" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx36(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx36(ComposerCancelButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx36(CircleStopIcon, {}) }) });
+  return /* @__PURE__ */ jsx36(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx36(ComposerCancelButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx36(CircleStopIcon, {}) }) });
 });
 ComposerCancel.displayName = "ComposerCancel";
 var exports5 = {
@@ -3511,7 +3798,7 @@ var UserActionBarEdit = forwardRef23((props, ref) => {
   } = useThreadConfig();
   const allowEdit = useAllowEdit();
   if (!allowEdit) return null;
-  return /* @__PURE__ */ jsx38(actionBar_exports.Edit, { asChild: true, children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx38(PencilIcon, {}) }) });
+  return /* @__PURE__ */ jsx38(actionBar_exports.Edit, { asChild: true, children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx38(PencilIcon, {}) }) });
 });
 UserActionBarEdit.displayName = "UserActionBarEdit";
 var exports7 = {
@@ -3589,7 +3876,7 @@ var EditComposerCancel = forwardRef25(
         editComposer: { cancel: { label = "Cancel" } = {} } = {}
       } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx40(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx40(Button, { variant: "ghost", ...props, ref, children: label }) });
+    return /* @__PURE__ */ jsx40(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx40(Button, { variant: "ghost", ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerCancel.displayName = "EditComposerCancel";
@@ -3598,7 +3885,7 @@ var EditComposerSend = forwardRef25(
     const {
       strings: { editComposer: { send: { label = "Send" } = {} } = {} } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx40(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx40(Button, { ...props, ref, children: label }) });
+    return /* @__PURE__ */ jsx40(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx40(Button, { ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerSend.displayName = "EditComposerSend";
@@ -3666,7 +3953,7 @@ var ThreadScrollToBottom = forwardRef26((props, ref) => {
       thread: { scrollToBottom: { tooltip = "Scroll to bottom" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx41(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx41(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx41(ArrowDownIcon, {}) }) });
+  return /* @__PURE__ */ jsx41(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx41(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx41(ArrowDownIcon, {}) }) });
 });
 ThreadScrollToBottom.displayName = "ThreadScrollToBottom";
 var exports10 = {
@@ -3679,7 +3966,7 @@ var exports10 = {
 var thread_default = Object.assign(Thread, exports10);
 
 // src/ui/assistant-modal.tsx
-import { jsx as jsx42, jsxs as jsxs14 } from "react/jsx-runtime";
+import { Fragment as Fragment6, jsx as jsx42, jsxs as jsxs14 } from "react/jsx-runtime";
 var AssistantModal = (config) => {
   return /* @__PURE__ */ jsxs14(AssistantModalRoot, { config, children: [
     /* @__PURE__ */ jsx42(AssistantModalTrigger, {}),
@@ -3717,7 +4004,7 @@ var AssistantModalButton = forwardRef27(({ "data-state": state, ...rest }, ref)
     } = {}
   } = useThreadConfig();
   const tooltip = state === "open" ? openTooltip : closedTooltip;
-  return /* @__PURE__ */ jsxs14(
+  return /* @__PURE__ */ jsx42(
     ModalButtonStyled,
     {
       side: "left",
@@ -3725,17 +4012,22 @@ var AssistantModalButton = forwardRef27(({ "data-state": state, ...rest }, ref)
       "data-state": state,
       ...rest,
       ref,
-      children: [
-        /* @__PURE__ */ jsx42(BotIcon, { "data-state": state, className: "aui-modal-button-closed-icon" }),
+      children: rest.children ?? /* @__PURE__ */ jsxs14(Fragment6, { children: [
+        /* @__PURE__ */ jsx42(
+          BotIcon,
+          {
+            "data-state": state,
+            className: "aui-modal-button-closed-icon"
+          }
+        ),
         /* @__PURE__ */ jsx42(
           ChevronDownIcon,
           {
             "data-state": state,
             className: "aui-modal-button-open-icon"
           }
-        ),
-        /* @__PURE__ */ jsx42("span", { className: "aui-sr-only", children: tooltip })
-      ]
+        )
+      ] })
     }
   );
 });
@@ -3751,16 +4043,6 @@ var exports11 = {
   Content: AssistantModalContent
 };
 var assistant_modal_default = Object.assign(AssistantModal, exports11);
-
-// src/internal.ts
-var internal_exports = {};
-__export(internal_exports, {
-  BaseAssistantRuntime: () => BaseAssistantRuntime,
-  MessageRepository: () => MessageRepository,
-  ProxyConfigProvider: () => ProxyConfigProvider,
-  TooltipIconButton: () => TooltipIconButton,
-  useSmooth: () => useSmooth
-});
 export {
   actionBar_exports as ActionBarPrimitive,
   assistant_action_bar_default as AssistantActionBar,
@@ -3774,6 +4056,7 @@ export {
   composer_exports as ComposerPrimitive,
   content_part_default as ContentPart,
   contentPart_exports as ContentPartPrimitive,
+  EdgeChatAdapter,
   edit_composer_default as EditComposer,
   internal_exports as INTERNAL,
   message_exports as MessagePrimitive,
@@ -3783,8 +4066,11 @@ export {
   thread_welcome_default as ThreadWelcome,
   user_action_bar_default as UserActionBar,
   user_message_default as UserMessage,
+  fromCoreMessages,
+  fromLanguageModelMessages,
   makeAssistantTool,
   makeAssistantToolUI,
+  toLanguageModelMessages,
   useActionBarCopy,
   useActionBarEdit,
   useActionBarReload,