@brainfish-ai/components 0.25.4 → 0.26.0

package/dist/chat-search.d.ts

@@ -112,7 +112,7 @@ declare interface AnswerAttachment {
 
 declare type AnswerBlock = MarkdownTextBlock | ActionButtonsBlock | ActionInputFormBlock;
 
-declare type AnswerListAction = SetAnswers | ClearAll | AppendNewAnswer | SetSearchResults | AppendAnswerChunk | SetAnswerError | SetUserFeedback | ClearUserFeedback | SetFollowUpQuestions | AppendBlock | CompleteAnswer | MarkAsUncertain | InvokeAction | GetActionInputs | AddActionButtons | NoArticlesFound;
+declare type AnswerListAction = SetAnswers | ClearAll | AppendNewAnswer | SetSearchResults | AppendAnswerChunk | SetAnswerError | SetUserFeedback | ClearUserFeedback | SetFollowUpQuestions | AppendBlock | CompleteAnswer | MarkAsUncertain | InvokeAction | GetActionInputs | AddActionButtons | NoArticlesFound | SetAwaitingKnowledgeGroup | SetKnowledgeGroupSelected;
 
 declare const AnswersActions: {
     readonly SET_ANSWERS: "answers/set_answers";
@@ -131,9 +131,11 @@ declare const AnswersActions: {
     readonly GET_ACTION_INPUTS: "answers/request_action_inputs";
     readonly INVOKE_ACTION: "answers/invoke_action";
     readonly ADD_ACTION_BUTTONS: "answers/add_action_buttons";
+    readonly SET_AWAITING_KNOWLEDGE_GROUP: "answers/set_awaiting_knowledge_group";
+    readonly SET_KNOWLEDGE_GROUP_SELECTED: "answers/set_knowledge_group_selected";
 };
 
-declare type AnswerState = 'fetching-search-results' | 'fetching-stream' | 'streaming' | 'getting-action-inputs' | 'completed';
+declare type AnswerState = 'fetching-search-results' | 'fetching-stream' | 'streaming' | 'getting-action-inputs' | 'awaiting-knowledge-group' | 'completed';
 
 declare type AppendAnswerChunk = {
     type: typeof AnswersActions.APPEND_ANSWER_CHUNK;
@@ -255,6 +257,8 @@ export declare interface ChatSearchProps {
     isSearchWidget?: boolean;
     hideCitations?: boolean;
     conversationInlineSlot?: ReactNode;
+    liveHumanAgentChatSlot?: ReactNode;
+    knowledgeSourceGroups?: KnowledgeSourceGroup[];
 }
 
 export declare const ChatSearchProvider: default_2.FC<{
@@ -333,6 +337,12 @@ declare type InvokeAction = {
 
 declare type JSONSchema<T> = JSONSchemaType<T>;
 
+export declare interface KnowledgeSourceGroup {
+    id: string;
+    name: string;
+    collectionIds: string[];
+}
+
 declare type MarkAsUncertain = {
     type: typeof AnswersActions.MARK_AS_UNCERTAIN;
     payload: {
@@ -405,6 +415,13 @@ declare type SetAnswers = {
     };
 };
 
+declare type SetAwaitingKnowledgeGroup = {
+    type: typeof AnswersActions.SET_AWAITING_KNOWLEDGE_GROUP;
+    payload: {
+        index?: number;
+    };
+};
+
 declare type SetFollowUpQuestions = {
     type: typeof AnswersActions.SET_FOLLOW_UP_QUESTIONS;
     payload: {
@@ -413,6 +430,13 @@ declare type SetFollowUpQuestions = {
     };
 };
 
+declare type SetKnowledgeGroupSelected = {
+    type: typeof AnswersActions.SET_KNOWLEDGE_GROUP_SELECTED;
+    payload: {
+        index?: number;
+    };
+};
+
 declare type SetSearchResults = {
     type: typeof AnswersActions.SET_SEARCH_RESULTS;
     payload: {

package/dist/esm/chunks/{ChatSearch.CL2VnSod.js → ChatSearch.CeQrTOVx.js}

@@ -29,7 +29,7 @@ import { C as Combobox } from './combobox.CJKym3Z1.js';
 import { T as TwoLevelCombobox } from './two-level-combobox.BXs2z9u5.js';
 import { u as useBooleanFlagValue, F as FeatureFlagProvider } from './feature-flags.DeDEcnd1.js';
 import { X as X$1 } from 'lucide-react';
-import { l as loadConversation, c as createConversationId, a as uploadFileToApi, b as useAutocomplete, s as sendFeedbackReason, d as sendFeedback, e as searchApi, f as fetchAnswerStream, g as fetchFollowUpQuestions } from './hooks.m-nIJmio.js';
+import { l as loadConversation, c as createConversationId, a as uploadFileToApi, b as useAutocomplete, s as sendFeedbackReason, d as sendFeedback, e as searchApi, f as fetchAnswerStream, g as fetchFollowUpQuestions } from './hooks.BWVaVAT-.js';
 import { ScrollArea } from '../components/ui/scroll-area.js';
 
 import '../ChatSearch.css';function Suggestions({ suggestions, onQuestionClick, title = "Suggested questions" }) {
@@ -3509,7 +3509,10 @@ const AnswersActions = {
   // action flow
   GET_ACTION_INPUTS: "answers/request_action_inputs",
   INVOKE_ACTION: "answers/invoke_action",
-  ADD_ACTION_BUTTONS: "answers/add_action_buttons"
+  ADD_ACTION_BUTTONS: "answers/add_action_buttons",
+  // knowledge group clarification
+  SET_AWAITING_KNOWLEDGE_GROUP: "answers/set_awaiting_knowledge_group",
+  SET_KNOWLEDGE_GROUP_SELECTED: "answers/set_knowledge_group_selected"
 };
 const setAnswers = ({ answers }) => ({
   type: AnswersActions.SET_ANSWERS,
@@ -3599,6 +3602,18 @@ const noArticlesFound = ({ index } = {}) => ({
     index
   }
 });
+const setAwaitingKnowledgeGroup = ({ index } = {}) => ({
+  type: AnswersActions.SET_AWAITING_KNOWLEDGE_GROUP,
+  payload: {
+    index
+  }
+});
+const setKnowledgeGroupSelected = ({ index } = {}) => ({
+  type: AnswersActions.SET_KNOWLEDGE_GROUP_SELECTED,
+  payload: {
+    index
+  }
+});
 const getActionInputs = ({
   searchIntentId,
   actionId,
@@ -3789,6 +3804,20 @@ const reducer = (draft, action) => {
       }
       return;
     }
+    case AnswersActions.SET_AWAITING_KNOWLEDGE_GROUP: {
+      const answer = getTargetAnswer(draft, action.payload.index);
+      if (answer) {
+        answer.state = "awaiting-knowledge-group";
+      }
+      return;
+    }
+    case AnswersActions.SET_KNOWLEDGE_GROUP_SELECTED: {
+      const answer = getTargetAnswer(draft, action.payload.index);
+      if (answer) {
+        answer.state = "fetching-search-results";
+      }
+      return;
+    }
     default: {
       throw new Error(`Answer List Reducer: unknown action type '${action.type}'`);
     }
@@ -3975,6 +4004,24 @@ function AttachmentThumbnails({ items, size = "sm", onRemove, className }) {
   ))));
 }
 
+const MotionButton$2 = motion(Button);
+function KnowledgeGroupSelector({ groups, onGroupSelect }) {
+  if (!groups.length) return null;
+  return /* @__PURE__ */ React__default.createElement(motion.div, { initial: { opacity: 0, y: 20 }, animate: { opacity: 1, y: 0 }, className: "w-full px-4 py-3" }, /* @__PURE__ */ React__default.createElement("p", { className: "text-sm text-muted-foreground mb-2 text-right" }, "Please select an option"), /* @__PURE__ */ React__default.createElement("div", { className: "flex flex-wrap justify-end gap-2" }, groups.map((group, index) => /* @__PURE__ */ React__default.createElement(
+    MotionButton$2,
+    {
+      key: group.id,
+      initial: { opacity: 0, scale: 0.95 },
+      animate: { opacity: 1, scale: 1 },
+      transition: { delay: index * 0.05 },
+      onClick: () => onGroupSelect(group),
+      variant: "outline",
+      className: "border-border font-normal h-auto rounded-full px-4 py-1.5 text-sm"
+    },
+    group.name
+  ))));
+}
+
 var ActionType = /* @__PURE__ */ ((ActionType2) => {
   ActionType2["CallPhone"] = "call_phone";
   ActionType2["OpenUrl"] = "open_url";
@@ -4748,15 +4795,18 @@ function Answer({
   nextBestActions,
   isLastAnswer,
   onFeedbackReasonSubmit,
-  hasError = false
+  hasError = false,
+  knowledgeSourceGroups,
+  onKnowledgeGroupSelect
 }) {
   const [isCopied, setIsCopied] = useState(false);
   const [isExpanded, setIsExpanded] = useState(false);
   const answer = blocksToPlainText(blocks);
+  const isAwaitingKnowledgeGroup = state === "awaiting-knowledge-group";
   const isFetching = state === "fetching-search-results" || state === "fetching-stream";
   const isFetchingInitialAnswer = isFetching && blocks.length === 1;
   const isFetchingSearchResults = state === "fetching-search-results";
-  const isIdle = state === "getting-action-inputs" || state === "completed";
+  const isIdle = state === "getting-action-inputs" || state === "completed" || isAwaitingKnowledgeGroup;
   const isCompleted = state === "completed";
   const showError = hasError && isCompleted;
   const handleCopy = async () => {
@@ -4802,7 +4852,7 @@ function Answer({
       }
     )),
     isFetchingInitialAnswer && /* @__PURE__ */ React__default.createElement("div", { className: "px-4 pt-0 pb-2" }, /* @__PURE__ */ React__default.createElement("p", { className: "text-sm text-muted-foreground" }, isFetchingSearchResults ? textConfig.loadingSearchText : textConfig.loadingAnswerText)),
-    /* @__PURE__ */ React__default.createElement("div", { className: "flex flex-col" }, blocks?.map((block, i) => /* @__PURE__ */ React__default.createElement(
+    isAwaitingKnowledgeGroup && knowledgeSourceGroups && onKnowledgeGroupSelect ? /* @__PURE__ */ React__default.createElement(KnowledgeGroupSelector, { groups: knowledgeSourceGroups, onGroupSelect: onKnowledgeGroupSelect }) : /* @__PURE__ */ React__default.createElement("div", { className: "flex flex-col" }, blocks?.map((block, i) => /* @__PURE__ */ React__default.createElement(
       "div",
       {
         key: i,
@@ -5421,7 +5471,8 @@ const FollowUpSearchBar = React__default.forwardRef(
     attachedFiles = [],
     isUploading = false,
     onAddFiles,
-    onRemoveFile
+    onRemoveFile,
+    disabled = false
   }, ref) => {
     const [localQuery, setLocalQuery] = useState(initialQuery);
     const [isDragOver, setIsDragOver] = useState(false);
@@ -5444,7 +5495,7 @@ const FollowUpSearchBar = React__default.forwardRef(
     useEffect(() => {
       setLocalQuery(initialQuery);
     }, [initialQuery]);
-    const isSendDisabled = isSearching || !localQuery.trim() || isUploading;
+    const isSendDisabled = disabled || isSearching || !localQuery.trim() || isUploading;
     const handleKeyPress = (e) => {
       if (e.key === "Enter" && !e.shiftKey && !isSendDisabled) {
         e.preventDefault();
@@ -5551,7 +5602,7 @@ const FollowUpSearchBar = React__default.forwardRef(
             size: "icon",
             className: "rounded-md size-7 hover:bg-opacity-80 disabled:opacity-50 shrink-0",
             onClick: () => fileInputRef.current?.click(),
-            disabled: attachedFiles.length >= 5,
+            disabled: disabled || attachedFiles.length >= 5,
             "aria-label": textConfig.attachImageText
           },
           /* @__PURE__ */ React__default.createElement(Image, { weight: "regular", className: "size-4 text-secondary-foreground" })
@@ -5564,7 +5615,8 @@ const FollowUpSearchBar = React__default.forwardRef(
             onBlur: handleBlur,
             onKeyDown: handleKeyPress,
             placeholder: textConfig.followUpPlaceholder,
-            className: "min-h-[44px] w-full resize-none border-0 shadow-none bg-transparent px-0 py-2.5 focus:outline-none focus:ring-0 focus-visible:ring-0",
+            disabled,
+            className: "min-h-11 w-full resize-none border-0 shadow-none bg-transparent px-0 py-2.5 focus:outline-none focus:ring-0 focus-visible:ring-0 disabled:opacity-50 disabled:cursor-not-allowed",
             rows: 1
           }
         ), /* @__PURE__ */ React__default.createElement(
@@ -5838,7 +5890,9 @@ const ChatSearchComponent = forwardRef(
     isSearchWidget = false,
     isAgentAssist = false,
     hideCitations = false,
-    conversationInlineSlot
+    conversationInlineSlot,
+    liveHumanAgentChatSlot,
+    knowledgeSourceGroups
   }, ref) => {
     const {
       answers,
@@ -5880,6 +5934,8 @@ const ChatSearchComponent = forwardRef(
     const [currentCollectionId, setCurrentCollectionId] = useState(selectedCollectionId);
     const [currentRegion, setCurrentRegion] = useState(selectedRegion);
     const [showFollowUp, setShowFollowUp] = useState(true);
+    const selectedKnowledgeGroupRef = useRef(null);
+    const pendingSearchRef = useRef(null);
     const primaryTextareaRef = useRef(null);
     const answerRefs = useRef([]);
     const containerRef = useRef(null);
@@ -6164,12 +6220,14 @@ const ChatSearchComponent = forwardRef(
       conversationId,
       attributes,
       secretAttributes: secretAttributes2,
-      allowedRegions: allowedRegions2
+      allowedRegions: allowedRegions2,
+      knowledgeGroupCollectionIds
     }) => {
+      const collectionId = knowledgeGroupCollectionIds ?? currentCollectionId;
       const searchResponse = await searchApi({
         endpoint: searchEndpoint,
         query: searchQuery,
-        collectionId: currentCollectionId,
+        collectionId,
         headers,
         conversationId,
         attributes,
@@ -6208,14 +6266,31 @@ const ChatSearchComponent = forwardRef(
       await Promise.resolve();
       followUpSearchRef.current?.focus();
       const pageContext = await fetchPageContext(conversationId, searchQuery);
+      const needsGroupSelection = knowledgeSourceGroups && knowledgeSourceGroups.length > 0 && !selectedKnowledgeGroupRef.current;
+      if (needsGroupSelection) {
+        pendingSearchRef.current = {
+          searchQuery,
+          conversationId,
+          attributes: { ...userData },
+          secretAttributes,
+          allowedRegions,
+          pageContext: pageContext ?? void 0,
+          fileAttachments: fileAttachments.length > 0 ? fileAttachments : void 0
+        };
+        answerListDispatch(setAwaitingKnowledgeGroup());
+        setIsSearching(false);
+        return;
+      }
       try {
         const attributes = { ...userData };
+        const knowledgeGroupCollectionIds = selectedKnowledgeGroupRef.current?.collectionIds;
         const searchQueryId = await handleSearchApiCall({
           searchQuery,
           conversationId,
           attributes,
           secretAttributes,
-          allowedRegions
+          allowedRegions,
+          knowledgeGroupCollectionIds
         });
         if (searchQueryId) {
           event && trackEvent?.(event.name, {
@@ -6249,6 +6324,50 @@ const ChatSearchComponent = forwardRef(
         setIsSearching(false);
       }
     };
+    const handleKnowledgeGroupSelect = async (group) => {
+      selectedKnowledgeGroupRef.current = group;
+      const pending = pendingSearchRef.current;
+      if (!pending) return;
+      pendingSearchRef.current = null;
+      answerListDispatch(setKnowledgeGroupSelected());
+      setIsSearching(true);
+      try {
+        const searchQueryId = await handleSearchApiCall({
+          searchQuery: pending.searchQuery,
+          conversationId: pending.conversationId,
+          attributes: pending.attributes,
+          secretAttributes: pending.secretAttributes,
+          allowedRegions: pending.allowedRegions,
+          knowledgeGroupCollectionIds: group.collectionIds
+        });
+        if (searchQueryId) {
+          trackEvent?.("Knowledge Group Selected", {
+            conversationId: pending.conversationId,
+            searchQuery: pending.searchQuery,
+            searchQueryId,
+            knowledgeGroupId: group.id,
+            knowledgeGroupName: group.name
+          });
+          await generateAnswerForQuery({
+            searchQueryId,
+            conversationId: pending.conversationId,
+            context: pending.pageContext,
+            attachments: pending.fileAttachments
+          });
+        } else {
+          answerListDispatch(noArticlesFound());
+        }
+      } catch (error) {
+        console.error("Error generating answer after group selection:", error);
+        answerListDispatch(
+          setAnswerError({
+            error: mergedTextConfig.errorText || "Sorry, we were unable to process the request right now. Please try again in a moment."
+          })
+        );
+      } finally {
+        setIsSearching(false);
+      }
+    };
     const createNewThread = (firstQuery) => {
       if (currentReaderRef.current) {
         const reader = currentReaderRef.current;
@@ -6263,6 +6382,8 @@ const ChatSearchComponent = forwardRef(
       setActiveAnswerIndex(0);
       clearAttachedFiles();
       clearCurrentConversationId();
+      selectedKnowledgeGroupRef.current = null;
+      pendingSearchRef.current = null;
       trackEvent?.("New Thread Created", { oldConversationId: currentConversationId });
       if (primaryTextareaRef.current) {
         primaryTextareaRef.current.style.height = "auto";
@@ -6614,7 +6735,9 @@ const ChatSearchComponent = forwardRef(
                 handleActionClick: handleNextBestActionClick,
                 isLastAnswer: i === answers.length - 1,
                 onFeedbackReasonSubmit: (reason) => void handleFeedbackReasonSubmit(reason, answer.searchQueryId),
-                hasError: answer.hasError
+                hasError: answer.hasError,
+                knowledgeSourceGroups,
+                onKnowledgeGroupSelect: (group) => void handleKnowledgeGroupSelect(group)
               }
             ),
             i === answers.length - 1 && !disableFollowUpQuestions && answer.followUpQuestions && answer.state === "completed" && !answer.hasError && /* @__PURE__ */ React__default.createElement(
@@ -6662,22 +6785,27 @@ const ChatSearchComponent = forwardRef(
         conversationInlineSlot != null && /* @__PURE__ */ React__default.createElement("div", { className: "-mt-6 w-full max-w-[100vw] px-3 pb-0 pt-0 text-foreground md:px-0" }, conversationInlineSlot),
         /* @__PURE__ */ React__default.createElement("div", { ref: bottomRef })
       ),
-      showResults && showFollowUp && /* @__PURE__ */ React__default.createElement(
-        FollowUpSearchBar,
-        {
-          ref: followUpSearchRef,
-          initialQuery: followUpQuery,
-          onQueryChange: setFollowUpQuery,
-          isSearching,
-          handleFollowUpSearch,
-          textConfig: mergedTextConfig,
-          enableFileAttachment,
-          attachedFiles,
-          isUploading,
-          onAddFiles: addFiles,
-          onRemoveFile: removeFile
-        }
-      )
+      showResults && showFollowUp && !liveHumanAgentChatSlot && (() => {
+        const isAwaitingGroup = answers.some((a) => a.state === "awaiting-knowledge-group");
+        return /* @__PURE__ */ React__default.createElement(
+          FollowUpSearchBar,
+          {
+            ref: followUpSearchRef,
+            initialQuery: followUpQuery,
+            onQueryChange: setFollowUpQuery,
+            isSearching,
+            handleFollowUpSearch,
+            textConfig: mergedTextConfig,
+            enableFileAttachment,
+            attachedFiles,
+            isUploading,
+            onAddFiles: addFiles,
+            onRemoveFile: removeFile,
+            disabled: isAwaitingGroup
+          }
+        );
+      })(),
+      liveHumanAgentChatSlot
     );
   }
 );
@@ -6694,4 +6822,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.CL2VnSod.js.map
+//# sourceMappingURL=ChatSearch.CeQrTOVx.js.map