Merge pull request #6134 from zcong1993/main

fix: fix isModelNotavailableInServer logic for bytedance models
ChatGPTNextWeb · Feb 4, 2025 · 5db6775 · 5db6775
2 parents b6881c7 + 92f57fb
commit 5db6775
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/app/utils/model.ts b/app/utils/model.ts
@@ -1,4 +1,4 @@
-import { DEFAULT_MODELS } from "../constant";
+import { DEFAULT_MODELS, ServiceProvider } from "../constant";
 import { LLMModel } from "../client/api";
 
 const CustomSeq = {
@@ -246,6 +246,11 @@ export function isModelNotavailableInServer(
     ? providerNames
     : [providerNames];
   for (const providerName of providerNamesArray) {
+    // if model provider is bytedance, use model config name to check if not avaliable
+    if (providerName === ServiceProvider.ByteDance) {
+      return !Object.values(modelTable).filter((v) => v.name === modelName)?.[0]
+        ?.available;
+    }
     const fullName = `${modelName}@${providerName.toLowerCase()}`;
     if (modelTable?.[fullName]?.available === true) return false;
   }