@assistant-ui/react 0.5.5 → 0.5.6

package/dist/index.d.mts

@@ -239,6 +239,7 @@ declare class LocalThreadRuntime implements ThreadRuntime {
     private abortController;
     private readonly repository;
     readonly capabilities: Readonly<{
+        switchToBranch: true;
         edit: true;
         reload: true;
         cancel: true;
@@ -641,6 +642,7 @@ declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
     private assistantOptimisticId;
     private useStore;
     get capabilities(): {
+        switchToBranch: boolean;
         edit: boolean;
         reload: boolean;
         cancel: boolean;
@@ -683,13 +685,15 @@ type AddToolResultOptions = {
     toolCallId: string;
     result: any;
 };
+type RuntimeCapabilities = {
+    switchToBranch: boolean;
+    edit: boolean;
+    reload: boolean;
+    cancel: boolean;
+    copy: boolean;
+};
 type ThreadActionsState = Readonly<{
-    capabilities: Readonly<{
-        edit: boolean;
-        reload: boolean;
-        cancel: boolean;
-        copy: boolean;
-    }>;
+    capabilities: Readonly<RuntimeCapabilities>;
     getBranches: (messageId: string) => readonly string[];
     switchToBranch: (branchId: string) => void;
     append: (message: AppendMessage) => void;
@@ -1219,6 +1223,9 @@ type AssistantMessageConfig = {
         Text?: TextContentPartComponent | undefined;
     } | undefined;
 };
+type BranchPickerConfig = {
+    allowBranchPicker?: boolean | undefined;
+};
 type StringsConfig = {
     assistantModal?: {
         open: {
@@ -1291,6 +1298,7 @@ type ThreadConfig = {
     welcome?: ThreadWelcomeConfig;
     assistantMessage?: AssistantMessageConfig;
     userMessage?: UserMessageConfig;
+    branchPicker?: BranchPickerConfig;
     strings?: StringsConfig;
 };
 declare const useThreadConfig: () => Omit<ThreadConfig, "runtime">;

package/dist/index.d.ts

@@ -239,6 +239,7 @@ declare class LocalThreadRuntime implements ThreadRuntime {
     private abortController;
     private readonly repository;
     readonly capabilities: Readonly<{
+        switchToBranch: true;
         edit: true;
         reload: true;
         cancel: true;
@@ -641,6 +642,7 @@ declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
     private assistantOptimisticId;
     private useStore;
     get capabilities(): {
+        switchToBranch: boolean;
         edit: boolean;
         reload: boolean;
         cancel: boolean;
@@ -683,13 +685,15 @@ type AddToolResultOptions = {
     toolCallId: string;
     result: any;
 };
+type RuntimeCapabilities = {
+    switchToBranch: boolean;
+    edit: boolean;
+    reload: boolean;
+    cancel: boolean;
+    copy: boolean;
+};
 type ThreadActionsState = Readonly<{
-    capabilities: Readonly<{
-        edit: boolean;
-        reload: boolean;
-        cancel: boolean;
-        copy: boolean;
-    }>;
+    capabilities: Readonly<RuntimeCapabilities>;
     getBranches: (messageId: string) => readonly string[];
     switchToBranch: (branchId: string) => void;
     append: (message: AppendMessage) => void;
@@ -1219,6 +1223,9 @@ type AssistantMessageConfig = {
         Text?: TextContentPartComponent | undefined;
     } | undefined;
 };
+type BranchPickerConfig = {
+    allowBranchPicker?: boolean | undefined;
+};
 type StringsConfig = {
     assistantModal?: {
         open: {
@@ -1291,6 +1298,7 @@ type ThreadConfig = {
     welcome?: ThreadWelcomeConfig;
     assistantMessage?: AssistantMessageConfig;
     userMessage?: UserMessageConfig;
+    branchPicker?: BranchPickerConfig;
     strings?: StringsConfig;
 };
 declare const useThreadConfig: () => Omit<ThreadConfig, "runtime">;

package/dist/index.js

@@ -3440,6 +3440,7 @@ var shouldContinue = (result) => result.status?.type === "requires-action" && re
 
 // src/runtimes/local/LocalThreadRuntime.tsx
 var CAPABILITIES = Object.freeze({
+  switchToBranch: true,
   edit: true,
   reload: true,
   cancel: true,
@@ -3713,6 +3714,7 @@ var ExternalStoreThreadRuntime = class {
   useStore;
   get capabilities() {
     return {
+      switchToBranch: this.store.setMessages !== void 0,
       edit: this.store.onEdit !== void 0,
       reload: this.store.onReload !== void 0,
       cancel: this.store.onCancel !== void 0,
@@ -3725,6 +3727,8 @@ var ExternalStoreThreadRuntime = class {
     return this.repository.getBranches(messageId);
   }
   switchToBranch(branchId) {
+    if (!this.store.setMessages)
+      throw new Error("Runtime does not support switching branches.");
     this.repository.switchToBranch(branchId);
     this.updateMessages(this.repository.getMessages());
   }
@@ -3954,7 +3958,15 @@ var import_react60 = require("react");
 var import_react59 = require("react");
 var import_lucide_react2 = require("lucide-react");
 var import_jsx_runtime31 = require("react/jsx-runtime");
+var useAllowBranchPicker = () => {
+  const { branchPicker: { allowBranchPicker = true } = {} } = useThreadConfig();
+  const { useThreadActions } = useThreadContext();
+  const branchPickerSupported = useThreadActions((t) => t.capabilities.edit);
+  return branchPickerSupported && allowBranchPicker;
+};
 var BranchPicker = () => {
+  const allowBranchPicker = useAllowBranchPicker();
+  if (!allowBranchPicker) return null;
   return /* @__PURE__ */ (0, import_jsx_runtime31.jsxs)(BranchPickerRoot, { hideWhenSingleBranch: true, children: [
     /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(BranchPickerPrevious2, {}),
     /* @__PURE__ */ (0, import_jsx_runtime31.jsx)(BranchPickerState, {}),