@assistant-ui/react 0.5.36 → 0.5.37

package/dist/index.mjs

@@ -1298,6 +1298,19 @@ var LocalRuntime = class extends BaseAssistantRuntime {
       this.thread.adapter
     );
   }
+  reset({
+    initialMessages
+  } = {}) {
+    this.switchToThread(null);
+    if (!initialMessages) return;
+    const messages = fromCoreMessages(initialMessages);
+    this.thread.import({
+      messages: messages.map((m, idx) => ({
+        parentId: messages[idx - 1]?.id ?? null,
+        message: m
+      }))
+    });
+  }
 };
 
 // src/runtimes/local/useLocalRuntime.tsx
@@ -1723,6 +1736,12 @@ var makeAssistantActionsStore = (runtimeRef) => create10(
   })
 );
 
+// src/context/stores/AssistantRuntime.tsx
+import { create as create11 } from "zustand";
+var makeAssistantRuntimeStore = (runtime) => {
+  return create11(() => runtime);
+};
+
 // src/context/providers/AssistantProvider.tsx
 import { jsx as jsx7 } from "react/jsx-runtime";
 var AssistantProvider = ({ children, runtime }) => {
@@ -1731,15 +1750,25 @@ var AssistantProvider = ({ children, runtime }) => {
     runtimeRef.current = runtime;
   });
   const [context] = useState8(() => {
+    const useAssistantRuntime = makeAssistantRuntimeStore(runtime);
     const useModelConfig = makeAssistantModelConfigStore();
     const useToolUIs = makeAssistantToolUIsStore();
     const useAssistantActions = makeAssistantActionsStore(runtimeRef);
-    return { useModelConfig, useToolUIs, useAssistantActions };
+    return {
+      useModelConfig,
+      useToolUIs,
+      useAssistantRuntime,
+      useAssistantActions
+    };
   });
   const getModelConfig = context.useModelConfig();
   useEffect4(() => {
     return runtime.registerModelConfigProvider(getModelConfig);
   }, [runtime, getModelConfig]);
+  useEffect4(
+    () => context.useAssistantRuntime.setState(runtime, true),
+    [runtime, context]
+  );
   return /* @__PURE__ */ jsx7(AssistantContext.Provider, { value: context, children: /* @__PURE__ */ jsx7(ThreadProvider, { provider: runtime, children }) });
 };
 
@@ -2561,7 +2590,7 @@ import { memo as memo2 } from "react";
 
 // src/context/providers/ContentPartProvider.tsx
 import { useEffect as useEffect9, useState as useState10 } from "react";
-import { create as create11 } from "zustand";
+import { create as create12 } from "zustand";
 import { jsx as jsx18 } from "react/jsx-runtime";
 var COMPLETE_STATUS = {
   type: "complete"
@@ -2600,7 +2629,7 @@ var getContentPartState = ({ message }, useContentPart, partIndex) => {
 var useContentPartContext2 = (partIndex) => {
   const { useMessage } = useMessageContext();
   const [context] = useState10(() => {
-    const useContentPart = create11(
+    const useContentPart = create12(
       () => getContentPartState(useMessage.getState(), void 0, partIndex)
     );
     return { useContentPart };
@@ -3108,14 +3137,14 @@ import { memo as memo3 } from "react";
 
 // src/context/providers/MessageProvider.tsx
 import { useEffect as useEffect12, useState as useState11 } from "react";
-import { create as create14 } from "zustand";
+import { create as create15 } from "zustand";
 
 // src/context/stores/EditComposer.ts
-import { create as create12 } from "zustand";
+import { create as create13 } from "zustand";
 var makeEditComposerStore = ({
   onEdit,
   onSend
-}) => create12()((set, get) => ({
+}) => create13()((set, get) => ({
   get value() {
     return get().text;
   },
@@ -3143,8 +3172,8 @@ var makeEditComposerStore = ({
 }));
 
 // src/context/stores/MessageUtils.ts
-import { create as create13 } from "zustand";
-var makeMessageUtilsStore = () => create13((set) => ({
+import { create as create14 } from "zustand";
+var makeMessageUtilsStore = () => create14((set) => ({
   isCopied: false,
   setIsCopied: (value) => {
     set({ isCopied: value });
@@ -3179,7 +3208,7 @@ var getMessageState = (messages, getBranches, useMessage, messageIndex) => {
 var useMessageContext2 = (messageIndex) => {
   const { useThreadMessages, useThreadActions } = useThreadContext();
   const [context] = useState11(() => {
-    const useMessage = create14(
+    const useMessage = create15(
       () => getMessageState(
         useThreadMessages.getState(),
         useThreadActions.getState().getBranches,