lloydzhou před 1 rokem
rodič
revize
31d9444264
1 změnil soubory, kde provedl 2 přidání a 2 odebrání
  1. 2 2
      app/utils/model.ts

+ 2 - 2
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) {