diff --git a/app/utils/model.ts b/app/utils/model.ts index 6b5173a5977..249987726ad 100644 --- a/app/utils/model.ts +++ b/app/utils/model.ts @@ -3,7 +3,7 @@ import { LLMModel } from "../client/api"; const customProvider = (modelName: string) => ({ id: modelName, - providerName: "", + providerName: "Custom", providerType: "custom", }); @@ -50,7 +50,7 @@ export function collectModelTable( // 1. find model by name(), and set available value let count = 0; for (const fullName in modelTable) { - if (fullName.includes(name)) { + if (fullName.split("@").shift() == name) { count += 1; modelTable[fullName]["available"] = available; if (displayName) {