Skip to content

Commit 1a5286b

Browse files
🐛 fix: feature flag knowledge_base doesn't affect ActionBar (lobehub#3609)
* 🐛 fix: Feature flag `knoeledge_base` doesn't affect ActionBar * 🐛 fix: Feature flag `knoeledge_base` doesn't affect ActionBar
1 parent d1d0350 commit 1a5286b

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed

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

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ import { useTranslation } from 'react-i18next';
55

66
import TipGuide from '@/components/TipGuide';
77
import { isServerMode } from '@/const/version';
8+
import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
89
import { useUserStore } from '@/store/user';
910
import { preferenceSelectors } from '@/store/user/selectors';
1011

@@ -15,11 +16,17 @@ const enableKnowledge = isServerMode;
1516
const Knowledge = memo(() => {
1617
const { t } = useTranslation('chat');
1718

19+
const { enableKnowledgeBase } = useServerConfigStore(featureFlagsSelectors);
20+
1821
const [showTip, updateGuideState] = useUserStore((s) => [
1922
preferenceSelectors.showUploadFileInKnowledgeBaseTip(s),
2023
s.updateGuideState,
2124
]);
2225

26+
if (!enableKnowledgeBase) {
27+
return null;
28+
}
29+
2330
const content = (
2431
<DropdownMenu>
2532
<ActionIcon
Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,12 @@
11
import { isServerMode } from '@/const/version';
2+
import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
23

34
import ClientMode from './ClientMode';
45
import ServerMode from './ServerMode';
56

6-
const Upload = isServerMode ? ServerMode : ClientMode;
7+
const Upload = () => {
8+
const { enableKnowledgeBase } = useServerConfigStore(featureFlagsSelectors);
9+
return isServerMode && enableKnowledgeBase ? <ServerMode /> : <ClientMode />;
10+
};
711

812
export default Upload;

0 commit comments

Comments
 (0)