فهرست منبع

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

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 4 ماه پیش
والد
کامیت
599d410d99
1فایلهای تغییر یافته به همراه6 افزوده شده و 2 حذف شده
  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

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