@assistant-ui/react 0.5.34 → 0.5.36

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-QBSJKWMT.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";
@@ -1614,13 +1615,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);
@@ -1693,7 +1730,7 @@ var AssistantProvider = ({ children, runtime }) => {
   useInsertionEffect3(() => {
     runtimeRef.current = runtime;
   });
-  const [context] = useState7(() => {
+  const [context] = useState8(() => {
     const useModelConfig = makeAssistantModelConfigStore();
     const useToolUIs = makeAssistantToolUIsStore();
     const useAssistantActions = makeAssistantActionsStore(runtimeRef);
@@ -2285,7 +2322,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 +2346,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,7 +2560,7 @@ 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 { useEffect as useEffect9, useState as useState10 } from "react";
 import { create as create11 } from "zustand";
 import { jsx as jsx18 } from "react/jsx-runtime";
 var COMPLETE_STATUS = {
@@ -2562,7 +2599,7 @@ var getContentPartState = ({ message }, useContentPart, partIndex) => {
 };
 var useContentPartContext2 = (partIndex) => {
   const { useMessage } = useMessageContext();
-  const [context] = useState9(() => {
+  const [context] = useState10(() => {
     const useContentPart = create11(
       () => getContentPartState(useMessage.getState(), void 0, partIndex)
     );
@@ -3070,7 +3107,7 @@ 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 { useEffect as useEffect12, useState as useState11 } from "react";
 import { create as create14 } from "zustand";
 
 // src/context/stores/EditComposer.ts
@@ -3141,7 +3178,7 @@ var getMessageState = (messages, getBranches, useMessage, messageIndex) => {
 };
 var useMessageContext2 = (messageIndex) => {
   const { useThreadMessages, useThreadActions } = useThreadContext();
-  const [context] = useState10(() => {
+  const [context] = useState11(() => {
     const useMessage = create14(
       () => getMessageState(
         useThreadMessages.getState(),
@@ -4030,6 +4067,7 @@ export {
   useContentPartDisplay,
   useContentPartImage,
   useContentPartText,
+  useDangerousInBrowserRuntime,
   useEdgeRuntime,
   useExternalStoreRuntime,
   useLocalRuntime,