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