|
|
|
@@ -83,11 +83,15 @@ class DataPostProcessor: |
|
|
|
if reranking_model: |
|
|
|
try: |
|
|
|
model_manager = ModelManager() |
|
|
|
reranking_provider_name = reranking_model.get("reranking_provider_name") |
|
|
|
reranking_model_name = reranking_model.get("reranking_model_name") |
|
|
|
if not reranking_provider_name or not reranking_model_name: |
|
|
|
return None |
|
|
|
rerank_model_instance = model_manager.get_model_instance( |
|
|
|
tenant_id=tenant_id, |
|
|
|
provider=reranking_model["reranking_provider_name"], |
|
|
|
provider=reranking_provider_name, |
|
|
|
model_type=ModelType.RERANK, |
|
|
|
model=reranking_model["reranking_model_name"], |
|
|
|
model=reranking_model_name, |
|
|
|
) |
|
|
|
return rerank_model_instance |
|
|
|
except InvokeAuthorizationError: |