From 31d94442644232e272ecca36c8f16e4399997633 Mon Sep 17 00:00:00 2001
From: lloydzhou <lloydzhou@qq.com>
Date: Thu, 4 Jul 2024 19:38:26 +0800
Subject: [PATCH] hotfix

---
 app/utils/model.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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