@assistant-ui/react 0.5.85 → 0.5.87

package/dist/index.mjs

@@ -10,7 +10,7 @@ import {
   toLanguageModelMessages,
   toLanguageModelTools,
   toolResultStream
-} from "./chunk-TWIMAOZR.mjs";
+} from "./chunk-HLFNKW7L.mjs";
 import {
   __export
 } from "./chunk-BJPOCE4O.mjs";
@@ -437,6 +437,62 @@ function useComposerRuntime(options) {
   return messageRuntime ? messageRuntime.composer : threadRuntime?.composer ?? null;
 }
 
+// src/context/react/AttachmentContext.ts
+import { createContext as createContext5, useContext as useContext2 } from "react";
+var AttachmentContext = createContext5(
+  null
+);
+function useAttachmentContext(options) {
+  const context = useContext2(AttachmentContext);
+  if (!options?.optional && !context)
+    throw new Error(
+      "This component must be used within a ComposerPrimitive.Attachments or MessagePrimitive.Attachments component."
+    );
+  return context;
+}
+function useThreadComposerAttachmentContext(options) {
+  const context = useAttachmentContext(options);
+  if (!context) return null;
+  if (context.source !== "thread-composer")
+    throw new Error(
+      "This component must be used within a thread's ComposerPrimitive.Attachments component."
+    );
+  return context;
+}
+function useEditComposerAttachmentContext(options) {
+  const context = useAttachmentContext(options);
+  if (!context) return null;
+  if (context.source !== "edit-composer")
+    throw new Error(
+      "This component must be used within a messages's ComposerPrimitive.Attachments component."
+    );
+  return context;
+}
+function useMessageAttachmentContext(options) {
+  const context = useAttachmentContext(options);
+  if (!context) return null;
+  if (context.source !== "message")
+    throw new Error(
+      "This component must be used within a MessagePrimitive.Attachments component."
+    );
+  return context;
+}
+function useAttachmentRuntime(options) {
+  const attachmentRuntime = useAttachmentContext(options);
+  if (!attachmentRuntime) return null;
+  return attachmentRuntime.useAttachmentRuntime();
+}
+var { useAttachment } = createContextStoreHook(
+  useAttachmentContext,
+  "useAttachment"
+);
+var { useAttachment: useThreadComposerAttachment } = createContextStoreHook(useThreadComposerAttachmentContext, "useAttachment");
+var { useAttachment: useEditComposerAttachment } = createContextStoreHook(useEditComposerAttachmentContext, "useAttachment");
+var { useAttachment: useMessageAttachment } = createContextStoreHook(
+  useMessageAttachmentContext,
+  "useAttachment"
+);
+
 // src/hooks/useAppendMessage.tsx
 import { useCallback } from "react";
 var useAppendMessage = () => {
@@ -1244,64 +1300,6 @@ AttachmentPrimitiveRoot.displayName = "AttachmentPrimitive.Root";
 
 // src/primitives/attachment/AttachmentThumb.tsx
 import { forwardRef as forwardRef11 } from "react";
-
-// src/context/react/AttachmentContext.ts
-import { createContext as createContext5, useContext as useContext2 } from "react";
-var AttachmentContext = createContext5(
-  null
-);
-function useAttachmentContext(options) {
-  const context = useContext2(AttachmentContext);
-  if (!options?.optional && !context)
-    throw new Error(
-      "This component must be used within a ComposerPrimitive.Attachments or MessagePrimitive.Attachments component."
-    );
-  return context;
-}
-function useThreadComposerAttachmentContext(options) {
-  const context = useAttachmentContext(options);
-  if (!context) return null;
-  if (context.source !== "thread-composer")
-    throw new Error(
-      "This component must be used within a thread's ComposerPrimitive.Attachments component."
-    );
-  return context;
-}
-function useEditComposerAttachmentContext(options) {
-  const context = useAttachmentContext(options);
-  if (!context) return null;
-  if (context.source !== "edit-composer")
-    throw new Error(
-      "This component must be used within a messages's ComposerPrimitive.Attachments component."
-    );
-  return context;
-}
-function useMessageAttachmentContext(options) {
-  const context = useAttachmentContext(options);
-  if (!context) return null;
-  if (context.source !== "message")
-    throw new Error(
-      "This component must be used within a MessagePrimitive.Attachments component."
-    );
-  return context;
-}
-function useAttachmentRuntime(options) {
-  const attachmentRuntime = useAttachmentContext(options);
-  if (!attachmentRuntime) return null;
-  return attachmentRuntime.useAttachmentRuntime();
-}
-var { useAttachment } = createContextStoreHook(
-  useAttachmentContext,
-  "useAttachment"
-);
-var { useAttachment: useThreadComposerAttachment } = createContextStoreHook(useThreadComposerAttachmentContext, "useAttachment");
-var { useAttachment: useEditComposerAttachment } = createContextStoreHook(useEditComposerAttachmentContext, "useAttachment");
-var { useAttachment: useMessageAttachment } = createContextStoreHook(
-  useMessageAttachmentContext,
-  "useAttachment"
-);
-
-// src/primitives/attachment/AttachmentThumb.tsx
 import { Primitive as Primitive8 } from "@radix-ui/react-primitive";
 import { jsxs } from "react/jsx-runtime";
 var AttachmentPrimitiveThumb = forwardRef11((props, ref) => {
@@ -3061,7 +3059,7 @@ var subscribeToMainThread = (runtime, callback) => {
 };
 
 // src/runtimes/local/useLocalRuntime.tsx
-import { useEffect as useEffect17, useMemo as useMemo11, useState as useState13 } from "react";
+import { useEffect as useEffect17, useMemo as useMemo11, useState as useState12 } from "react";
 
 // src/utils/ProxyConfigProvider.ts
 var ProxyConfigProvider = class {
@@ -4101,9 +4099,6 @@ var streamUtils = {
   streamPartDecoderStream
 };
 
-// src/runtimes/edge/useEdgeRuntime.ts
-import { useState as useState12 } from "react";
-
 // src/runtimes/edge/streams/assistantDecoderStream.ts
 function assistantDecoderStream() {
   const toolCallNames = /* @__PURE__ */ new Map();
@@ -4182,6 +4177,7 @@ function assistantDecoderStream() {
         }
         case "9" /* ToolCall */: {
           const { toolCallId, toolName, args } = value;
+          toolCallNames.set(toolCallId, toolName);
           const argsText = JSON.stringify(args);
           controller.enqueue({
             type: "tool-call-delta",
@@ -4283,8 +4279,10 @@ var splitLocalRuntimeOptions = (options) => {
 // src/runtimes/edge/useEdgeRuntime.ts
 var useEdgeRuntime = (options) => {
   const { localRuntimeOptions, otherOptions } = splitLocalRuntimeOptions(options);
-  const [adapter] = useState12(() => new EdgeChatAdapter(otherOptions));
-  return useLocalRuntime(adapter, localRuntimeOptions);
+  return useLocalRuntime(
+    new EdgeChatAdapter(otherOptions),
+    localRuntimeOptions
+  );
 };
 
 // src/runtimes/local/shouldContinue.tsx
@@ -4834,7 +4832,7 @@ var useLocalRuntime = (adapter, options = {}) => {
       chatModel: adapter
     }
   };
-  const [runtime] = useState13(() => new LocalRuntimeCore(opt));
+  const [runtime] = useState12(() => new LocalRuntimeCore(opt));
   useEffect17(() => {
     runtime.setOptions(opt);
   });
@@ -4842,7 +4840,7 @@ var useLocalRuntime = (adapter, options = {}) => {
 };
 
 // src/runtimes/external-store/useExternalStoreRuntime.tsx
-import { useEffect as useEffect18, useMemo as useMemo12, useState as useState14 } from "react";
+import { useEffect as useEffect18, useMemo as useMemo12, useState as useState13 } from "react";
 
 // src/runtimes/external-store/ExternalStoreThreadManagementAdapter.tsx
 var EMPTY_ARRAY2 = Object.freeze([]);
@@ -5228,7 +5226,7 @@ var ExternalStoreRuntimeCore = class extends BaseAssistantRuntimeCore {
 
 // src/runtimes/external-store/useExternalStoreRuntime.tsx
 var useExternalStoreRuntime = (store) => {
-  const [runtime] = useState14(() => new ExternalStoreRuntimeCore(store));
+  const [runtime] = useState13(() => new ExternalStoreRuntimeCore(store));
   useEffect18(() => {
     runtime.setStore(store);
   });
@@ -5385,7 +5383,7 @@ var shallowArrayEqual = (a, b) => {
 };
 
 // src/runtimes/dangerous-in-browser/useDangerousInBrowserRuntime.ts
-import { useState as useState15 } from "react";
+import { useState as useState14 } from "react";
 
 // src/runtimes/dangerous-in-browser/DangerousInBrowserAdapter.ts
 var DangerousInBrowserAdapter = class {
@@ -5414,7 +5412,7 @@ var DangerousInBrowserAdapter = class {
 // src/runtimes/dangerous-in-browser/useDangerousInBrowserRuntime.ts
 var useDangerousInBrowserRuntime = (options) => {
   const { localRuntimeOptions, otherOptions } = splitLocalRuntimeOptions(options);
-  const [adapter] = useState15(() => new DangerousInBrowserAdapter(otherOptions));
+  const [adapter] = useState14(() => new DangerousInBrowserAdapter(otherOptions));
   return useLocalRuntime(adapter, localRuntimeOptions);
 };
 
@@ -6001,7 +5999,7 @@ CircleStopIcon.displayName = "CircleStopIcon";
 import {
   forwardRef as forwardRef28,
   useEffect as useEffect19,
-  useState as useState16
+  useState as useState15
 } from "react";
 import { CircleXIcon, FileIcon } from "lucide-react";
 
@@ -6044,7 +6042,7 @@ var AttachmentRoot = withDefaults(attachment_exports.Root, {
 });
 AttachmentRoot.displayName = "AttachmentRoot";
 var useFileSrc = (file) => {
-  const [src, setSrc] = useState16(void 0);
+  const [src, setSrc] = useState15(void 0);
   useEffect19(() => {
     if (!file) {
       setSrc(void 0);
@@ -6069,7 +6067,7 @@ var useAttachmentSrc = () => {
   return useFileSrc(file) ?? src;
 };
 var AttachmentPreview = ({ src }) => {
-  const [isLoaded, setIsLoaded] = useState16(false);
+  const [isLoaded, setIsLoaded] = useState15(false);
   return (
     // eslint-disable-next-line @next/next/no-img-element
     /* @__PURE__ */ jsx46(
@@ -6730,6 +6728,8 @@ export {
   useAssistantRuntimeStore,
   useAssistantTool,
   useAssistantToolUI,
+  useAttachment,
+  useAttachmentRuntime,
   useBranchPickerCount,
   useBranchPickerNext,
   useBranchPickerNumber,