@assistant-ui/react 0.0.29 → 0.1.1

package/dist/index.mjs

@@ -1,24 +1,26 @@
-var __defProp = Object.defineProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
+import {
+  MessageRepository,
+  ProxyConfigProvider,
+  generateId,
+  mergeModelConfigs
+} from "./chunk-NSBOH42A.mjs";
+import {
+  AssistantContext,
+  ContentPartContext,
+  MessageContext,
+  ThreadContext,
+  useComposerContext,
+  useContentPartContext,
+  useMessageContext,
+  useThreadContext
+} from "./chunk-CY4TTHR7.mjs";
+import {
+  __export
+} from "./chunk-J5LGTIGS.mjs";
 
 // src/actions/useCopyMessage.tsx
 import { useCallback } from "react";
 
-// src/context/MessageContext.ts
-import { createContext, useContext } from "react";
-var MessageContext = createContext(null);
-var useMessageContext = () => {
-  const context = useContext(MessageContext);
-  if (!context)
-    throw new Error(
-      "This component can only be used inside a component passed to <ThreadPrimitive.Messages components={...} />."
-    );
-  return context;
-};
-
 // src/utils/combined/useCombinedStore.ts
 import { useMemo } from "react";
 
@@ -76,18 +78,6 @@ var useCopyMessage = ({ copiedDuration = 3e3 }) => {
 
 // src/actions/useReloadMessage.tsx
 import { useCallback as useCallback2 } from "react";
-
-// src/context/ThreadContext.ts
-import { createContext as createContext2, useContext as useContext2 } from "react";
-var ThreadContext = createContext2(null);
-var useThreadContext = () => {
-  const context = useContext2(ThreadContext);
-  if (!context)
-    throw new Error("This component must be used within an AssistantProvider.");
-  return context;
-};
-
-// src/actions/useReloadMessage.tsx
 var useReloadMessage = () => {
   const { useThread, useViewport } = useThreadContext();
   const { useMessage } = useMessageContext();
@@ -436,20 +426,6 @@ var MessageProvider = ({
   return /* @__PURE__ */ jsx4(MessageContext.Provider, { value: context, children });
 };
 
-// src/context/ComposerContext.ts
-import { useContext as useContext3, useMemo as useMemo2 } from "react";
-var useComposerContext = () => {
-  const { useComposer } = useThreadContext();
-  const { useComposer: useEditComposer } = useContext3(MessageContext) ?? {};
-  return useMemo2(
-    () => ({
-      useComposer: useEditComposer ?? useComposer,
-      type: useEditComposer ? "edit" : "new"
-    }),
-    [useEditComposer, useComposer]
-  );
-};
-
 // src/primitives/composer/ComposerIf.tsx
 var useComposerIf = (props) => {
   const { useComposer } = useComposerContext();
@@ -769,22 +745,6 @@ var MessageRoot = forwardRef9(
 // src/context/providers/ContentPartProvider.tsx
 import { useEffect as useEffect5, useState as useState2 } from "react";
 import { create as create3 } from "zustand";
-
-// src/context/ContentPartContext.ts
-import { createContext as createContext3, useContext as useContext4 } from "react";
-var ContentPartContext = createContext3(
-  null
-);
-var useContentPartContext = () => {
-  const context = useContext4(ContentPartContext);
-  if (!context)
-    throw new Error(
-      "This component can only be used inside a component passed to <MessagePrimitive.Content components={...} >."
-    );
-  return context;
-};
-
-// src/context/providers/ContentPartProvider.tsx
 import { jsx as jsx13 } from "react/jsx-runtime";
 var syncContentPart = ({ message }, useContentPart, partIndex) => {
   const part = message.content[partIndex];
@@ -883,11 +843,11 @@ var MessageContent = ({
 import {
   Primitive as Primitive9
 } from "@radix-ui/react-primitive";
-import { forwardRef as forwardRef10, useMemo as useMemo3 } from "react";
+import { forwardRef as forwardRef10, useMemo as useMemo2 } from "react";
 import { jsx as jsx15 } from "react/jsx-runtime";
 var MessageInProgress = forwardRef10((props, ref) => {
   const { useMessage } = useMessageContext();
-  useMemo3(() => {
+  useMemo2(() => {
     useMessage.getState().setInProgressIndicator(/* @__PURE__ */ jsx15(Primitive9.span, { ...props, ref }));
   }, [useMessage, props, ref]);
   return null;
@@ -1016,585 +976,135 @@ __export(contentPart_exports, {
   InProgressIndicator: () => ContentPartInProgressIndicator
 });
 
-// ../react-ai-sdk/src/rsc/useVercelRSCRuntime.tsx
-import { useEffect as useEffect7, useInsertionEffect, useState as useState3 } from "react";
-
-// ../react-ai-sdk/src/rsc/VercelRSCRuntime.tsx
-import { create as create4 } from "zustand";
-
-// ../react-ai-sdk/src/rsc/useVercelRSCSync.tsx
-import { useEffect as useEffect6, useMemo as useMemo4 } from "react";
-
-// ../react-ai-sdk/src/utils/ThreadMessageConverter.ts
-var ThreadMessageConverter = class {
-  cache = /* @__PURE__ */ new WeakMap();
-  convertMessages(messages, converter, keyMapper = (key) => key) {
-    return messages.map((m) => {
-      const key = keyMapper(m);
-      const cached = this.cache.get(key);
-      const newMessage = converter(m, cached);
-      this.cache.set(key, newMessage);
-      return newMessage;
-    });
-  }
-};
-
-// ../react-ai-sdk/src/rsc/getVercelRSCMessage.tsx
-var symbolInnerRSCMessage = Symbol("innerVercelRSCMessage");
-var getVercelRSCMessage = (message) => {
-  return message[symbolInnerRSCMessage];
-};
-
-// ../react-ai-sdk/src/rsc/useVercelRSCSync.tsx
-var vercelToThreadMessage = (converter, rawMessage) => {
-  const message = converter(rawMessage);
-  return {
-    id: message.id,
-    role: message.role,
-    content: [{ type: "ui", display: message.display }],
-    createdAt: message.createdAt ?? /* @__PURE__ */ new Date(),
-    ...{ status: "done" },
-    [symbolInnerRSCMessage]: rawMessage
-  };
-};
-var useVercelRSCSync = (adapter, updateData) => {
-  const [converter, convertCallback] = useMemo4(() => {
-    const rscConverter = adapter.convertMessage ?? ((m) => m);
-    const convertCallback2 = (m, cache) => {
-      if (cache) return cache;
-      return vercelToThreadMessage(rscConverter, m);
-    };
-    return [new ThreadMessageConverter(), convertCallback2];
-  }, [adapter.convertMessage]);
-  useEffect6(() => {
-    updateData(converter.convertMessages(adapter.messages, convertCallback));
-  }, [updateData, converter, convertCallback, adapter.messages]);
-};
+// src/runtime/local/useLocalRuntime.tsx
+import { useInsertionEffect, useState as useState3 } from "react";
 
-// ../react-ai-sdk/src/rsc/VercelRSCRuntime.tsx
-var EMPTY_BRANCHES = Object.freeze([]);
-var VercelRSCRuntime = class {
+// src/runtime/local/LocalRuntime.tsx
+var LocalRuntime = class {
   constructor(adapter) {
     this.adapter = adapter;
-    this.useAdapter = create4(() => ({
-      adapter
-    }));
   }
-  useAdapter;
   _subscriptions = /* @__PURE__ */ new Set();
-  isRunning = false;
-  messages = [];
-  withRunning = (callback) => {
-    this.isRunning = true;
-    return callback.finally(() => {
-      this.isRunning = false;
-    });
-  };
-  getBranches() {
-    return EMPTY_BRANCHES;
-  }
-  switchToBranch() {
-    throw new Error(
-      "Branch switching is not supported by VercelRSCAssistantProvider."
-    );
-  }
-  async append(message) {
-    if (message.parentId !== (this.messages.at(-1)?.id ?? null)) {
-      if (!this.adapter.edit)
-        throw new Error(
-          "Message editing is not enabled, please provide an edit callback to VercelRSCAssistantProvider."
-        );
-      await this.withRunning(this.adapter.edit(message));
-    } else {
-      await this.withRunning(this.adapter.append(message));
-    }
-  }
-  async startRun(parentId) {
-    if (!this.adapter.reload)
-      throw new Error(
-        "Message reloading is not enabled, please provide a reload callback to VercelRSCAssistantProvider."
-      );
-    await this.withRunning(this.adapter.reload(parentId));
-  }
-  cancelRun() {
-    if (process.env["NODE_ENV"] === "development") {
-      console.warn(
-        "Run cancellation is not supported by VercelRSCAssistantProvider."
-      );
-    }
-  }
-  subscribe(callback) {
-    this._subscriptions.add(callback);
-    return () => this._subscriptions.delete(callback);
-  }
-  onAdapterUpdated() {
-    if (this.useAdapter.getState().adapter !== this.adapter) {
-      this.useAdapter.setState({ adapter: this.adapter });
-    }
-  }
-  updateData = (messages) => {
-    this.messages = messages;
-    for (const callback of this._subscriptions) callback();
-  };
-  unstable_synchronizer = () => {
-    const { adapter } = this.useAdapter();
-    useVercelRSCSync(adapter, this.updateData);
-    return null;
-  };
-};
-
-// ../react-ai-sdk/src/rsc/useVercelRSCRuntime.tsx
-var useVercelRSCRuntime = (adapter) => {
-  const [runtime] = useState3(() => new VercelRSCRuntime(adapter));
-  useInsertionEffect(() => {
-    runtime.adapter = adapter;
-  });
-  useEffect7(() => {
-    runtime.onAdapterUpdated();
-  });
-  return runtime;
-};
-
-// ../react-ai-sdk/src/ui/use-chat/useVercelUseChatRuntime.tsx
-import { useEffect as useEffect10, useInsertionEffect as useInsertionEffect2, useState as useState4 } from "react";
-
-// ../react-ai-sdk/src/ui/VercelAIRuntime.tsx
-import { create as create5 } from "zustand";
-
-// ../react-ai-sdk/src/ui/getVercelAIMessage.tsx
-var symbolInnerAIMessage = Symbol("innerVercelAIUIMessage");
-var getVercelAIMessage = (message) => {
-  return message[symbolInnerAIMessage];
-};
-
-// ../react-ai-sdk/src/ui/utils/sliceMessagesUntil.tsx
-var sliceMessagesUntil = (messages, messageId) => {
-  if (messageId == null) return [];
-  let messageIdx = messages.findIndex((m) => m.id === messageId);
-  if (messageIdx === -1)
-    throw new Error(
-      "useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui."
-    );
-  while (messages[messageIdx + 1]?.role === "assistant") {
-    messageIdx++;
-  }
-  return messages.slice(0, messageIdx + 1);
-};
-
-// ../react-ai-sdk/src/ui/utils/useVercelAIComposerSync.tsx
-import { useEffect as useEffect8 } from "react";
-var useVercelAIComposerSync = (vercel) => {
-  const { useComposer } = useThreadContext();
-  useEffect8(() => {
-    useComposer.setState({
-      value: vercel.input,
-      setValue: vercel.setInput
-    });
-  }, [useComposer, vercel.input, vercel.setInput]);
-};
-
-// ../react-ai-sdk/src/ui/utils/useVercelAIThreadSync.tsx
-import { useEffect as useEffect9, useMemo as useMemo5 } from "react";
-var getIsRunning = (vercel) => {
-  if ("isLoading" in vercel) return vercel.isLoading;
-  return vercel.status === "in_progress";
-};
-var vercelToThreadMessage2 = (messages, status) => {
-  const firstMessage = messages[0];
-  if (!firstMessage) throw new Error("No messages found");
-  const common = {
-    id: firstMessage.id,
-    createdAt: firstMessage.createdAt ?? /* @__PURE__ */ new Date(),
-    [symbolInnerAIMessage]: messages
-  };
-  switch (firstMessage.role) {
-    case "user":
-      if (messages.length > 1) {
-        throw new Error(
-          "Multiple user messages found. This is likely an internal bug in assistant-ui."
-        );
-      }
-      return {
-        ...common,
-        role: "user",
-        content: [{ type: "text", text: firstMessage.content }]
-      };
-    case "assistant":
-      return {
-        ...common,
-        role: "assistant",
-        content: messages.flatMap((message) => [
-          ...message.content ? [{ type: "text", text: message.content }] : [],
-          ...message.toolInvocations?.map(
-            (t) => ({
-              type: "tool-call",
-              name: t.toolName,
-              args: t.args,
-              result: "result" in t ? t.result : void 0
-            })
-          ) ?? []
-        ]),
-        status
-      };
-    default:
-      throw new Error(
-        `You have a message with an unsupported role. The role ${firstMessage.role} is not supported.`
-      );
-  }
-};
-var hasItems = (messages) => messages.length > 0;
-var chunkedMessages = (messages) => {
-  const chunks = [];
-  let currentChunk = [];
-  for (const message of messages) {
-    if (message.role === "assistant") {
-      currentChunk.push(message);
-    } else {
-      if (hasItems(currentChunk)) {
-        chunks.push(currentChunk);
-        currentChunk = [];
-      }
-      chunks.push([message]);
-    }
-  }
-  if (hasItems(currentChunk)) {
-    chunks.push(currentChunk);
-  }
-  return chunks;
-};
-var shallowArrayEqual = (a, b) => {
-  if (a.length !== b.length) return false;
-  for (let i = 0; i < a.length; i++) {
-    if (a[i] !== b[i]) return false;
+  _configProviders = /* @__PURE__ */ new Set();
+  abortController = null;
+  repository = new MessageRepository();
+  get messages() {
+    return this.repository.getMessages();
   }
-  return true;
-};
-var useVercelAIThreadSync = (vercel, updateData) => {
-  const isRunning = getIsRunning(vercel);
-  const converter = useMemo5(() => new ThreadMessageConverter(), []);
-  useEffect9(() => {
-    const lastMessageId = vercel.messages.at(-1)?.id;
-    const convertCallback = (messages2, cache) => {
-      const status = lastMessageId === messages2[0].id && isRunning ? "in_progress" : "done";
-      if (cache && shallowArrayEqual(cache.content, messages2) && (cache.role === "user" || cache.status === status))
-        return cache;
-      return vercelToThreadMessage2(messages2, status);
-    };
-    const messages = converter.convertMessages(
-      chunkedMessages(vercel.messages),
-      convertCallback,
-      (m) => m[0]
-    );
-    updateData(isRunning, messages);
-  }, [updateData, isRunning, vercel.messages, converter]);
-};
-
-// ../react-ai-sdk/src/ui/VercelAIRuntime.tsx
-var hasUpcomingMessage = (isRunning, messages) => {
-  return isRunning && messages[messages.length - 1]?.role !== "assistant";
-};
-var VercelAIRuntime = class {
-  constructor(vercel) {
-    this.vercel = vercel;
-    this.useVercel = create5(() => ({
-      vercel
-    }));
+  get isRunning() {
+    return this.abortController != null;
   }
-  _subscriptions = /* @__PURE__ */ new Set();
-  repository = new MessageRepository();
-  assistantOptimisticId = null;
-  useVercel;
-  messages = [];
-  isRunning = false;
   getBranches(messageId) {
     return this.repository.getBranches(messageId);
   }
   switchToBranch(branchId) {
     this.repository.switchToBranch(branchId);
-    this.updateVercelMessages(this.repository.getMessages());
+    this.notifySubscribers();
   }
   async append(message) {
-    if (message.content.length !== 1 || message.content[0]?.type !== "text")
-      throw new Error("Only text content is supported by Vercel AI SDK.");
-    const newMessages = sliceMessagesUntil(
-      this.vercel.messages,
-      message.parentId
-    );
-    this.vercel.setMessages(newMessages);
-    await this.vercel.append({
+    const userMessageId = generateId();
+    const userMessage = {
+      id: userMessageId,
       role: "user",
-      content: message.content[0].text
-    });
+      content: message.content,
+      createdAt: /* @__PURE__ */ new Date()
+    };
+    this.repository.addOrUpdateMessage(message.parentId, userMessage);
+    await this.startRun(userMessageId);
   }
   async startRun(parentId) {
-    const reloadMaybe = "reload" in this.vercel ? this.vercel.reload : void 0;
-    if (!reloadMaybe)
-      throw new Error(
-        "Reload is not supported by Vercel AI SDK's useAssistant."
-      );
-    const newMessages = sliceMessagesUntil(this.vercel.messages, parentId);
-    this.vercel.setMessages(newMessages);
-    await reloadMaybe();
+    const id = generateId();
+    this.repository.resetHead(parentId);
+    const messages = this.repository.getMessages();
+    const message = {
+      id,
+      role: "assistant",
+      status: "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].map((p) => p())),
+        onUpdate: updateHandler
+      });
+      updateHandler(result);
+      message.status = "done";
+      this.repository.addOrUpdateMessage(parentId, { ...message });
+    } catch (e) {
+      message.status = "error";
+      this.repository.addOrUpdateMessage(parentId, { ...message });
+      console.error(e);
+    } finally {
+      this.abortController = null;
+      this.notifySubscribers();
+    }
   }
   cancelRun() {
-    const previousMessage = this.vercel.messages.at(-1);
-    this.vercel.stop();
-    if (this.assistantOptimisticId) {
-      this.repository.deleteMessage(this.assistantOptimisticId);
-      this.assistantOptimisticId = null;
-    }
-    let messages = this.repository.getMessages();
-    if (previousMessage?.role === "user" && previousMessage.id === messages.at(-1)?.id) {
-      this.vercel.setInput(previousMessage.content);
-      this.repository.deleteMessage(previousMessage.id);
-      messages = this.repository.getMessages();
-    }
-    setTimeout(() => {
-      this.updateVercelMessages(messages);
-    }, 0);
+    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);
   }
-  updateVercelMessages = (messages) => {
-    this.vercel.setMessages(
-      messages.flatMap(getVercelAIMessage).filter((m) => m != null)
-    );
-  };
-  onVercelUpdated() {
-    if (this.useVercel.getState().vercel !== this.vercel) {
-      this.useVercel.setState({ vercel: this.vercel });
-    }
+  registerModelConfigProvider(provider) {
+    this._configProviders.add(provider);
+    return () => this._configProviders.delete(provider);
   }
-  updateData = (isRunning, vm) => {
-    for (let i = 0; i < vm.length; i++) {
-      const message = vm[i];
-      const parent = vm[i - 1];
-      this.repository.addOrUpdateMessage(parent?.id ?? null, message);
-    }
-    if (this.assistantOptimisticId) {
-      this.repository.deleteMessage(this.assistantOptimisticId);
-      this.assistantOptimisticId = null;
-    }
-    if (hasUpcomingMessage(isRunning, vm)) {
-      this.assistantOptimisticId = this.repository.appendOptimisticMessage(
-        vm.at(-1)?.id ?? null,
-        {
-          role: "assistant",
-          content: [{ type: "text", text: "" }]
-        }
-      );
-    }
-    this.repository.resetHead(
-      this.assistantOptimisticId ?? vm.at(-1)?.id ?? null
-    );
-    this.messages = this.repository.getMessages();
-    this.isRunning = isRunning;
-    for (const callback of this._subscriptions) callback();
-  };
-  unstable_synchronizer = () => {
-    const { vercel } = this.useVercel();
-    useVercelAIThreadSync(vercel, this.updateData);
-    useVercelAIComposerSync(vercel);
-    return null;
-  };
 };
 
-// ../react-ai-sdk/src/ui/use-chat/useVercelUseChatRuntime.tsx
-var useVercelUseChatRuntime = (chatHelpers) => {
-  const [runtime] = useState4(() => new VercelAIRuntime(chatHelpers));
-  useInsertionEffect2(() => {
-    runtime.vercel = chatHelpers;
-  });
-  useEffect10(() => {
-    runtime.onVercelUpdated();
+// src/runtime/local/useLocalRuntime.tsx
+var useLocalRuntime = (adapter) => {
+  const [runtime] = useState3(() => new LocalRuntime(adapter));
+  useInsertionEffect(() => {
+    runtime.adapter = adapter;
   });
   return runtime;
 };
 
-// ../react-ai-sdk/src/ui/use-assistant/useVercelUseAssistantRuntime.tsx
-import { useEffect as useEffect11, useInsertionEffect as useInsertionEffect3, useState as useState5 } from "react";
-var useVercelUseAssistantRuntime = (assistantHelpers) => {
-  const [runtime] = useState5(() => new VercelAIRuntime(assistantHelpers));
-  useInsertionEffect3(() => {
-    runtime.vercel = assistantHelpers;
-  });
-  useEffect11(() => {
-    runtime.onVercelUpdated();
-  });
-  return runtime;
-};
+// src/context/providers/AssistantRuntimeProvider.tsx
+import { memo } from "react";
 
-// src/runtime/utils/idUtils.tsx
-import { customAlphabet } from "nanoid/non-secure";
-var generateId = customAlphabet(
-  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
-  7
-);
-var optimisticPrefix = "__optimistic__";
-var generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;
+// src/context/providers/AssistantProvider.tsx
+import { useEffect as useEffect7, useInsertionEffect as useInsertionEffect3, useRef as useRef5, useState as useState5 } from "react";
 
-// src/runtime/utils/MessageRepository.tsx
-var findHead = (message) => {
-  if (message.next) return findHead(message.next);
-  return message;
-};
-var MessageRepository = class {
-  messages = /* @__PURE__ */ new Map();
-  // message_id -> item
-  head = null;
-  root = {
-    children: []
-  };
-  performOp(newParent, child, operation) {
-    const parentOrRoot = child.prev ?? this.root;
-    const newParentOrRoot = newParent ?? this.root;
-    if (operation === "relink" && parentOrRoot === newParentOrRoot) return;
-    if (operation !== "link") {
-      parentOrRoot.children = parentOrRoot.children.filter(
-        (m) => m !== child.current.id
-      );
-      if (child.prev?.next === child) {
-        const fallbackId = child.prev.children.at(-1);
-        const fallback = fallbackId ? this.messages.get(fallbackId) : null;
-        if (fallback === void 0) {
-          throw new Error(
-            "MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui."
-          );
-        }
-        child.prev.next = fallback;
-      }
-    }
-    if (operation !== "cut") {
-      newParentOrRoot.children = [
-        ...newParentOrRoot.children,
-        child.current.id
-      ];
-      if (newParent && (findHead(child) === this.head || newParent.next === null)) {
-        newParent.next = child;
-      }
-      child.prev = newParent;
-    }
-  }
-  getMessages() {
-    const messages = new Array(this.head?.level ?? 0);
-    for (let current = this.head; current; current = current.prev) {
-      messages[current.level] = current.current;
-    }
-    return messages;
-  }
-  addOrUpdateMessage(parentId, message) {
-    const existingItem = this.messages.get(message.id);
-    const prev = parentId ? this.messages.get(parentId) : null;
-    if (prev === void 0)
-      throw new Error(
-        "MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui."
-      );
-    if (existingItem) {
-      existingItem.current = message;
-      this.performOp(prev, existingItem, "relink");
-      return;
-    }
-    const newItem = {
-      prev,
-      current: message,
-      next: null,
-      children: [],
-      level: prev ? prev.level + 1 : 0
-    };
-    this.messages.set(message.id, newItem);
-    this.performOp(prev, newItem, "link");
-    if (this.head === prev) {
-      this.head = newItem;
-    }
-  }
-  appendOptimisticMessage(parentId, message) {
-    let optimisticId;
-    do {
-      optimisticId = generateOptimisticId();
-    } while (this.messages.has(optimisticId));
-    this.addOrUpdateMessage(parentId, {
-      ...message,
-      id: optimisticId,
-      createdAt: /* @__PURE__ */ new Date(),
-      ...message.role === "assistant" ? { status: "in_progress" } : void 0
-    });
-    return optimisticId;
-  }
-  deleteMessage(messageId, replacementId) {
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(deleteMessage): Optimistic message not found. This is likely an internal bug in assistant-ui."
-      );
-    const replacement = replacementId === void 0 ? message.prev : replacementId === null ? null : this.messages.get(replacementId);
-    if (replacement === void 0)
-      throw new Error(
-        "MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui."
-      );
-    for (const child of message.children) {
-      const childMessage = this.messages.get(child);
-      if (!childMessage)
-        throw new Error(
-          "MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui."
-        );
-      this.performOp(replacement, childMessage, "relink");
-    }
-    this.performOp(null, message, "cut");
-    this.messages.delete(messageId);
-    if (this.head === message) {
-      this.head = replacement ? findHead(replacement) : null;
-    }
-  }
-  getBranches(messageId) {
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui."
-      );
-    const { children } = message.prev ?? this.root;
-    return children;
-  }
-  switchToBranch(messageId) {
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui."
-      );
-    if (message.prev) {
-      message.prev.next = message;
-    }
-    this.head = findHead(message);
-  }
-  resetHead(messageId) {
-    if (messageId === null) {
-      this.head = null;
-      return;
-    }
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui."
-      );
-    this.head = message;
-    for (let current = message; current; current = current.prev) {
-      if (current.prev) {
-        current.prev.next = current;
-      }
+// src/context/stores/AssistantModelConfig.ts
+import { create as create4 } from "zustand";
+var makeAssistantModelConfigStore = () => create4(() => {
+  const proxy = new ProxyConfigProvider();
+  return {
+    getModelConfig: () => {
+      return proxy.getModelConfig();
+    },
+    registerModelConfigProvider: (provider) => {
+      return proxy.registerModelConfigProvider(provider);
     }
-  }
-};
-
-// src/context/providers/AssistantRuntimeProvider.tsx
-import { memo } from "react";
+  };
+});
 
 // src/context/providers/ThreadProvider.tsx
-import { useEffect as useEffect12, useInsertionEffect as useInsertionEffect4, useRef as useRef4, useState as useState6 } from "react";
+import { useEffect as useEffect6, useInsertionEffect as useInsertionEffect2, useRef as useRef4, useState as useState4 } from "react";
 
 // src/context/stores/Composer.ts
-import { create as create6 } from "zustand";
-var makeComposerStore = (useThread) => create6()((set, get, store) => {
+import { create as create5 } from "zustand";
+var makeComposerStore = (useThread) => create5()((set, get, store) => {
   return {
     ...makeBaseComposer(set, get, store),
     isEditing: true,
@@ -1616,9 +1126,9 @@ var makeComposerStore = (useThread) => create6()((set, get, store) => {
 });
 
 // src/context/stores/Thread.ts
-import { create as create7 } from "zustand";
+import { create as create6 } from "zustand";
 var makeThreadStore = (runtimeRef) => {
-  const useThread = create7(() => ({
+  const useThread = create6(() => ({
     messages: runtimeRef.current.messages,
     isRunning: runtimeRef.current.isRunning,
     getBranches: (messageId) => runtimeRef.current.getBranches(messageId),
@@ -1640,10 +1150,10 @@ var makeThreadStore = (runtimeRef) => {
 };
 
 // src/context/stores/ThreadViewport.tsx
-import { create as create8 } from "zustand";
+import { create as create7 } from "zustand";
 var makeThreadViewportStore = () => {
   const scrollToBottomListeners = /* @__PURE__ */ new Set();
-  return create8(() => ({
+  return create7(() => ({
     isAtBottom: true,
     scrollToBottom: () => {
       for (const listener of scrollToBottomListeners) {
@@ -1666,10 +1176,10 @@ var ThreadProvider = ({
   runtime
 }) => {
   const runtimeRef = useRef4(runtime);
-  useInsertionEffect4(() => {
+  useInsertionEffect2(() => {
     runtimeRef.current = runtime;
   });
-  const [{ context, onRuntimeUpdate }] = useState6(() => {
+  const [{ context, onRuntimeUpdate }] = useState4(() => {
     const { useThread, onRuntimeUpdate: onRuntimeUpdate2 } = makeThreadStore(runtimeRef);
     const useViewport = makeThreadViewportStore();
     const useComposer = makeComposerStore(useThread);
@@ -1682,7 +1192,7 @@ var ThreadProvider = ({
       onRuntimeUpdate: onRuntimeUpdate2
     };
   });
-  useEffect12(() => {
+  useEffect6(() => {
     onRuntimeUpdate();
     return runtime.subscribe(onRuntimeUpdate);
   }, [onRuntimeUpdate, runtime]);
@@ -1693,145 +1203,30 @@ var ThreadProvider = ({
   ] });
 };
 
-// src/context/providers/AssistantRuntimeProvider.tsx
+// src/context/providers/AssistantProvider.tsx
 import { jsx as jsx22 } from "react/jsx-runtime";
-var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
-  return /* @__PURE__ */ jsx22(ThreadProvider, { runtime, children });
+var AssistantProvider = ({ children, runtime }) => {
+  const runtimeRef = useRef5(runtime);
+  useInsertionEffect3(() => {
+    runtimeRef.current = runtime;
+  });
+  const [context] = useState5(() => {
+    const useModelConfig = makeAssistantModelConfigStore();
+    return { useModelConfig };
+  });
+  const getModelCOnfig = context.useModelConfig((c) => c.getModelConfig);
+  useEffect7(() => {
+    return runtime.registerModelConfigProvider(getModelCOnfig);
+  }, [runtime, getModelCOnfig]);
+  return /* @__PURE__ */ jsx22(AssistantContext.Provider, { value: context, children: /* @__PURE__ */ jsx22(ThreadProvider, { runtime, children }) });
 };
-var AssistantRuntimeProvider = memo(AssistantRuntimeProviderImpl);
 
-// src/runtime/vercel-deprecated/VercelAIAssistantProvider.tsx
+// src/context/providers/AssistantRuntimeProvider.tsx
 import { jsx as jsx23 } from "react/jsx-runtime";
-var VercelUseChatRuntimeProvider = ({
-  chat,
-  children
-}) => {
-  const runtime = useVercelUseChatRuntime(chat);
-  return /* @__PURE__ */ jsx23(AssistantRuntimeProvider, { runtime, children });
-};
-var VercelUseAssistantRuntimeProvider = ({
-  assistant,
-  children
-}) => {
-  const runtime = useVercelUseAssistantRuntime(assistant);
-  return /* @__PURE__ */ jsx23(AssistantRuntimeProvider, { runtime, children });
-};
-var VercelAIAssistantProvider = ({
-  children,
-  ...rest
-}) => {
-  if ("chat" in rest) {
-    return /* @__PURE__ */ jsx23(VercelUseChatRuntimeProvider, { chat: rest.chat, children });
-  }
-  return /* @__PURE__ */ jsx23(VercelUseAssistantRuntimeProvider, { assistant: rest.assistant, children });
-};
-
-// src/runtime/vercel-deprecated/VercelRSCAssistantProvider.tsx
-import { jsx as jsx24 } from "react/jsx-runtime";
-var VercelRSCAssistantProvider = ({
-  children,
-  ...adapter
-}) => {
-  const runtime = useVercelRSCRuntime(adapter);
-  return /* @__PURE__ */ jsx24(AssistantRuntimeProvider, { runtime, children });
-};
-
-// src/runtime/local/useLocalRuntime.tsx
-import { useInsertionEffect as useInsertionEffect5, useState as useState7 } from "react";
-
-// src/runtime/local/LocalRuntime.tsx
-var LocalRuntime = class {
-  constructor(adapter) {
-    this.adapter = adapter;
-  }
-  _subscriptions = /* @__PURE__ */ new Set();
-  abortController = null;
-  repository = new MessageRepository();
-  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: "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,
-        onUpdate: updateHandler
-      });
-      updateHandler(result);
-      message.status = "done";
-      this.repository.addOrUpdateMessage(parentId, { ...message });
-    } catch (e) {
-      message.status = "error";
-      this.repository.addOrUpdateMessage(parentId, { ...message });
-      console.error(e);
-    } finally {
-      this.abortController = null;
-      this.notifySubscribers();
-    }
-  }
-  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);
-  }
-};
-
-// src/runtime/local/useLocalRuntime.tsx
-var useLocalRuntime = (adapter) => {
-  const [runtime] = useState7(() => new LocalRuntime(adapter));
-  useInsertionEffect5(() => {
-    runtime.adapter = adapter;
-  });
-  return runtime;
+var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
+  return /* @__PURE__ */ jsx23(AssistantProvider, { runtime, children });
 };
+var AssistantRuntimeProvider = memo(AssistantRuntimeProviderImpl);
 export {
   actionBar_exports as ActionBarPrimitive,
   AssistantRuntimeProvider,
@@ -1840,23 +1235,11 @@ export {
   contentPart_exports as ContentPartPrimitive,
   message_exports as MessagePrimitive,
   thread_exports as ThreadPrimitive,
-  VercelAIAssistantProvider,
-  VercelRSCAssistantProvider,
-  getVercelAIMessage,
-  getVercelRSCMessage,
-  MessageRepository as unstable_MessageRepository,
-  useComposerContext as unstable_useComposerContext,
-  useContentPartContext as unstable_useContentPartContext,
-  useLocalRuntime as unstable_useLocalRuntime,
-  useMessageContext as unstable_useMessageContext,
-  useThreadContext as unstable_useThreadContext,
   useBeginMessageEdit,
   useCopyMessage,
   useGoToNextBranch,
   useGoToPreviousBranch,
-  useReloadMessage,
-  useVercelRSCRuntime,
-  useVercelUseAssistantRuntime,
-  useVercelUseChatRuntime
+  useLocalRuntime,
+  useReloadMessage
 };
 //# sourceMappingURL=index.mjs.map