feat: customise feed manage according to brain type

QuivrHQ · Dec 13, 2023 · 6b1f3fa · 6b1f3fa
1 parent 3315ee5
commit 6b1f3fa
Show file tree

Hide file tree

Showing 4 changed files with 54 additions and 6 deletions.
diff --git a/...tionsModal/components/FeedCardTrigger.tsx → ...nents/FeedCardTrigger/FeedCardTrigger.tsx b/...tionsModal/components/FeedCardTrigger.tsx → ...nents/FeedCardTrigger/FeedCardTrigger.tsx
@@ -1,18 +1,18 @@
-import { useTranslation } from "react-i18next";
-import { LuChevronRight, LuFilePlus } from "react-icons/lu";
+import { LuChevronRight } from "react-icons/lu";
 
 import { useKnowledgeToFeedContext } from "@/lib/context/KnowledgeToFeedProvider/hooks/useKnowledgeToFeedContext";
 
-import { Button } from "./Button";
+import { useFeedCardTrigger } from "./hooks/useFeedCardTrigger";
+import { Button } from "../Button";
 
 export const FeedCardTrigger = (): JSX.Element => {
-  const { t } = useTranslation("chat");
   const { setShouldDisplayFeedCard } = useKnowledgeToFeedContext();
+  const { label, Icon } = useFeedCardTrigger();
 
   return (
     <Button
-      label={t("add_document")}
-      startIcon={<LuFilePlus size={18} />}
+      label={label}
+      startIcon={<Icon size={18} />}
       endIcon={<LuChevronRight size={18} />}
       className="w-full"
       onClick={() => setShouldDisplayFeedCard(true)}

diff --git a/.../ChatInput/components/ActionsModal/components/FeedCardTrigger/hooks/useFeedCardTrigger.ts b/.../ChatInput/components/ActionsModal/components/FeedCardTrigger/hooks/useFeedCardTrigger.ts
@@ -0,0 +1,20 @@
+import { Fragment } from "react";
+
+import { useBrainContext } from "@/lib/context/BrainProvider/hooks/useBrainContext";
+
+import { useFeedCardTriggerUtils } from "./useFeedCardTriggerUtils";
+
+// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
+export const useFeedCardTrigger = () => {
+  const { brainTypeToIcon, brainTypeToLabel } = useFeedCardTriggerUtils();
+  const { currentBrain } = useBrainContext();
+
+  const isBrainTypeDefined = currentBrain?.brain_type !== undefined;
+
+  return {
+    label: isBrainTypeDefined ? brainTypeToLabel[currentBrain.brain_type] : "",
+    Icon: isBrainTypeDefined
+      ? brainTypeToIcon[currentBrain.brain_type]
+      : Fragment,
+  };
+};
diff --git a/...Input/components/ActionsModal/components/FeedCardTrigger/hooks/useFeedCardTriggerUtils.ts b/...Input/components/ActionsModal/components/FeedCardTrigger/hooks/useFeedCardTriggerUtils.ts
@@ -0,0 +1,27 @@
+import { useTranslation } from "react-i18next";
+import { IconType } from "react-icons/lib";
+import { LuBot, LuFilePlus, LuUnlock } from "react-icons/lu";
+
+import { BrainType } from "@/lib/types/brainConfig";
+
+// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
+export const useFeedCardTriggerUtils = () => {
+  const { t } = useTranslation(["chat", "brain"]);
+
+  const brainTypeToLabel: Record<BrainType, string> = {
+    doc: t("chat:add_document"),
+    api: t("brain:update_secrets_button"),
+    composite: t("brain:manage_brain"),
+  };
+
+  const brainTypeToIcon: Record<BrainType, IconType> = {
+    doc: LuFilePlus,
+    api: LuUnlock,
+    composite: LuBot,
+  };
+
+  return {
+    brainTypeToIcon,
+    brainTypeToLabel,
+  };
+};
diff --git a/...tionsBar/components/ChatInput/components/ActionsModal/components/FeedCardTrigger/index.ts b/...tionsBar/components/ChatInput/components/ActionsModal/components/FeedCardTrigger/index.ts
@@ -0,0 +1 @@
+export * from "./FeedCardTrigger";