@brainfish-ai/components 0.13.15 → 0.14.2

package/dist/components/chat-search/PrimarySearch.d.ts

@@ -1,5 +1,5 @@
 import { default as React } from 'react';
-import { Suggestion } from './types';
+import { Collection, Suggestion } from './types';
 export declare const PrimarySearch: React.ForwardRefExoticComponent<{
     query: string;
     onQueryChange: (query: string) => void;
@@ -9,12 +9,18 @@ export declare const PrimarySearch: React.ForwardRefExoticComponent<{
         attachImageText?: string;
         allDocumentsText?: string;
     };
-    collections?: {
-        id: string;
-        name: string;
-    }[] | undefined;
+    collections?: Collection[] | undefined;
     currentCollectionId?: string | undefined;
     onCollectionChange?: ((collectionId: string) => void) | undefined;
+    allowedRegions?: string[] | undefined;
+    currentRegion?: string | undefined;
+    onRegionChange?: ((region: string) => void) | undefined;
+    apiKey?: string | undefined;
+    fetchRegionConfig?: ((apiKey: string, region: string) => Promise<{
+        settings?: {
+            collections?: Collection[];
+        };
+    } | null>) | undefined;
     disableImageAttachment?: boolean | undefined;
     attachedImages?: {
         file: File;

package/dist/components/chat-search/types.d.ts

@@ -150,6 +150,14 @@ export interface ChatSearchProps {
     collections?: Array<Collection>;
     selectedCollectionId?: string;
     onCollectionChange?: (collectionId: string) => void;
+    selectedRegion?: string;
+    onRegionChange?: (region: string) => void;
+    apiKey?: string;
+    fetchRegionConfig?: (apiKey: string, region: string) => Promise<{
+        settings?: {
+            collections?: Collection[];
+        };
+    } | null>;
     searchEndpoint: string;
     answerEndpoint: string;
     answerStreamEndpoint: string;

package/dist/components/combobox/combobox.d.ts

@@ -1,14 +1,18 @@
 import * as React from 'react';
+export interface ComboboxItem {
+    value: string;
+    label: string;
+    icon?: React.ReactNode;
+}
 export interface ComboboxProps {
-    options: {
-        value: string;
-        label: string;
-    }[];
+    options: ComboboxItem[];
     defaultValueLabel: string;
     placeholder: string;
     noResultsLabel: string;
     className?: string;
+    dropdownWidth?: number;
+    value?: string;
     onOpenChange?: (open: boolean) => void;
     onChange?: (value: string) => void;
 }
-export declare function Combobox({ options, defaultValueLabel, placeholder, noResultsLabel, className, onOpenChange, onChange, }: ComboboxProps): React.JSX.Element;
+export declare function Combobox({ options, defaultValueLabel, placeholder, noResultsLabel, className, onOpenChange, onChange, dropdownWidth, value: controlledValue, }: ComboboxProps): React.JSX.Element;

package/dist/components/two-level-combobox/index.d.ts

@@ -0,0 +1 @@
+export * from './two-level-combobox';

package/dist/components/two-level-combobox/two-level-combobox.d.ts

@@ -0,0 +1,15 @@
+import { ComboboxItem } from '../combobox';
+import * as React from 'react';
+export interface TwoLevelComboboxProps {
+    firstLevelItems: ComboboxItem[];
+    fetchSecondLevelItems: (firstLevelId: string) => Promise<ComboboxItem[]>;
+    onSelectionChange?: (firstLevel: ComboboxItem | null, secondLevel: ComboboxItem | null) => void;
+    firstLevelLabel?: string;
+    secondLevelLabel?: string;
+    firstLevelPlaceholder?: string;
+    secondLevelPlaceholder?: string;
+    className?: string;
+    debugMode?: boolean;
+    displayLabel?: boolean;
+}
+export declare function TwoLevelCombobox({ firstLevelItems, fetchSecondLevelItems, onSelectionChange, firstLevelLabel, secondLevelLabel, firstLevelPlaceholder, className, debugMode, displayLabel, }: TwoLevelComboboxProps): React.JSX.Element;

package/dist/components/ui/div-button.d.ts

@@ -0,0 +1,11 @@
+import { VariantProps } from 'class-variance-authority';
+import * as React from "react";
+declare const buttonVariants: (props?: ({
+    variant?: "link" | "dark" | "default" | "destructive" | "outline" | "secondary" | "ghost" | null | undefined;
+    size?: "default" | "icon" | "sm" | "lg" | null | undefined;
+} & import('class-variance-authority/dist/types').ClassProp) | undefined) => string;
+export interface DivButtonProps extends React.HTMLAttributes<HTMLDivElement>, VariantProps<typeof buttonVariants> {
+    asChild?: boolean;
+}
+declare const DivButton: React.ForwardRefExoticComponent<DivButtonProps & React.RefAttributes<HTMLDivElement>>;
+export { DivButton, buttonVariants };

package/dist/esm/chunks/{ChatSearch.BdkUMxIG.js → ChatSearch.BVivOuWt.js}

@@ -2,7 +2,7 @@ import React__default, { useState, useCallback, useMemo, createContext, useConte
 import { motion, AnimatePresence } from 'framer-motion';
 import { TooltipProvider, Tooltip, TooltipTrigger, TooltipContent } from '../components/ui/tooltip.js';
 import { Button } from '../components/ui/button.js';
-import { XCircle, Check, Copy, CaretDown, ArrowSquareOut, CaretRight, ArrowsVertical, ArrowDown, MagnifyingGlass, X, CheckCircle, Circle, Image, ArrowRight, ArrowLeft, ArrowsInSimple, ArrowsOutSimple } from '@phosphor-icons/react';
+import { XCircle, Check, Copy, CaretDown, ArrowSquareOut, CaretRight, ArrowsVertical, ArrowDown, Globe, LockSimple, MagnifyingGlass, X, Image, ArrowRight, ArrowLeft, ArrowsInSimple, ArrowsOutSimple } from '@phosphor-icons/react';
 import { g as getDefaultExportFromCjs, F as FormattedMessage, c as addPopupWidgetUtm } from './FormattedMessage.C7CKFL5X.js';
 import { appendErrors, useForm, Controller } from 'react-hook-form';
 import { validateFieldsNatively, toNestErrors } from '@hookform/resolvers';
@@ -20,7 +20,9 @@ import { MemoizedIcon } from '../components/ui/icon.js';
 import { Collapsible, CollapsibleTrigger, CollapsibleContent } from '../components/ui/collapsible.js';
 import { F as Feedback } from './feedback.NWn6_mYe.js';
 import { G as GeneratingStar } from './generating-star.B0RUXRff.js';
-import { DropdownMenu, DropdownMenuTrigger, DropdownMenuContent, DropdownMenuItem } from '../components/ui/dropdown-menu.js';
+import { getCountry } from 'countries-and-timezones';
+import { C as Combobox } from './combobox.BRm6ZDEB.js';
+import { T as TwoLevelCombobox } from './two-level-combobox.DFonmrmc.js';
 import { ScrollArea } from '../components/ui/scroll-area.js';
 
 import '../ChatSearch.css';function Suggestions({ suggestions, onQuestionClick, title = "Suggested questions" }) {
@@ -4252,6 +4254,7 @@ async function searchApi({
   secretAttributes,
   allowedRegions
 }) {
+  const isAgentAssist = !!headers?.["x-brainfish-agent-assist"] && headers?.["x-brainfish-agent-assist"] !== "false";
   const payload = await makeRequest(endpoint, {
     method: "POST",
     headers,
@@ -4263,7 +4266,8 @@ async function searchApi({
       ...conversationId && { conversationId },
       ...attributes && { attributes },
       ...secretAttributes && { secretAttributes },
-      ...allowedRegions && { allowedRegions }
+      ...allowedRegions && { allowedRegions },
+      ...isAgentAssist && { source: "agent_assist" }
     }
   });
   return {
@@ -4747,6 +4751,10 @@ const PrimarySearch = React__default.forwardRef(
     collections = [],
     currentCollectionId,
     onCollectionChange,
+    allowedRegions = [],
+    onRegionChange,
+    apiKey,
+    fetchRegionConfig,
     disableImageAttachment = false,
     attachedImages = [],
     onImageUpload,
@@ -4757,6 +4765,65 @@ const PrimarySearch = React__default.forwardRef(
   }, ref) => {
     const fileInputRef = useRef(null);
     const [showSuggestions, setShowSuggestions] = useState(false);
+    const regionItems = [
+      {
+        value: "all",
+        label: "All regions",
+        icon: /* @__PURE__ */ React__default.createElement(Globe, { className: "size-4" })
+      },
+      ...allowedRegions.map((region) => {
+        const country = getCountry(region);
+        return {
+          value: region,
+          label: country?.name || region,
+          icon: /* @__PURE__ */ React__default.createElement(
+            "img",
+            {
+              alt: `${country?.name || region}`,
+              src: `https://purecatamphetamine.github.io/country-flag-icons/3x2/${region}.svg`,
+              className: "w-5"
+            }
+          )
+        };
+      })
+    ];
+    const fetchCollectionsForRegion = React__default.useCallback(
+      async (regionId) => {
+        if (!apiKey || !fetchRegionConfig) {
+          console.warn("No apiKey or fetchRegionConfig provided, falling back to current collections");
+          return collections.map((collection) => ({
+            value: collection.id,
+            label: collection.name,
+            icon: collection.siteEnabled ? /* @__PURE__ */ React__default.createElement(Globe, { className: "size-4" }) : /* @__PURE__ */ React__default.createElement(LockSimple, { className: "size-4" })
+          }));
+        }
+        try {
+          const regionConfig = await fetchRegionConfig(apiKey, regionId);
+          const regionCollections = regionConfig?.settings?.collections || [];
+          return regionCollections.map((collection) => ({
+            value: collection.id,
+            label: collection.name,
+            icon: collection.siteEnabled ? /* @__PURE__ */ React__default.createElement(Globe, { className: "size-4" }) : /* @__PURE__ */ React__default.createElement(LockSimple, { className: "size-4" })
+          }));
+        } catch (error) {
+          console.error("Error fetching collections for region:", error);
+          return collections.map((collection) => ({
+            value: collection.id,
+            label: collection.name,
+            icon: collection.siteEnabled ? /* @__PURE__ */ React__default.createElement(Globe, { className: "size-4" }) : /* @__PURE__ */ React__default.createElement(LockSimple, { className: "size-4" })
+          }));
+        }
+      },
+      [apiKey, fetchRegionConfig, collections]
+    );
+    const handleSelectionChange = (region, collection) => {
+      if (region && onRegionChange) {
+        onRegionChange(region.value);
+      }
+      if (collection && onCollectionChange) {
+        onCollectionChange(collection.value);
+      }
+    };
     const adjustTextareaHeight = (textarea) => {
       textarea.style.height = "auto";
       textarea.style.height = `${textarea.scrollHeight}px`;
@@ -4868,19 +4935,40 @@ const PrimarySearch = React__default.forwardRef(
         },
         /* @__PURE__ */ React__default.createElement(X, { className: "size-3 text-white", "aria-label": "Remove image" })
       )))),
-      !shouldShowSuggestions && /* @__PURE__ */ React__default.createElement("div", { className: "flex items-center justify-between px-3 py-2 border-t border-border" }, /* @__PURE__ */ React__default.createElement("div", { className: "flex gap-1", "data-name": "collections-dropdown" }, collections.length > 0 && /* @__PURE__ */ React__default.createElement(DropdownMenu, null, /* @__PURE__ */ React__default.createElement(DropdownMenuTrigger, { asChild: true }, /* @__PURE__ */ React__default.createElement(
-        Button,
-        {
-          variant: "ghost",
-          size: "sm",
-          className: cn(
-            "transition-all duration-200 rounded-md h-8",
-            "flex items-center gap-1 hover:bg-opacity-80 text-foreground"
-          )
-        },
-        collections.find((c) => c.id === currentCollectionId)?.name || textConfig.allDocumentsText || "All documents",
-        /* @__PURE__ */ React__default.createElement(CaretDown, { className: "size-4 text-secondary-foreground" })
-      )), /* @__PURE__ */ React__default.createElement(DropdownMenuContent, { className: "max-w-48" }, collections.map((collection) => /* @__PURE__ */ React__default.createElement(DropdownMenuItem, { key: collection.id, onClick: () => onCollectionChange?.(collection.id) }, collection.id === currentCollectionId ? /* @__PURE__ */ React__default.createElement(CheckCircle, { className: "rounded-full text-primary-500 !size-5", weight: "fill" }) : /* @__PURE__ */ React__default.createElement(Circle, { className: "rounded-full !size-5 text-gray-300" }), /* @__PURE__ */ React__default.createElement("span", { className: "text-sm truncate" }, collection.name))))), !disableImageAttachment && /* @__PURE__ */ React__default.createElement(React__default.Fragment, null, /* @__PURE__ */ React__default.createElement(
+      !shouldShowSuggestions && /* @__PURE__ */ React__default.createElement("div", { className: "flex items-center justify-between px-3 py-2 border-t border-border" }, /* @__PURE__ */ React__default.createElement("div", { className: "flex gap-1", "data-name": "collections-dropdown" }, allowedRegions.length > 0 && collections.length > 0 ? (
+        // Two-level selection: Region -> Collection
+        /* @__PURE__ */ React__default.createElement(
+          TwoLevelCombobox,
+          {
+            firstLevelItems: regionItems,
+            fetchSecondLevelItems: fetchCollectionsForRegion,
+            onSelectionChange: handleSelectionChange,
+            firstLevelLabel: "Region",
+            secondLevelLabel: "Collection",
+            firstLevelPlaceholder: "All documents",
+            className: "min-w-[200px]",
+            displayLabel: false
+          }
+        )
+      ) : collections.length > 0 ? (
+        // Single-level selection: Collection only (fallback when no regions)
+        /* @__PURE__ */ React__default.createElement(
+          Combobox,
+          {
+            options: collections.map((collection) => ({
+              value: collection.id,
+              label: collection.name,
+              icon: collection.siteEnabled ? /* @__PURE__ */ React__default.createElement(Globe, { className: "size-4" }) : /* @__PURE__ */ React__default.createElement(LockSimple, { className: "size-4" })
+            })),
+            value: currentCollectionId === "all" ? void 0 : currentCollectionId,
+            defaultValueLabel: "All documents",
+            placeholder: "Select collection",
+            noResultsLabel: "No collections found",
+            dropdownWidth: 200,
+            onChange: (value) => onCollectionChange?.(value)
+          }
+        )
+      ) : null, !disableImageAttachment && /* @__PURE__ */ React__default.createElement(React__default.Fragment, null, /* @__PURE__ */ React__default.createElement(
         "input",
         {
           type: "file",
@@ -5304,6 +5392,10 @@ const ChatSearchComponent = forwardRef(
     collections = [],
     selectedCollectionId,
     onCollectionChange,
+    selectedRegion,
+    onRegionChange,
+    apiKey,
+    fetchRegionConfig,
     searchEndpoint,
     answerStreamEndpoint,
     feedbackEndpoint,
@@ -5355,6 +5447,7 @@ const ChatSearchComponent = forwardRef(
     const [showScrollArrow, setShowScrollArrow] = useState(false);
     const [attachedImages, setAttachedImages] = useState([]);
     const [currentCollectionId, setCurrentCollectionId] = useState(selectedCollectionId);
+    const [currentRegion, setCurrentRegion] = useState(selectedRegion);
     const [showFollowUp, setShowFollowUp] = useState(true);
     const primaryTextareaRef = useRef(null);
     const answerRefs = useRef([]);
@@ -5720,6 +5813,9 @@ const ChatSearchComponent = forwardRef(
     useEffect(() => {
       setCurrentCollectionId(selectedCollectionId);
     }, [selectedCollectionId]);
+    useEffect(() => {
+      setCurrentRegion(selectedRegion);
+    }, [selectedRegion]);
     const adjustTextareaHeight = (textarea) => {
       textarea.style.height = "auto";
       textarea.style.height = `${textarea.scrollHeight}px`;
@@ -5827,6 +5923,11 @@ const ChatSearchComponent = forwardRef(
       setCurrentCollectionId(collectionId);
       onCollectionChange?.(collectionId);
     };
+    const handleRegionChange = (regionId) => {
+      setCurrentRegion(regionId);
+      onRegionChange?.(regionId);
+      trackEvent?.("Region Changed", { region: regionId, conversationId: currentConversationId });
+    };
     useEffect(() => {
       const handleUrlChange = () => {
         const urlParams = new URLSearchParams(window.location.search);
@@ -5914,6 +6015,11 @@ const ChatSearchComponent = forwardRef(
               collections,
               currentCollectionId,
               onCollectionChange: handleCollectionChange,
+              allowedRegions,
+              currentRegion,
+              onRegionChange: handleRegionChange,
+              apiKey,
+              fetchRegionConfig,
               disableImageAttachment,
               attachedImages,
               onImageUpload: handleImageUpload,
@@ -6025,4 +6131,4 @@ const ChatSearch = forwardRef(({ featureFlags, ...props }, ref) => /* @__PURE__
 ChatSearch.displayName = "ChatSearch";
 
 export { ChatSearch as C, ChatSearchProvider as a, useIsChatSearchDirty as b, useChatSearch as u };
-//# sourceMappingURL=ChatSearch.BdkUMxIG.js.map
+//# sourceMappingURL=ChatSearch.BVivOuWt.js.map