@assistant-ui/react 0.5.4 → 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());
   }
@@ -3769,7 +3773,10 @@ var useExternalStoreRuntime = (store) => {
 };
 
 // src/ui/thread-config.tsx
-import { createContext as createContext5, useContext as useContext5 } from "react";
+import {
+  createContext as createContext5,
+  useContext as useContext5
+} from "react";
 import { Fragment as Fragment3, jsx as jsx29 } from "react/jsx-runtime";
 var ThreadConfigContext = createContext5({});
 var useThreadConfig = () => {
@@ -3871,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, {}),
@@ -4154,10 +4169,10 @@ var ThreadWelcomeSuggestion = ({
   return /* @__PURE__ */ jsx37(
     ThreadWelcomeSuggestionStyled,
     {
-      prompt: prompt ?? text,
+      prompt,
       method: "replace",
       autoSend: true,
-      children: /* @__PURE__ */ jsx37("span", { className: "aui-thread-welcome-suggestion-text", children: text })
+      children: /* @__PURE__ */ jsx37("span", { className: "aui-thread-welcome-suggestion-text", children: text ?? prompt })
     }
   );
 };