This commit is contained in:
lloydzhou 2024-07-04 19:38:26 +08:00
parent 97aa72ec5b
commit 31d9444264
1 changed files with 2 additions and 2 deletions

View File

@ -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) {