Skip to content

Commit 52726d4

Browse files
committed
feat: support close model provider display
1 parent 4f93704 commit 52726d4

File tree

4 files changed

+17
-1
lines changed

4 files changed

+17
-1
lines changed

app/api/config/route.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ const DANGER_CONFIG = {
1717
defaultModel: serverConfig.defaultModel,
1818
edgeTTSVoiceName: serverConfig.edgeTTSVoiceName,
1919
isUseOpenAIEndpointForAllModels: serverConfig.isUseOpenAIEndpointForAllModels,
20+
disableModelProviderDisplay: serverConfig.disableModelProviderDisplay,
2021
};
2122

2223
declare global {

app/components/chat.tsx

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -570,6 +570,11 @@ export function ChatActions(props: {
570570
// eslint-disable-next-line react-hooks/exhaustive-deps
571571
[],
572572
);
573+
const isDisableModelProviderDisplay = useMemo(
574+
() => accessStore.isDisableModelProviderDisplay(),
575+
// eslint-disable-next-line react-hooks/exhaustive-deps
576+
[],
577+
);
573578

574579
useEffect(() => {
575580
const show = isVisionModel(currentModel);
@@ -690,7 +695,7 @@ export function ChatActions(props: {
690695
defaultSelectedValue={`${currentModel}@${currentProviderName}`}
691696
items={models.map((m) => ({
692697
title: `${m.displayName}${
693-
m?.provider?.providerName
698+
m?.provider?.providerName && !isDisableModelProviderDisplay
694699
? "(" + m?.provider?.providerName + ")"
695700
: ""
696701
}`,

app/config/server.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -231,5 +231,7 @@ export const getServerSideConfig = () => {
231231
edgeTTSVoiceName: process.env.EDGE_TTS_VOICE_NAME ?? "zh-CN-YunxiNeural",
232232
isUseOpenAIEndpointForAllModels:
233233
!!process.env.USE_OPENAI_ENDPOINT_FOR_ALL_MODELS,
234+
235+
disableModelProviderDisplay: !!process.env.DISABLE_MODEL_PROVIDER_DISPLAY,
234236
};
235237
};

app/store/access.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,8 @@ const DEFAULT_ACCESS_STATE = {
9494
edgeTTSVoiceName: "zh-CN-YunxiNeural",
9595

9696
isUseOpenAIEndpointForAllModels: false,
97+
98+
disableModelProviderDisplay: false,
9799
};
98100

99101
export const useAccessStore = createPersistStore(
@@ -106,6 +108,12 @@ export const useAccessStore = createPersistStore(
106108
return get().needCode;
107109
},
108110

111+
isDisableModelProviderDisplay() {
112+
this.fetch();
113+
114+
return get().disableModelProviderDisplay;
115+
},
116+
109117
useOpenAIEndpointForAllModels() {
110118
this.fetch();
111119

0 commit comments

Comments
 (0)