Parcourir la source

fix: validate reranking model attributes before processing (#11930)

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/0.14.2
-LAN- il y a 10 mois
Parent
révision
599d410d99
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 6 ajouts et 2 suppressions
  1. 6
    2
      api/core/rag/data_post_processor/data_post_processor.py

+ 6
- 2
api/core/rag/data_post_processor/data_post_processor.py Voir le fichier

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

Chargement…
Annuler
Enregistrer