fix: 修复本地模型加载失败时,错误信息未输出

1Panel-dev · Sep 20, 2024 · b8d1512 · b8d1512
1 parent a980604
commit b8d1512
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 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
@@ -35,7 +35,7 @@ def embed_query(self, text: str) -> List[float]:
         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 @@ def embed_documents(self, texts: List[str]) -> List[List[float]]:
         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
@@ -61,7 +61,7 @@ def compress_documents(self, documents: Sequence[Document], query: str, callback
         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):