zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
shaohuzhang1 2024-09-20 19:27:08 +08:00 committed by shaohuzhang1
parent a980604617
commit b8d1512eca
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/setting/models_provider/impl/local_model_provider/model/embedding.py
View File

@ -35,7 +35,7 @@ class WebLocalEmbedding(MaxKBBaseModel, BaseModel, Embeddings):
result = res.json() result = res.json()
if result.get('code', 500) == 200: if result.get('code', 500) == 200:
return result.get('data') return result.get('data')
raise Exception(result.get('msg')) raise Exception(result.get('message'))
def embed_documents(self, texts: List[str]) -> List[List[float]]: def embed_documents(self, texts: List[str]) -> List[List[float]]:
bind = f'{CONFIG.get("LOCAL_MODEL_HOST")}:{CONFIG.get("LOCAL_MODEL_PORT")}' bind = f'{CONFIG.get("LOCAL_MODEL_HOST")}:{CONFIG.get("LOCAL_MODEL_PORT")}'
@ -44,7 +44,7 @@ class WebLocalEmbedding(MaxKBBaseModel, BaseModel, Embeddings):
result = res.json() result = res.json()
if result.get('code', 500) == 200: if result.get('code', 500) == 200:
return result.get('data') return result.get('data')
raise Exception(result.get('msg')) raise Exception(result.get('message'))
class LocalEmbedding(MaxKBBaseModel, HuggingFaceEmbeddings): class LocalEmbedding(MaxKBBaseModel, HuggingFaceEmbeddings):

2
apps/setting/models_provider/impl/local_model_provider/model/reranker.py
View File

@ -61,7 +61,7 @@ class WebLocalBaseReranker(MaxKBBaseModel, BaseDocumentCompressor):
if result.get('code', 500) == 200: if result.get('code', 500) == 200:
return [Document(page_content=document.get('page_content'), metadata=document.get('metadata')) for document return [Document(page_content=document.get('page_content'), metadata=document.get('metadata')) for document
in result.get('data')] in result.get('data')]
raise Exception(result.get('msg')) raise Exception(result.get('message'))
class LocalBaseReranker(MaxKBBaseModel, BaseDocumentCompressor): class LocalBaseReranker(MaxKBBaseModel, BaseDocumentCompressor):