fix: fix isModelNotavailableInServer logic for bytedance models

This commit is contained in:
zcong1993 2025-02-03 16:58:42 +08:00
parent 1e20b64048
commit 92f57fb18f
No known key found for this signature in database
GPG Key ID: CE68FEFC68F0E3D6

View File

@ -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;
}