@assistant-ui/react 0.0.29 → 0.1.1

package/dist/index.js

@@ -37,24 +37,12 @@ __export(src_exports, {
   ContentPartPrimitive: () => contentPart_exports,
   MessagePrimitive: () => message_exports,
   ThreadPrimitive: () => thread_exports,
-  VercelAIAssistantProvider: () => VercelAIAssistantProvider,
-  VercelRSCAssistantProvider: () => VercelRSCAssistantProvider,
-  getVercelAIMessage: () => getVercelAIMessage,
-  getVercelRSCMessage: () => getVercelRSCMessage,
-  unstable_MessageRepository: () => MessageRepository,
-  unstable_useComposerContext: () => useComposerContext,
-  unstable_useContentPartContext: () => useContentPartContext,
-  unstable_useLocalRuntime: () => useLocalRuntime,
-  unstable_useMessageContext: () => useMessageContext,
-  unstable_useThreadContext: () => useThreadContext,
   useBeginMessageEdit: () => useBeginMessageEdit,
   useCopyMessage: () => useCopyMessage,
   useGoToNextBranch: () => useGoToNextBranch,
   useGoToPreviousBranch: () => useGoToPreviousBranch,
-  useReloadMessage: () => useReloadMessage,
-  useVercelRSCRuntime: () => useVercelRSCRuntime,
-  useVercelUseAssistantRuntime: () => useVercelUseAssistantRuntime,
-  useVercelUseChatRuntime: () => useVercelUseChatRuntime
+  useLocalRuntime: () => useLocalRuntime,
+  useReloadMessage: () => useReloadMessage
 });
 module.exports = __toCommonJS(src_exports);
 
@@ -137,7 +125,7 @@ var ThreadContext = (0, import_react5.createContext)(null);
 var useThreadContext = () => {
   const context = (0, import_react5.useContext)(ThreadContext);
   if (!context)
-    throw new Error("This component must be used within an AssistantProvider.");
+    throw new Error("This component must be used within an AssistantRuntimeProvider.");
   return context;
 };
 
@@ -1041,414 +1029,26 @@ __export(contentPart_exports, {
   InProgressIndicator: () => ContentPartInProgressIndicator
 });
 
-// ../react-ai-sdk/src/rsc/useVercelRSCRuntime.tsx
-var import_react30 = require("react");
-
-// ../react-ai-sdk/src/rsc/VercelRSCRuntime.tsx
-var import_zustand4 = require("zustand");
-
-// ../react-ai-sdk/src/rsc/useVercelRSCSync.tsx
+// src/runtime/local/useLocalRuntime.tsx
 var import_react29 = require("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] = (0, import_react29.useMemo)(() => {
-    const rscConverter = adapter.convertMessage ?? ((m) => m);
-    const convertCallback2 = (m, cache) => {
-      if (cache) return cache;
-      return vercelToThreadMessage(rscConverter, m);
-    };
-    return [new ThreadMessageConverter(), convertCallback2];
-  }, [adapter.convertMessage]);
-  (0, import_react29.useEffect)(() => {
-    updateData(converter.convertMessages(adapter.messages, convertCallback));
-  }, [updateData, converter, convertCallback, adapter.messages]);
-};
-
-// ../react-ai-sdk/src/rsc/VercelRSCRuntime.tsx
-var EMPTY_BRANCHES = Object.freeze([]);
-var VercelRSCRuntime = class {
-  constructor(adapter) {
-    this.adapter = adapter;
-    this.useAdapter = (0, import_zustand4.create)(() => ({
-      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] = (0, import_react30.useState)(() => new VercelRSCRuntime(adapter));
-  (0, import_react30.useInsertionEffect)(() => {
-    runtime.adapter = adapter;
-  });
-  (0, import_react30.useEffect)(() => {
-    runtime.onAdapterUpdated();
-  });
-  return runtime;
-};
-
-// ../react-ai-sdk/src/ui/use-chat/useVercelUseChatRuntime.tsx
-var import_react35 = require("react");
-
-// ../react-ai-sdk/src/ui/VercelAIRuntime.tsx
-var import_zustand5 = require("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
-var import_react32 = require("react");
-var useVercelAIComposerSync = (vercel) => {
-  const { useComposer } = useThreadContext();
-  (0, import_react32.useEffect)(() => {
-    useComposer.setState({
-      value: vercel.input,
-      setValue: vercel.setInput
-    });
-  }, [useComposer, vercel.input, vercel.setInput]);
-};
+// src/utils/ModelConfigTypes.ts
+var mergeModelConfigs = (configs) => {
+  return configs.reduce((acc, config) => {
+    if (config.system) {
+      if (acc.system) {
+        acc.system += `
 
-// ../react-ai-sdk/src/ui/utils/useVercelAIThreadSync.tsx
-var import_react33 = require("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 = [];
+${config.system}`;
+      } else {
+        acc.system = config.system;
       }
-      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;
-  }
-  return true;
-};
-var useVercelAIThreadSync = (vercel, updateData) => {
-  const isRunning = getIsRunning(vercel);
-  const converter = (0, import_react33.useMemo)(() => new ThreadMessageConverter(), []);
-  (0, import_react33.useEffect)(() => {
-    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 = (0, import_zustand5.create)(() => ({
-      vercel
-    }));
-  }
-  _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());
-  }
-  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({
-      role: "user",
-      content: message.content[0].text
-    });
-  }
-  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();
-  }
-  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);
-  }
-  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 });
-    }
-  }
-  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: "" }]
-        }
-      );
+    if (config.tools) {
+      acc.tools = { ...acc.tools, ...config.tools };
     }
-    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] = (0, import_react35.useState)(() => new VercelAIRuntime(chatHelpers));
-  (0, import_react35.useInsertionEffect)(() => {
-    runtime.vercel = chatHelpers;
-  });
-  (0, import_react35.useEffect)(() => {
-    runtime.onVercelUpdated();
-  });
-  return runtime;
-};
-
-// ../react-ai-sdk/src/ui/use-assistant/useVercelUseAssistantRuntime.tsx
-var import_react36 = require("react");
-var useVercelUseAssistantRuntime = (assistantHelpers) => {
-  const [runtime] = (0, import_react36.useState)(() => new VercelAIRuntime(assistantHelpers));
-  (0, import_react36.useInsertionEffect)(() => {
-    runtime.vercel = assistantHelpers;
-  });
-  (0, import_react36.useEffect)(() => {
-    runtime.onVercelUpdated();
-  });
-  return runtime;
+    return acc;
+  }, {});
 };
 
 // src/runtime/utils/idUtils.tsx
@@ -1611,15 +1211,154 @@ var MessageRepository = class {
   }
 };
 
+// src/runtime/local/LocalRuntime.tsx
+var LocalRuntime = class {
+  constructor(adapter) {
+    this.adapter = adapter;
+  }
+  _subscriptions = /* @__PURE__ */ new Set();
+  _configProviders = /* @__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,
+        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() {
+    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);
+  }
+  registerModelConfigProvider(provider) {
+    this._configProviders.add(provider);
+    return () => this._configProviders.delete(provider);
+  }
+};
+
+// src/runtime/local/useLocalRuntime.tsx
+var useLocalRuntime = (adapter) => {
+  const [runtime] = (0, import_react29.useState)(() => new LocalRuntime(adapter));
+  (0, import_react29.useInsertionEffect)(() => {
+    runtime.adapter = adapter;
+  });
+  return runtime;
+};
+
 // src/context/providers/AssistantRuntimeProvider.tsx
-var import_react38 = require("react");
+var import_react33 = require("react");
+
+// src/context/providers/AssistantProvider.tsx
+var import_react32 = require("react");
+
+// src/context/AssistantContext.ts
+var import_react30 = require("react");
+var AssistantContext = (0, import_react30.createContext)(
+  null
+);
+
+// src/context/stores/AssistantModelConfig.ts
+var import_zustand4 = require("zustand");
+
+// src/utils/ProxyConfigProvider.ts
+var ProxyConfigProvider = class {
+  _providers = /* @__PURE__ */ new Set();
+  getModelConfig() {
+    return mergeModelConfigs([...this._providers].map((p) => p()));
+  }
+  registerModelConfigProvider(provider) {
+    this._providers.add(provider);
+    return () => {
+      this._providers.delete(provider);
+    };
+  }
+};
+
+// src/context/stores/AssistantModelConfig.ts
+var makeAssistantModelConfigStore = () => (0, import_zustand4.create)(() => {
+  const proxy = new ProxyConfigProvider();
+  return {
+    getModelConfig: () => {
+      return proxy.getModelConfig();
+    },
+    registerModelConfigProvider: (provider) => {
+      return proxy.registerModelConfigProvider(provider);
+    }
+  };
+});
 
 // src/context/providers/ThreadProvider.tsx
-var import_react37 = require("react");
+var import_react31 = require("react");
 
 // src/context/stores/Composer.ts
-var import_zustand6 = require("zustand");
-var makeComposerStore = (useThread) => (0, import_zustand6.create)()((set, get, store) => {
+var import_zustand5 = require("zustand");
+var makeComposerStore = (useThread) => (0, import_zustand5.create)()((set, get, store) => {
   return {
     ...makeBaseComposer(set, get, store),
     isEditing: true,
@@ -1641,9 +1380,9 @@ var makeComposerStore = (useThread) => (0, import_zustand6.create)()((set, get,
 });
 
 // src/context/stores/Thread.ts
-var import_zustand7 = require("zustand");
+var import_zustand6 = require("zustand");
 var makeThreadStore = (runtimeRef) => {
-  const useThread = (0, import_zustand7.create)(() => ({
+  const useThread = (0, import_zustand6.create)(() => ({
     messages: runtimeRef.current.messages,
     isRunning: runtimeRef.current.isRunning,
     getBranches: (messageId) => runtimeRef.current.getBranches(messageId),
@@ -1665,10 +1404,10 @@ var makeThreadStore = (runtimeRef) => {
 };
 
 // src/context/stores/ThreadViewport.tsx
-var import_zustand8 = require("zustand");
+var import_zustand7 = require("zustand");
 var makeThreadViewportStore = () => {
   const scrollToBottomListeners = /* @__PURE__ */ new Set();
-  return (0, import_zustand8.create)(() => ({
+  return (0, import_zustand7.create)(() => ({
     isAtBottom: true,
     scrollToBottom: () => {
       for (const listener of scrollToBottomListeners) {
@@ -1690,11 +1429,11 @@ var ThreadProvider = ({
   children,
   runtime
 }) => {
-  const runtimeRef = (0, import_react37.useRef)(runtime);
-  (0, import_react37.useInsertionEffect)(() => {
+  const runtimeRef = (0, import_react31.useRef)(runtime);
+  (0, import_react31.useInsertionEffect)(() => {
     runtimeRef.current = runtime;
   });
-  const [{ context, onRuntimeUpdate }] = (0, import_react37.useState)(() => {
+  const [{ context, onRuntimeUpdate }] = (0, import_react31.useState)(() => {
     const { useThread, onRuntimeUpdate: onRuntimeUpdate2 } = makeThreadStore(runtimeRef);
     const useViewport = makeThreadViewportStore();
     const useComposer = makeComposerStore(useThread);
@@ -1707,7 +1446,7 @@ var ThreadProvider = ({
       onRuntimeUpdate: onRuntimeUpdate2
     };
   });
-  (0, import_react37.useEffect)(() => {
+  (0, import_react31.useEffect)(() => {
     onRuntimeUpdate();
     return runtime.subscribe(onRuntimeUpdate);
   }, [onRuntimeUpdate, runtime]);
@@ -1718,145 +1457,30 @@ var ThreadProvider = ({
   ] });
 };
 
-// src/context/providers/AssistantRuntimeProvider.tsx
+// src/context/providers/AssistantProvider.tsx
 var import_jsx_runtime22 = require("react/jsx-runtime");
-var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(ThreadProvider, { runtime, children });
+var AssistantProvider = ({ children, runtime }) => {
+  const runtimeRef = (0, import_react32.useRef)(runtime);
+  (0, import_react32.useInsertionEffect)(() => {
+    runtimeRef.current = runtime;
+  });
+  const [context] = (0, import_react32.useState)(() => {
+    const useModelConfig = makeAssistantModelConfigStore();
+    return { useModelConfig };
+  });
+  const getModelCOnfig = context.useModelConfig((c) => c.getModelConfig);
+  (0, import_react32.useEffect)(() => {
+    return runtime.registerModelConfigProvider(getModelCOnfig);
+  }, [runtime, getModelCOnfig]);
+  return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(AssistantContext.Provider, { value: context, children: /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(ThreadProvider, { runtime, children }) });
 };
-var AssistantRuntimeProvider = (0, import_react38.memo)(AssistantRuntimeProviderImpl);
 
-// src/runtime/vercel-deprecated/VercelAIAssistantProvider.tsx
+// src/context/providers/AssistantRuntimeProvider.tsx
 var import_jsx_runtime23 = require("react/jsx-runtime");
-var VercelUseChatRuntimeProvider = ({
-  chat,
-  children
-}) => {
-  const runtime = useVercelUseChatRuntime(chat);
-  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(AssistantRuntimeProvider, { runtime, children });
-};
-var VercelUseAssistantRuntimeProvider = ({
-  assistant,
-  children
-}) => {
-  const runtime = useVercelUseAssistantRuntime(assistant);
-  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(AssistantRuntimeProvider, { runtime, children });
-};
-var VercelAIAssistantProvider = ({
-  children,
-  ...rest
-}) => {
-  if ("chat" in rest) {
-    return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(VercelUseChatRuntimeProvider, { chat: rest.chat, children });
-  }
-  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(VercelUseAssistantRuntimeProvider, { assistant: rest.assistant, children });
-};
-
-// src/runtime/vercel-deprecated/VercelRSCAssistantProvider.tsx
-var import_jsx_runtime24 = require("react/jsx-runtime");
-var VercelRSCAssistantProvider = ({
-  children,
-  ...adapter
-}) => {
-  const runtime = useVercelRSCRuntime(adapter);
-  return /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(AssistantRuntimeProvider, { runtime, children });
-};
-
-// src/runtime/local/useLocalRuntime.tsx
-var import_react39 = require("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] = (0, import_react39.useState)(() => new LocalRuntime(adapter));
-  (0, import_react39.useInsertionEffect)(() => {
-    runtime.adapter = adapter;
-  });
-  return runtime;
+var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
+  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(AssistantProvider, { runtime, children });
 };
+var AssistantRuntimeProvider = (0, import_react33.memo)(AssistantRuntimeProviderImpl);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ActionBarPrimitive,
@@ -1866,23 +1490,11 @@ var useLocalRuntime = (adapter) => {
   ContentPartPrimitive,
   MessagePrimitive,
   ThreadPrimitive,
-  VercelAIAssistantProvider,
-  VercelRSCAssistantProvider,
-  getVercelAIMessage,
-  getVercelRSCMessage,
-  unstable_MessageRepository,
-  unstable_useComposerContext,
-  unstable_useContentPartContext,
-  unstable_useLocalRuntime,
-  unstable_useMessageContext,
-  unstable_useThreadContext,
   useBeginMessageEdit,
   useCopyMessage,
   useGoToNextBranch,
   useGoToPreviousBranch,
-  useReloadMessage,
-  useVercelRSCRuntime,
-  useVercelUseAssistantRuntime,
-  useVercelUseChatRuntime
+  useLocalRuntime,
+  useReloadMessage
 });
 //# sourceMappingURL=index.js.map