From af21c57e77560087eb2911e88b24398fc6b2d0ff Mon Sep 17 00:00:00 2001
From: lyf <1910527151@qq.com>
Date: Wed, 7 Aug 2024 10:15:39 +0800
Subject: [PATCH] ban gpt4
---
app/api/openai.ts | 2 +-
app/config/server.ts | 10 ++++++++--
2 files changed, 9 insertions(+), 3 deletions(-)
diff --git a/app/api/openai.ts b/app/api/openai.ts
index 6d11d6792..d32e45dd9 100644
--- a/app/api/openai.ts
+++ b/app/api/openai.ts
@@ -13,7 +13,7 @@ function getModels(remoteModelRes: OpenAIListModelResponse) {
if (config.disableGPT4) {
remoteModelRes.data = remoteModelRes.data.filter(
- (m) => !m.id.startsWith("gpt-4"),
+ (m) => !m.id.startsWith("gpt-4") || m.id.startsWith("gpt-40-mini"),
);
}
diff --git a/app/config/server.ts b/app/config/server.ts
index 1fba454e8..5bfa2c2df 100644
--- a/app/config/server.ts
+++ b/app/config/server.ts
@@ -119,10 +119,16 @@ export const getServerSideConfig = () => {
if (disableGPT4) {
if (customModels) customModels += ",";
- customModels += DEFAULT_MODELS.filter((m) => m.name.startsWith("gpt-4"))
+ customModels += DEFAULT_MODELS.filter(
+ (m) => m.name.startsWith("gpt-4") && !m.name.startsWith("gpt-4o-mini"),
+ )
.map((m) => "-" + m.name)
.join(",");
- if (defaultModel.startsWith("gpt-4")) defaultModel = "";
+ if (
+ defaultModel.startsWith("gpt-4") &&
+ !defaultModel.startsWith("gpt-4o-mini")
+ )
+ defaultModel = "";
}
const isStability = !!process.env.STABILITY_API_KEY;