Selaa lähdekoodia

Merge pull request #6134 from zcong1993/main

fix: fix isModelNotavailableInServer logic for bytedance models
RiverRay 10 kuukautta sitten
vanhempi
commit
5db6775cb8
1 muutettua tiedostoa jossa 6 lisäystä ja 1 poistoa
  1. 6 1
      app/utils/model.ts

+ 6 - 1
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;
   }