@assistant-ui/react 0.5.35 → 0.5.37

package/dist/index.mjs

@@ -1,6 +1,7 @@
 "use client";
 import {
   PipeableTransformStream,
+  getEdgeRuntimeStream,
   mergeModelConfigs,
   runResultStream,
   streamPartEncoderStream,
@@ -9,7 +10,7 @@ import {
   toLanguageModelMessages,
   toLanguageModelTools,
   toolResultStream
-} from "./chunk-LX7F2T2K.mjs";
+} from "./chunk-5ZTUOAPH.mjs";
 import {
   __export
 } from "./chunk-BJPOCE4O.mjs";
@@ -18,7 +19,7 @@ import {
 import { memo } from "react";
 
 // src/context/providers/AssistantProvider.tsx
-import { useEffect as useEffect4, useInsertionEffect as useInsertionEffect3, useRef as useRef2, useState as useState7 } from "react";
+import { useEffect as useEffect4, useInsertionEffect as useInsertionEffect3, useRef as useRef2, useState as useState8 } from "react";
 
 // src/context/react/AssistantContext.ts
 import { createContext, useContext } from "react";
@@ -97,7 +98,7 @@ var makeAssistantToolUIsStore = () => create2((set) => {
 });
 
 // src/context/providers/ThreadProvider.tsx
-import { useEffect as useEffect3, useInsertionEffect as useInsertionEffect2, useState as useState6 } from "react";
+import { useEffect as useEffect3, useInsertionEffect as useInsertionEffect2, useState as useState7 } from "react";
 
 // src/context/react/ThreadContext.ts
 import { createContext as createContext2, useContext as useContext2 } from "react";
@@ -1297,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
@@ -1614,13 +1628,49 @@ var useExternalStoreRuntime = (store) => {
   return runtime;
 };
 
+// src/runtimes/dangerous-in-browser/useDangerousInBrowserRuntime.ts
+import { useState as useState6 } from "react";
+
+// src/runtimes/dangerous-in-browser/DangerousInBrowserAdapter.ts
+var DangerousInBrowserAdapter = class {
+  constructor(options) {
+    this.options = options;
+  }
+  async *run({ messages, abortSignal, config }) {
+    const res = await getEdgeRuntimeStream({
+      options: this.options,
+      abortSignal,
+      requestData: {
+        system: config.system,
+        messages: toCoreMessages(messages),
+        tools: config.tools ? toLanguageModelTools(config.tools) : [],
+        ...config.callSettings,
+        ...config.config
+      }
+    });
+    const stream = res.pipeThrough(toolResultStream(config.tools, abortSignal)).pipeThrough(runResultStream());
+    for await (const update of asAsyncIterable(stream)) {
+      yield update;
+    }
+  }
+};
+
+// src/runtimes/dangerous-in-browser/useDangerousInBrowserRuntime.ts
+var useDangerousInBrowserRuntime = ({
+  initialMessages,
+  ...options
+}) => {
+  const [adapter] = useState6(() => new DangerousInBrowserAdapter(options));
+  return useLocalRuntime(adapter, { initialMessages });
+};
+
 // src/context/providers/ThreadProvider.tsx
 import { jsx as jsx6, jsxs as jsxs2 } from "react/jsx-runtime";
 var ThreadProvider = ({
   children,
   provider
 }) => {
-  const [context] = useState6(() => {
+  const [context] = useState7(() => {
     const useThreadRuntime = makeThreadRuntimeStore(provider.thread);
     const useThread = makeThreadStore(useThreadRuntime);
     const useThreadMessages = makeThreadMessagesStore(useThreadRuntime);
@@ -1686,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 }) => {
@@ -1693,16 +1749,26 @@ var AssistantProvider = ({ children, runtime }) => {
   useInsertionEffect3(() => {
     runtimeRef.current = runtime;
   });
-  const [context] = useState7(() => {
+  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 }) });
 };
 
@@ -2285,7 +2351,7 @@ __export(assistantModal_exports, {
 });
 
 // src/primitives/assistantModal/AssistantModalRoot.tsx
-import { useState as useState8 } from "react";
+import { useState as useState9 } from "react";
 import * as PopoverPrimitive2 from "@radix-ui/react-popover";
 import { composeEventHandlers as composeEventHandlers2 } from "@radix-ui/primitive";
 
@@ -2309,7 +2375,7 @@ var usePopoverScope = PopoverPrimitive.createPopoverScope();
 // src/primitives/assistantModal/AssistantModalRoot.tsx
 import { jsx as jsx11 } from "react/jsx-runtime";
 var useAssistantModalOpenState = (defaultOpen = false) => {
-  const state = useState8(defaultOpen);
+  const state = useState9(defaultOpen);
   const [, setOpen] = state;
   useOnComposerFocus(() => {
     setOpen(true);
@@ -2523,8 +2589,8 @@ MessagePrimitiveIf.displayName = "MessagePrimitive.If";
 import { memo as memo2 } from "react";
 
 // src/context/providers/ContentPartProvider.tsx
-import { useEffect as useEffect9, useState as useState9 } from "react";
-import { create as create11 } from "zustand";
+import { useEffect as useEffect9, useState as useState10 } from "react";
+import { create as create12 } from "zustand";
 import { jsx as jsx18 } from "react/jsx-runtime";
 var COMPLETE_STATUS = {
   type: "complete"
@@ -2562,8 +2628,8 @@ var getContentPartState = ({ message }, useContentPart, partIndex) => {
 };
 var useContentPartContext2 = (partIndex) => {
   const { useMessage } = useMessageContext();
-  const [context] = useState9(() => {
-    const useContentPart = create11(
+  const [context] = useState10(() => {
+    const useContentPart = create12(
       () => getContentPartState(useMessage.getState(), void 0, partIndex)
     );
     return { useContentPart };
@@ -3070,15 +3136,15 @@ ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 import { memo as memo3 } from "react";
 
 // src/context/providers/MessageProvider.tsx
-import { useEffect as useEffect12, useState as useState10 } from "react";
-import { create as create14 } from "zustand";
+import { useEffect as useEffect12, useState as useState11 } from "react";
+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;
   },
@@ -3106,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 });
@@ -3141,8 +3207,8 @@ var getMessageState = (messages, getBranches, useMessage, messageIndex) => {
 };
 var useMessageContext2 = (messageIndex) => {
   const { useThreadMessages, useThreadActions } = useThreadContext();
-  const [context] = useState10(() => {
-    const useMessage = create14(
+  const [context] = useState11(() => {
+    const useMessage = create15(
       () => getMessageState(
         useThreadMessages.getState(),
         useThreadActions.getState().getBranches,
@@ -4030,6 +4096,7 @@ export {
   useContentPartDisplay,
   useContentPartImage,
   useContentPartText,
+  useDangerousInBrowserRuntime,
   useEdgeRuntime,
   useExternalStoreRuntime,
   useLocalRuntime,