@assistant-ui/react 0.5.5 → 0.5.6

package/dist/index.mjs

@@ -3357,6 +3357,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,
@@ -3630,6 +3631,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,
@@ -3642,6 +3644,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());
   }
@@ -3874,7 +3878,15 @@ import { forwardRef as forwardRef20 } from "react";
 import { forwardRef as forwardRef19 } from "react";
 import { ChevronLeftIcon, ChevronRightIcon } from "lucide-react";
 import { jsx as jsx31, jsxs as jsxs6 } from "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__ */ jsxs6(BranchPickerRoot, { hideWhenSingleBranch: true, children: [
     /* @__PURE__ */ jsx31(BranchPickerPrevious2, {}),
     /* @__PURE__ */ jsx31(BranchPickerState, {}),