feat: support close model provider display

Hk-Gosuto · Nov 3, 2024 · 52726d4 · 52726d4
1 parent 4f93704
commit 52726d4
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git a/app/api/config/route.ts b/app/api/config/route.ts
@@ -17,6 +17,7 @@ const DANGER_CONFIG = {
   defaultModel: serverConfig.defaultModel,
   edgeTTSVoiceName: serverConfig.edgeTTSVoiceName,
   isUseOpenAIEndpointForAllModels: serverConfig.isUseOpenAIEndpointForAllModels,
+  disableModelProviderDisplay: serverConfig.disableModelProviderDisplay,
 };
 
 declare global {

diff --git a/app/components/chat.tsx b/app/components/chat.tsx
@@ -570,6 +570,11 @@ export function ChatActions(props: {
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [],
   );
+  const isDisableModelProviderDisplay = useMemo(
+    () => accessStore.isDisableModelProviderDisplay(),
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+    [],
+  );
 
   useEffect(() => {
     const show = isVisionModel(currentModel);
@@ -690,7 +695,7 @@ export function ChatActions(props: {
             defaultSelectedValue={`${currentModel}@${currentProviderName}`}
             items={models.map((m) => ({
               title: `${m.displayName}${
-                m?.provider?.providerName
+                m?.provider?.providerName && !isDisableModelProviderDisplay
                   ? "(" + m?.provider?.providerName + ")"
                   : ""
               }`,

diff --git a/app/config/server.ts b/app/config/server.ts
@@ -231,5 +231,7 @@ export const getServerSideConfig = () => {
     edgeTTSVoiceName: process.env.EDGE_TTS_VOICE_NAME ?? "zh-CN-YunxiNeural",
     isUseOpenAIEndpointForAllModels:
       !!process.env.USE_OPENAI_ENDPOINT_FOR_ALL_MODELS,
+
+    disableModelProviderDisplay: !!process.env.DISABLE_MODEL_PROVIDER_DISPLAY,
   };
 };
diff --git a/app/store/access.ts b/app/store/access.ts
@@ -94,6 +94,8 @@ const DEFAULT_ACCESS_STATE = {
   edgeTTSVoiceName: "zh-CN-YunxiNeural",
 
   isUseOpenAIEndpointForAllModels: false,
+
+  disableModelProviderDisplay: false,
 };
 
 export const useAccessStore = createPersistStore(
@@ -106,6 +108,12 @@ export const useAccessStore = createPersistStore(
       return get().needCode;
     },
 
+    isDisableModelProviderDisplay() {
+      this.fetch();
+
+      return get().disableModelProviderDisplay;
+    },
+
     useOpenAIEndpointForAllModels() {
       this.fetch();