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();