AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 03:39:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
-LAN- 2024-12-21 21:23:12 +08:00 committed by GitHub
parent 5e37ab60d8
commit 599d410d99
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/core/rag/data_post_processor/data_post_processor.py
View File

@ -83,11 +83,15 @@ class DataPostProcessor:
if reranking_model: if reranking_model:
try: try:
model_manager = ModelManager() 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( rerank_model_instance = model_manager.get_model_instance(
tenant_id=tenant_id, tenant_id=tenant_id,
provider=reranking_model["reranking_provider_name"], provider=reranking_provider_name,
model_type=ModelType.RERANK, model_type=ModelType.RERANK,
model=reranking_model["reranking_model_name"], model=reranking_model_name,
) )
return rerank_model_instance return rerank_model_instance
except InvokeAuthorizationError: except InvokeAuthorizationError: