diff --git a/.env.template b/.env.template index 5f3fc02dab3..954510c7852 100644 --- a/.env.template +++ b/.env.template @@ -27,3 +27,8 @@ HIDE_USER_API_KEY= # Default: Empty # If you do not want users to use GPT-4, set this value to 1. DISABLE_GPT4= + +# (optional) +# Default: Empty +# If you do not want users to query balance, set this value to 1. +Hide_Balance_Query= \ No newline at end of file diff --git a/README.md b/README.md index 148c137f88d..231fb1f07ae 100644 --- a/README.md +++ b/README.md @@ -185,6 +185,12 @@ If you do not want users to input their own API key, set this value to 1. If you do not want users to use GPT-4, set this value to 1. +### `Hide_Balance_Query` (optional) + +> Default: Empty + +If you do not want users to query balance, set this value to 1. + ## Requirements NodeJS >= 18, Docker >= 20 diff --git a/README_CN.md b/README_CN.md index f1be5cc9872..b32b2d80aba 100644 --- a/README_CN.md +++ b/README_CN.md @@ -98,6 +98,10 @@ OpenAI 接口代理 URL,如果你手动配置了 openai 接口代理,请填 如果你不想让用户使用 GPT-4,将此环境变量设置为 1 即可。 +### `Hide_Balance_Query` (可选) + +如果你不想让用户查询余额,将此环境变量设置为 1 即可。 + ## 开发 点击下方按钮,开始二次开发: diff --git a/README_ES.md b/README_ES.md index cdd8359087f..78fe11533f6 100644 --- a/README_ES.md +++ b/README_ES.md @@ -96,6 +96,10 @@ Si no desea que los usuarios rellenen la clave de API ellos mismos, establezca e Si no desea que los usuarios utilicen GPT-4, establezca esta variable de entorno en 1. +### `Hide_Balance_Query` (Opcional) + +Si no desea que los usuarios consulte el saldo, establezca esta variable de entorno en 1. + ## explotación > No se recomienda encarecidamente desarrollar o implementar localmente, debido a algunas razones técnicas, es difícil configurar el agente API de OpenAI localmente, a menos que pueda asegurarse de que puede conectarse directamente al servidor OpenAI. diff --git a/app/api/config/route.ts b/app/api/config/route.ts index 2b3bcbf203e..6b95655885a 100644 --- a/app/api/config/route.ts +++ b/app/api/config/route.ts @@ -10,6 +10,7 @@ const DANGER_CONFIG = { needCode: serverConfig.needCode, hideUserApiKey: serverConfig.hideUserApiKey, enableGPT4: serverConfig.enableGPT4, + hideBalanceQuery: serverConfig.hideBalanceQuery, }; declare global { diff --git a/app/components/settings.tsx b/app/components/settings.tsx index 9e377478fb3..41fed620cdd 100644 --- a/app/components/settings.tsx +++ b/app/components/settings.tsx @@ -522,29 +522,31 @@ export function Settings() { ) : null} - - {!showUsage || loadingUsage ? ( -
- ) : ( - } - text={Locale.Settings.Usage.Check} - onClick={() => checkUsage(true)} - /> - )} - + {!accessStore.hideBalanceQuery ? ( + + {!showUsage || loadingUsage ? ( +
+ ) : ( + } + text={Locale.Settings.Usage.Check} + onClick={() => checkUsage(true)} + /> + )} + + ) : null} {!accessStore.hideUserApiKey ? ( { isVercel: !!process.env.VERCEL, hideUserApiKey: !!process.env.HIDE_USER_API_KEY, enableGPT4: !process.env.DISABLE_GPT4, + hideBalanceQuery: !!process.env.Hide_Balance_Query, }; }; diff --git a/app/store/access.ts b/app/store/access.ts index 0601903d348..e9d09bb8489 100644 --- a/app/store/access.ts +++ b/app/store/access.ts @@ -13,6 +13,7 @@ export interface AccessControlStore { needCode: boolean; hideUserApiKey: boolean; openaiUrl: string; + hideBalanceQuery: boolean; updateToken: (_: string) => void; updateCode: (_: string) => void; @@ -36,6 +37,7 @@ export const useAccessStore = create()( needCode: true, hideUserApiKey: false, openaiUrl: DEFAULT_OPENAI_URL, + hideBalanceQuery: false, enabledAccessControl() { get().fetch();