fix: put knowledge bases under feature flag

src/frontend/src/components/core/folderSidebarComponent/components/sideBarFolderButtons/index.tsx

@@ -26,6 +26,7 @@ import {
   ENABLE_CUSTOM_PARAM,
   ENABLE_DATASTAX_LANGFLOW,
   ENABLE_FILE_MANAGEMENT,
+  ENABLE_KNOWLEDGE_BASES,
   ENABLE_MCP_NOTICE,
 } from "@/customization/feature-flags";
 import { useCustomNavigate } from "@/customization/hooks/use-custom-navigate";
@@ -478,14 +479,16 @@ const SideBarFoldersButtonsComponent = ({
           <div className="grid w-full items-center gap-2 p-2">
             {/* TODO: Remove this on cleanup */}
             {ENABLE_DATASTAX_LANGFLOW && <CustomStoreButton />}{" "}
-            <SidebarMenuButton
-              onClick={handleKnowledgeNavigation}
-              size="md"
-              className="text-sm"
-            >
-              <ForwardedIconComponent name="Library" className="h-4 w-4" />
-              Knowledge
-            </SidebarMenuButton>
+            {ENABLE_KNOWLEDGE_BASES && (
+              <SidebarMenuButton
+                onClick={handleKnowledgeNavigation}
+                size="md"
+                className="text-sm"
+              >
+                <ForwardedIconComponent name="Library" className="h-4 w-4" />
+                Knowledge
+              </SidebarMenuButton>
+            )}
             <SidebarMenuButton
               onClick={handleFilesNavigation}
               size="md"

src/frontend/src/controllers/API/queries/flows/use-get-types.ts

@@ -1,3 +1,4 @@
+import { ENABLE_KNOWLEDGE_BASES } from "@/customization/feature-flags";
 import useFlowsManagerStore from "@/stores/flowsManagerStore";
 import { useTypesStore } from "@/stores/typesStore";
 import type {
@@ -30,6 +31,11 @@ export const useGetTypes: useQueryFunctionType<
         `${getURL("ALL")}?force_refresh=true`,
       );
       const data = response?.data;
+
+      if (!ENABLE_KNOWLEDGE_BASES) {
+        delete data.knowledge_bases;
+      }
+
       setTypes(data);
       return data;
     } catch (error) {

src/frontend/src/routes.tsx

@@ -16,6 +16,7 @@ import { BASENAME } from "./customization/config-constants";
 import {
   ENABLE_CUSTOM_PARAM,
   ENABLE_FILE_MANAGEMENT,
+  ENABLE_KNOWLEDGE_BASES,
 } from "./customization/feature-flags";
 import { CustomRoutesStore } from "./customization/utils/custom-routes-store";
 import { CustomRoutesStorePages } from "./customization/utils/custom-routes-store-pages";
@@ -89,10 +90,12 @@ const router = createBrowserRouter(
                         element={<CustomNavigate replace to="files" />}
                       />
                       <Route path="files" element={<FilesPage />} />
-                      <Route
-                        path="knowledge-bases"
-                        element={<KnowledgePage />}
-                      />
+                      {ENABLE_KNOWLEDGE_BASES && (
+                        <Route
+                          path="knowledge-bases"
+                          element={<KnowledgePage />}
+                        />
+                      )}
                     </Route>
                   )}
                   <Route

src/frontend/src/utils/styleUtils.ts

@@ -2,6 +2,7 @@ import { TwitterLogoIcon } from "@radix-ui/react-icons";
 import dynamicIconImports from "lucide-react/dynamicIconImports";
 import { lazy } from "react";
 import { FaApple, FaDiscord, FaGithub } from "react-icons/fa";
+import { ENABLE_KNOWLEDGE_BASES } from "@/customization/feature-flags";
 import { BotMessageSquareIcon } from "@/icons/BotMessageSquare";
 import { fontAwesomeIcons, isFontAwesomeIcon } from "@/icons/fontAwesomeIcons";
 import { GradientSave } from "@/icons/GradientSparkles";
@@ -211,11 +212,15 @@ export const SIDEBAR_CATEGORIES = [
   { display_name: "Agents", name: "agents", icon: "Bot" },
   { display_name: "Models", name: "models", icon: "BrainCog" },
   { display_name: "Data", name: "data", icon: "Database" },
-  {
-    display_name: "Knowledge Bases",
-    name: "knowledge_bases",
-    icon: "Library",
-  },
+  ...(ENABLE_KNOWLEDGE_BASES
+    ? [
+        {
+          display_name: "Knowledge Bases",
+          name: "knowledge_bases",
+          icon: "Library",
+        },
+      ]
+    : []),
   { display_name: "Vector Stores", name: "vectorstores", icon: "Layers" },
   { display_name: "Processing", name: "processing", icon: "ListFilter" },
   { display_name: "Logic", name: "logic", icon: "ArrowRightLeft" },