|
@@ -146,7 +146,7 @@ class RetrievalService:
|
|
|
)
|
|
|
|
|
|
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),
|
|
|
RerankMode.RERANKING_MODEL.value,
|
|
|
reranking_model, None, False)
|
|
@@ -180,7 +180,7 @@ class RetrievalService:
|
|
|
top_k=top_k
|
|
|
)
|
|
|
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),
|
|
|
RerankMode.RERANKING_MODEL.value,
|
|
|
reranking_model, None, False)
|