From 8c4fb7968d5fa844483f81b566f429d88261ae21 Mon Sep 17 00:00:00 2001
From: wxg0103 <727495428@qq.com>
Date: Thu, 17 Oct 2024 13:54:05 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8Dxinference=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E5=88=9B=E5=BB=BA=E5=A4=B1=E8=B4=A5=E7=9A=84=E7=BC=BA?=
 =?UTF-8?q?=E9=99=B7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

https://github.com/1Panel-dev/MaxKB/issues/1389
---
 .../impl/xinference_model_provider/xinference_model_provider.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/apps/setting/models_provider/impl/xinference_model_provider/xinference_model_provider.py b/apps/setting/models_provider/impl/xinference_model_provider/xinference_model_provider.py
index 0e432e061..ae2dc776c 100644
--- a/apps/setting/models_provider/impl/xinference_model_provider/xinference_model_provider.py
+++ b/apps/setting/models_provider/impl/xinference_model_provider/xinference_model_provider.py
@@ -407,4 +407,4 @@ class XinferenceModelProvider(IModelProvider):
     def get_model_info_by_name(model_list, model_name):
         if model_list is None:
             return []
-        return [model for model in model_list if model.get('model_name') == model_name]
+        return [model for model in model_list if model.get('model_name') == model_name or model.get('id') == model_name]