🐛 fix: feature flag knowledge_base doesn't affect ActionBar (lobehu…

…b#3609)

* 🐛 fix: Feature flag `knoeledge_base` doesn't affect ActionBar

* 🐛 fix: Feature flag `knoeledge_base` doesn't affect ActionBar

src/features/ChatInput/ActionBar/Knowledge/index.tsx

@@ -5,6 +5,7 @@ import { useTranslation } from 'react-i18next';
 
 import TipGuide from '@/components/TipGuide';
 import { isServerMode } from '@/const/version';
+import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
 import { useUserStore } from '@/store/user';
 import { preferenceSelectors } from '@/store/user/selectors';
 
@@ -15,11 +16,17 @@ const enableKnowledge = isServerMode;
 const Knowledge = memo(() => {
   const { t } = useTranslation('chat');
 
+  const { enableKnowledgeBase } = useServerConfigStore(featureFlagsSelectors);
+
   const [showTip, updateGuideState] = useUserStore((s) => [
     preferenceSelectors.showUploadFileInKnowledgeBaseTip(s),
     s.updateGuideState,
   ]);
 
+  if (!enableKnowledgeBase) {
+    return null;
+  }
+
   const content = (
     <DropdownMenu>
       <ActionIcon

src/features/ChatInput/ActionBar/Upload/index.tsx

@@ -1,8 +1,12 @@
 import { isServerMode } from '@/const/version';
+import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
 
 import ClientMode from './ClientMode';
 import ServerMode from './ServerMode';
 
-const Upload = isServerMode ? ServerMode : ClientMode;
+const Upload = () => {
+  const { enableKnowledgeBase } = useServerConfigStore(featureFlagsSelectors);
+  return isServerMode && enableKnowledgeBase ? <ServerMode /> : <ClientMode />;
+};
 
 export default Upload;