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