hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-15 05:46:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix rerank model value is empty string (#7937)

Browse Source
This commit is contained in:
Jyong
2024-09-03 21:25:21 +08:00
committed by GitHub
parent 4b262cae58
commit d8b6c053a2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/rag/datasource/retrieval_service.py
View File

@@ -146,7 +146,7 @@ class RetrievalService:
) )
if documents: if documents:
if reranking_model and retrival_method == RetrievalMethod.SEMANTIC_SEARCH.value: if reranking_model and reranking_model.get('reranking_model_name') and reranking_model.get('reranking_provider_name') and retrival_method == RetrievalMethod.SEMANTIC_SEARCH.value:
data_post_processor = DataPostProcessor(str(dataset.tenant_id), data_post_processor = DataPostProcessor(str(dataset.tenant_id),
RerankMode.RERANKING_MODEL.value, RerankMode.RERANKING_MODEL.value,
reranking_model, None, False) reranking_model, None, False)
@@ -180,7 +180,7 @@ class RetrievalService:
top_k=top_k top_k=top_k
) )
if documents: if documents:
if reranking_model and retrival_method == RetrievalMethod.FULL_TEXT_SEARCH.value: if reranking_model and reranking_model.get('reranking_model_name') and reranking_model.get('reranking_provider_name') and retrival_method == RetrievalMethod.FULL_TEXT_SEARCH.value:
data_post_processor = DataPostProcessor(str(dataset.tenant_id), data_post_processor = DataPostProcessor(str(dataset.tenant_id),
RerankMode.RERANKING_MODEL.value, RerankMode.RERANKING_MODEL.value,
reranking_model, None, False) reranking_model, None, False)