Преглед изворни кода

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

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/0.14.2
-LAN- пре 10 месеци
родитељ
комит
599d410d99
No account linked to committer's email address
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:

Loading…
Откажи
Сачувај