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) => ({ const customProvider = (modelName: string) => ({
id: modelName, id: modelName,
providerName: "", providerName: "Custom",
providerType: "custom", providerType: "custom",
}); });
@ -50,7 +50,7 @@ export function collectModelTable(
// 1. find model by name(), and set available value // 1. find model by name(), and set available value
let count = 0; let count = 0;
for (const fullName in modelTable) { for (const fullName in modelTable) {
if (fullName.includes(name)) { if (fullName.split("@").shift() == name) {
count += 1; count += 1;
modelTable[fullName]["available"] = available; modelTable[fullName]["available"] = available;
if (displayName) { if (displayName) {