hfix: disable ask cohere search (#1303)

fern-api · Aug 15, 2024 · 50ef250 · 50ef250
1 parent d0ef388
commit 50ef250
Showing 1 changed file with 15 additions and 11 deletions.
diff --git a/packages/ui/app/src/search/SearchHits.tsx b/packages/ui/app/src/search/SearchHits.tsx
@@ -5,7 +5,7 @@ import { useSetAtom } from "jotai";
 import { useRouter } from "next/router";
 import React, { PropsWithChildren, useEffect, useMemo, useRef, useState } from "react";
 import { useInfiniteHits, useInstantSearch } from "react-instantsearch";
-import { COHERE_ASK_AI, COHERE_INITIAL_MESSAGE, useBasePath, useCloseSearchDialog, useDomain } from "../atoms";
+import { COHERE_ASK_AI, COHERE_INITIAL_MESSAGE, useBasePath, useCloseSearchDialog, useFeatureFlags } from "../atoms";
 import { SearchHit } from "./SearchHit";
 import { AskCohereHit } from "./cohere/AskCohereHit";
 
@@ -16,7 +16,7 @@ export const EmptyStateView: React.FC<PropsWithChildren> = ({ children }) => {
 const COHERE_AI_HIT_ID = "cohere-ai-hit";
 
 export const SearchHits: React.FC = () => {
-    const isCohere = useDomain().includes("cohere");
+    const { isAiChatbotEnabledInPreview } = useFeatureFlags();
     const basePath = useBasePath();
     const { hits } = useInfiniteHits<SearchRecord>();
     const search = useInstantSearch();
@@ -48,17 +48,17 @@ export const SearchHits: React.FC = () => {
     useEffect(() => {
         const [firstHit] = hits;
         if (firstHit != null) {
-            setHoveredSearchHitId((id) => id ?? (isCohere ? COHERE_AI_HIT_ID : firstHit.objectID));
+            setHoveredSearchHitId((id) => id ?? (isAiChatbotEnabledInPreview ? COHERE_AI_HIT_ID : firstHit.objectID));
         }
-    }, [hits, isCohere]);
+    }, [hits, isAiChatbotEnabledInPreview]);
 
     useKeyboardPress({
         key: "Up",
         onPress: () => {
             if (hoveredSearchHit == null) {
                 setHoveredSearchHitId(null);
                 return;
-            } else if (hoveredSearchHit.index === 0 && isCohere) {
+            } else if (hoveredSearchHit.index === 0 && isAiChatbotEnabledInPreview) {
                 setHoveredSearchHitId(COHERE_AI_HIT_ID);
                 return;
             }
@@ -82,7 +82,7 @@ export const SearchHits: React.FC = () => {
                 return;
             }
 
-            if (hoveredSearchHit == null && isCohere) {
+            if (hoveredSearchHit == null && isAiChatbotEnabledInPreview) {
                 setHoveredSearchHitId(COHERE_AI_HIT_ID);
                 return;
             }
@@ -107,7 +107,11 @@ export const SearchHits: React.FC = () => {
 
     const navigateToHoveredHit = async () => {
         if (hoveredSearchHit == null) {
-            if (isCohere && hoveredSearchHitId === COHERE_AI_HIT_ID && search.results.query.length > 0) {
+            if (
+                isAiChatbotEnabledInPreview &&
+                hoveredSearchHitId === COHERE_AI_HIT_ID &&
+                search.results.query.length > 0
+            ) {
                 closeSearchDialog();
                 openCohere();
             }
@@ -136,7 +140,7 @@ export const SearchHits: React.FC = () => {
         capture: true,
     });
 
-    if ((hits.length === 0 && !isCohere) || search.results.query.length === 0) {
+    if ((hits.length === 0 && !isAiChatbotEnabledInPreview) || search.results.query.length === 0) {
         return null;
     }
 
@@ -146,7 +150,7 @@ export const SearchHits: React.FC = () => {
             className="p-2"
             scrollbars="vertical"
         >
-            {isCohere && (
+            {isAiChatbotEnabledInPreview && (
                 <AskCohereHit
                     setRef={(elem) => {
                         if (elem != null) {
@@ -176,7 +180,7 @@ export const SearchHits: React.FC = () => {
 };
 
 export const SearchMobileHits: React.FC<PropsWithChildren> = ({ children }) => {
-    const isCohere = useDomain().includes("cohere");
+    const { isAiChatbotEnabledInPreview } = useFeatureFlags();
     const { hits } = useInfiniteHits<SearchRecord>();
     const search = useInstantSearch();
 
@@ -193,7 +197,7 @@ export const SearchMobileHits: React.FC<PropsWithChildren> = ({ children }) => {
 
     return (
         <FernScrollArea className="mask-grad-top-4 px-2 pt-4">
-            {isCohere && (
+            {isAiChatbotEnabledInPreview && (
                 <AskCohereHit
                     setRef={(elem) => {
                         if (elem != null) {