@assistant-ui/react 0.11.19 → 0.11.21

package/package.json

@@ -28,7 +28,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.11.19",
+  "version": "0.11.21",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -61,10 +61,10 @@
     "@standard-schema/spec": "^1.0.0",
     "assistant-stream": "^0.2.29",
     "json-schema": "^0.4.0",
-    "nanoid": "5.1.5",
+    "nanoid": "5.1.6",
     "react-textarea-autosize": "^8.5.9",
-    "zod": "^4.0.17",
-    "zustand": "^5.0.7"
+    "zod": "^4.1.11",
+    "zustand": "^5.0.8"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -81,13 +81,13 @@
     }
   },
   "devDependencies": {
-    "@stryker-mutator/core": "^9.0.1",
-    "@stryker-mutator/vitest-runner": "^9.0.1",
+    "@stryker-mutator/core": "^9.1.1",
+    "@stryker-mutator/vitest-runner": "^9.1.1",
     "@types/json-schema": "^7.0.15",
-    "@types/node": "^24.3.0",
+    "@types/node": "^24.5.2",
     "eslint": "^9",
-    "eslint-config-next": "15.4.6",
-    "tsx": "^4.20.4",
+    "eslint-config-next": "15.5.4",
+    "tsx": "^4.20.6",
     "vitest": "^3.2.4",
     "@assistant-ui/x-buildutils": "0.0.1"
   },

package/src/augmentations.ts

@@ -0,0 +1,26 @@
+declare global {
+  interface Assistant {
+    Commands: unknown;
+  }
+}
+
+type GetAugmentation<
+  Key extends keyof Assistant,
+  ExpectedType,
+  FallbackType = ExpectedType,
+> = unknown extends Assistant[Key]
+  ? FallbackType
+  : Assistant[Key] extends ExpectedType
+    ? Assistant[Key]
+    : {
+        ErrorMessage: `There is an error in the type you provided for Assistant.${Key}`;
+      };
+
+type UserCommandsRecord = GetAugmentation<
+  "Commands",
+  Record<string, unknown>,
+  Record<string, never>
+>;
+
+export type UserCommands =
+  UserCommandsRecord extends Record<string, infer V> ? V : never;

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

@@ -1,4 +1,7 @@
-export { useAssistantTransportRuntime } from "./useAssistantTransportRuntime";
+export {
+  useAssistantTransportRuntime,
+  useAssistantTransportSendCommand,
+} from "./useAssistantTransportRuntime";
 export type {
   AssistantTransportConnectionMetadata,
   AssistantTransportCommand,

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

@@ -1,6 +1,7 @@
 import { ReadonlyJSONValue } from "assistant-stream/utils";
 import { ThreadMessage } from "../../../types";
 import { AttachmentAdapter, ThreadHistoryAdapter } from "..";
+import { UserCommands } from "../../../augmentations";
 
 // Message part types
 export type TextPart = {
@@ -42,7 +43,8 @@ export type AddToolResultCommand = {
 
 export type AssistantTransportCommand =
   | AddMessageCommand
-  | AddToolResultCommand;
+  | AddToolResultCommand
+  | UserCommands;
 
 // State types
 export type AssistantTransportState = {

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

@@ -21,6 +21,7 @@ import {
   AddToolResultCommand,
   UserMessagePart,
   QueuedCommand,
+  AssistantTransportCommand,
 } from "./types";
 import { useCommandQueue } from "./commandQueue";
 import { useRunManager } from "./runManager";
@@ -29,6 +30,38 @@ import { useToolInvocations } from "./useToolInvocations";
 import { toAISDKTools, getEnabledTools, createRequestHeaders } from "./utils";
 import { useRemoteThreadListRuntime } from "../remote-thread-list/useRemoteThreadListRuntime";
 import { InMemoryThreadListAdapter } from "../remote-thread-list/adapter/in-memory";
+import { useAssistantApi } from "../../../context/react";
+
+const symbolAssistantTransportExtras = Symbol("assistant-transport-extras");
+type AssistantTransportExtras = {
+  [symbolAssistantTransportExtras]: true;
+  sendCommand: (command: AssistantTransportCommand) => void;
+};
+
+const asAssistantTransportExtras = (
+  extras: unknown,
+): AssistantTransportExtras => {
+  if (
+    typeof extras !== "object" ||
+    extras == null ||
+    !(symbolAssistantTransportExtras in extras)
+  )
+    throw new Error(
+      "This method can only be called when you are using useAssistantTransportRuntime",
+    );
+
+  return extras as AssistantTransportExtras;
+};
+
+export const useAssistantTransportSendCommand = () => {
+  const api = useAssistantApi();
+
+  return (command: AssistantTransportCommand) => {
+    const extras = api.thread().getState().extras;
+    const transportExtras = asAssistantTransportExtras(extras);
+    transportExtras.sendCommand(command);
+  };
+};
 
 const useAssistantTransportThreadRuntime = <T,>(
   options: AssistantTransportOptions<T>,
@@ -162,6 +195,12 @@ const useAssistantTransportThreadRuntime = <T,>(
     state: converted.state,
     isRunning: converted.isRunning,
     adapters: options.adapters,
+    extras: {
+      [symbolAssistantTransportExtras]: true,
+      sendCommand: (command: AssistantTransportCommand) => {
+        commandQueue.enqueue(command);
+      },
+    } satisfies AssistantTransportExtras,
     onNew: async (message: AppendMessage): Promise<void> => {
       if (message.role !== "user")
         throw new Error("Only user messages are supported");

package/src/legacy-runtime/runtime-cores/core/ThreadRuntimeCore.tsx

@@ -14,6 +14,7 @@ import {
 
 export type RuntimeCapabilities = {
   readonly switchToBranch: boolean;
+  readonly switchBranchDuringRun: boolean;
   readonly edit: boolean;
   readonly reload: boolean;
   readonly cancel: boolean;

package/src/legacy-runtime/runtime-cores/external-store/ExternalStoreThreadRuntimeCore.tsx

@@ -43,6 +43,7 @@ export class ExternalStoreThreadRuntimeCore
 
   private _capabilities: RuntimeCapabilities = {
     switchToBranch: false,
+    switchBranchDuringRun: false,
     edit: false,
     reload: false,
     cancel: false,
@@ -108,6 +109,7 @@ export class ExternalStoreThreadRuntimeCore
     this.suggestions = store.suggestions ?? EMPTY_ARRAY;
     this._capabilities = {
       switchToBranch: this._store.setMessages !== undefined,
+      switchBranchDuringRun: false, // External store never supports branch switching during run
       edit: this._store.onEdit !== undefined,
       reload: this._store.onReload !== undefined,
       cancel: this._store.onCancel !== undefined,
@@ -228,6 +230,11 @@ export class ExternalStoreThreadRuntimeCore
     if (!this._store.setMessages)
       throw new Error("Runtime does not support switching branches.");
 
+    // Silently ignore branch switches while running
+    if (this._store.isRunning) {
+      return;
+    }
+
     this.repository.switchToBranch(branchId);
     this.updateMessages(this.repository.getMessages());
   }

package/src/legacy-runtime/runtime-cores/local/LocalThreadRuntimeCore.tsx

@@ -30,6 +30,7 @@ export class LocalThreadRuntimeCore
 {
   public readonly capabilities = {
     switchToBranch: true,
+    switchBranchDuringRun: true,
     edit: true,
     reload: true,
     cancel: true,

package/src/legacy-runtime/runtime-cores/remote-thread-list/EMPTY_THREAD_CORE.tsx

@@ -132,6 +132,7 @@ export const EMPTY_THREAD_CORE: ThreadRuntimeCore = {
 
   capabilities: {
     switchToBranch: false,
+    switchBranchDuringRun: false,
     edit: false,
     reload: false,
     cancel: false,

package/src/legacy-runtime/runtime-cores/remote-thread-list/RemoteThreadListHookInstanceManager.tsx

@@ -90,10 +90,7 @@ export class RemoteThreadListHookInstanceManager extends BaseSubscribable {
         throw new Error("Thread not found. This is a bug in assistant-ui.");
 
       aliveThread.runtime = threadBinding.getState();
-
-      if (isMounted) {
-        this._notifySubscribers();
-      }
+      this._notifySubscribers();
     }, [threadId, threadBinding]);
 
     const isMounted = useRef(false);

package/src/legacy-runtime/runtime-cores/remote-thread-list/adapter/cloud.tsx

@@ -16,14 +16,14 @@ import { InMemoryThreadListAdapter } from "./in-memory";
 import { CloudFileAttachmentAdapter } from "../../adapters";
 
 type ThreadData = {
-  externalId: string;
+  externalId: string | undefined;
 };
 
 type CloudThreadListAdapterOptions = {
   cloud?: AssistantCloud | undefined;
 
-  create?(): Promise<ThreadData>;
-  delete?(threadId: string): Promise<void>;
+  create?: (() => Promise<ThreadData>) | undefined;
+  delete?: ((threadId: string) => Promise<void>) | undefined;
 };
 
 const baseUrl =

package/src/legacy-runtime/runtime-cores/remote-thread-list/useRemoteThreadListRuntime.tsx

@@ -6,6 +6,7 @@ import { RemoteThreadListThreadListRuntimeCore } from "./RemoteThreadListThreadL
 import { RemoteThreadListOptions } from "./types";
 import { AssistantRuntimeImpl } from "../../../internal";
 import { AssistantRuntimeCore } from "../core/AssistantRuntimeCore";
+import { AssistantRuntime } from "../../runtime/AssistantRuntime";
 
 class RemoteThreadListRuntimeCore
   extends BaseAssistantRuntimeCore
@@ -28,7 +29,7 @@ class RemoteThreadListRuntimeCore
 
 export const useRemoteThreadListRuntime = (
   options: RemoteThreadListOptions,
-) => {
+): AssistantRuntime => {
   const [runtime] = useState(() => new RemoteThreadListRuntimeCore(options));
   useEffect(() => {
     runtime.threads.__internal_setOptions(options);

package/src/primitives/branchPicker/BranchPickerNext.tsx

@@ -10,9 +10,17 @@ import { useAssistantState, useAssistantApi } from "../../context";
 
 const useBranchPickerNext = () => {
   const api = useAssistantApi();
-  const disabled = useAssistantState(
-    ({ message }) => message.branchNumber >= message.branchCount,
-  );
+  const disabled = useAssistantState(({ thread, message }) => {
+    // Disabled if no next branch
+    if (message.branchNumber >= message.branchCount) return true;
+
+    // Disabled if running and capability not supported
+    if (thread.isRunning && !thread.capabilities.switchBranchDuringRun) {
+      return true;
+    }
+
+    return false;
+  });
 
   const callback = useCallback(() => {
     api.message().switchToBranch({ position: "next" });

package/src/primitives/branchPicker/BranchPickerPrevious.tsx

@@ -31,9 +31,17 @@ import { useAssistantState, useAssistantApi } from "../../context";
  */
 const useBranchPickerPrevious = () => {
   const api = useAssistantApi();
-  const disabled = useAssistantState(
-    ({ message }) => message.branchNumber <= 1,
-  );
+  const disabled = useAssistantState(({ thread, message }) => {
+    // Disabled if no previous branch
+    if (message.branchNumber <= 1) return true;
+
+    // Disabled if running and capability not supported
+    if (thread.isRunning && !thread.capabilities.switchBranchDuringRun) {
+      return true;
+    }
+
+    return false;
+  });
 
   const callback = useCallback(() => {
     api.message().switchToBranch({ position: "previous" });