@assistant-ui/react 0.11.39 → 0.11.41

package/package.json

@@ -28,7 +28,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.11.39",
+  "version": "0.11.41",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -48,8 +48,8 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-cloud": "^0.1.7",
-    "@assistant-ui/tap": "^0.2.1",
+    "assistant-cloud": "^0.1.8",
+    "@assistant-ui/tap": "^0.2.2",
     "@radix-ui/primitive": "^1.1.3",
     "@radix-ui/react-compose-refs": "^1.1.2",
     "@radix-ui/react-context": "^1.1.3",
@@ -59,7 +59,7 @@
     "@radix-ui/react-use-callback-ref": "^1.1.1",
     "@radix-ui/react-use-escape-keydown": "^1.1.1",
     "@standard-schema/spec": "^1.0.0",
-    "assistant-stream": "^0.2.39",
+    "assistant-stream": "^0.2.41",
     "json-schema": "^0.4.0",
     "nanoid": "5.1.6",
     "react-textarea-autosize": "^8.5.9",
@@ -88,7 +88,7 @@
     "eslint": "^9",
     "eslint-config-next": "16.0.3",
     "tsx": "^4.20.6",
-    "vitest": "^4.0.8",
+    "vitest": "^4.0.10",
     "@assistant-ui/x-buildutils": "0.0.1"
   },
   "publishConfig": {

package/src/legacy-runtime/runtime-cores/assistant-transport/runManager.ts

@@ -11,7 +11,7 @@ export function useRunManager(config: {
   onRun: (signal: AbortSignal) => Promise<void>;
   onFinish?: (() => void) | undefined;
   onCancel?: (() => void) | undefined;
-  onError?: ((error: Error) => void) | undefined;
+  onError?: ((error: Error) => void | Promise<void>) | undefined;
 }): RunManager {
   const [isRunning, setIsRunning] = useState(false);
   const stateRef = useRef({
@@ -37,7 +37,7 @@ export function useRunManager(config: {
         if (ac.signal.aborted) {
           onCancelRef.current?.();
         } else {
-          onErrorRef.current?.(error as Error);
+          await onErrorRef.current?.(error as Error);
         }
       } finally {
         onFinishRef.current?.();

package/src/legacy-runtime/runtime-cores/assistant-transport/types.ts

@@ -97,10 +97,17 @@ export type AssistantTransportOptions<T> = {
       commands: AssistantTransportCommand[];
       updateState: (updater: (state: T) => T) => void;
     },
-  ) => void;
+  ) => void | Promise<void>;
+  /**
+   * Called when commands are cancelled.
+   *
+   * When an error occurs, queued commands are automatically cancelled after `onError` settles.
+   * In this case, the `error` parameter contains the error that caused the cancellation.
+   */
   onCancel?: (params: {
     commands: AssistantTransportCommand[];
     updateState: (updater: (state: T) => T) => void;
+    error?: Error;
   }) => void;
   adapters?: {
     attachments?: AttachmentAdapter | undefined;

package/src/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.tsx

@@ -174,6 +174,9 @@ const useAssistantTransportThreadRuntime = <T,>(
         ...commandQueue.state.inTransit,
         ...commandQueue.state.queued,
       ];
+
+      commandQueue.reset();
+
       options.onCancel?.({
         commands: cmds,
         updateState: (updater) => {
@@ -181,19 +184,31 @@ const useAssistantTransportThreadRuntime = <T,>(
           rerender((prev) => prev + 1);
         },
       });
+    },
+    onError: async (error) => {
+      const inTransitCmds = [...commandQueue.state.inTransit];
+      const queuedCmds = [...commandQueue.state.queued];
 
       commandQueue.reset();
-    },
-    onError: (error) => {
-      const cmds = [...commandQueue.state.inTransit];
-      options.onError?.(error as Error, {
-        commands: cmds,
-        updateState: (updater) => {
-          agentStateRef.current = updater(agentStateRef.current);
-          rerender((prev) => prev + 1);
-        },
-      });
-      commandQueue.markDelivered();
+
+      try {
+        await options.onError?.(error as Error, {
+          commands: inTransitCmds,
+          updateState: (updater) => {
+            agentStateRef.current = updater(agentStateRef.current);
+            rerender((prev) => prev + 1);
+          },
+        });
+      } finally {
+        options.onCancel?.({
+          commands: queuedCmds,
+          updateState: (updater) => {
+            agentStateRef.current = updater(agentStateRef.current);
+            rerender((prev) => prev + 1);
+          },
+          error: error as Error,
+        });
+      }
     },
   });
 

package/src/primitives/messagePart/useMessagePartData.tsx

@@ -3,7 +3,6 @@
 import { useAssistantState } from "../../context";
 import { DataMessagePart } from "../../types";
 
-// eslint-disable-next-line @typescript-eslint/no-unused-vars
 export const useMessagePartData = <T = any,>(name?: string) => {
   const part = useAssistantState(({ part }) => {
     if (part.type !== "data") {