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