From b8d1512eca70e799a43cb16a1ecfb8af837a6651 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Fri, 20 Sep 2024 19:27:08 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=9C=AC=E5=9C=B0?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E5=8A=A0=E8=BD=BD=E5=A4=B1=E8=B4=A5=E6=97=B6?= =?UTF-8?q?,=E9=94=99=E8=AF=AF=E4=BF=A1=E6=81=AF=E6=9C=AA=E8=BE=93?= =?UTF-8?q?=E5=87=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../impl/local_model_provider/model/embedding.py | 4 ++-- .../impl/local_model_provider/model/reranker.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/apps/setting/models_provider/impl/local_model_provider/model/embedding.py b/apps/setting/models_provider/impl/local_model_provider/model/embedding.py index 38cf1aeb9..820b93e5f 100644 --- a/apps/setting/models_provider/impl/local_model_provider/model/embedding.py +++ b/apps/setting/models_provider/impl/local_model_provider/model/embedding.py @@ -35,7 +35,7 @@ class WebLocalEmbedding(MaxKBBaseModel, BaseModel, Embeddings): result = res.json() if result.get('code', 500) == 200: return result.get('data') - raise Exception(result.get('msg')) + raise Exception(result.get('message')) def embed_documents(self, texts: List[str]) -> List[List[float]]: bind = f'{CONFIG.get("LOCAL_MODEL_HOST")}:{CONFIG.get("LOCAL_MODEL_PORT")}' @@ -44,7 +44,7 @@ class WebLocalEmbedding(MaxKBBaseModel, BaseModel, Embeddings): result = res.json() if result.get('code', 500) == 200: return result.get('data') - raise Exception(result.get('msg')) + raise Exception(result.get('message')) class LocalEmbedding(MaxKBBaseModel, HuggingFaceEmbeddings): diff --git a/apps/setting/models_provider/impl/local_model_provider/model/reranker.py b/apps/setting/models_provider/impl/local_model_provider/model/reranker.py index 4ada1eb35..f5056b2cc 100644 --- a/apps/setting/models_provider/impl/local_model_provider/model/reranker.py +++ b/apps/setting/models_provider/impl/local_model_provider/model/reranker.py @@ -61,7 +61,7 @@ class WebLocalBaseReranker(MaxKBBaseModel, BaseDocumentCompressor): if result.get('code', 500) == 200: return [Document(page_content=document.get('page_content'), metadata=document.get('metadata')) for document in result.get('data')] - raise Exception(result.get('msg')) + raise Exception(result.get('message')) class LocalBaseReranker(MaxKBBaseModel, BaseDocumentCompressor):