hotfix

DDDDD12138 · Jul 4, 2024 · 31d9444 · 31d9444
1 parent 97aa72e
commit 31d9444
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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) {