fix Reranking mode is null

api/core/rag/retrieval/dataset_retrieval.py

@@ -436,7 +436,8 @@ class DatasetRetrieval:
                                                           if retrieval_model['score_threshold_enabled'] else None,
                                                           reranking_model=retrieval_model.get('reranking_model', None)
                                                           if retrieval_model['reranking_enable'] else None,
-                                                          reranking_mode=retrieval_model.get('reranking_mode', 'reranking_model'),
+                                                          reranking_mode=retrieval_model.get('reranking_mode')
+                                                          if retrieval_model.get('reranking_mode') else 'reranking_model',
                                                           weights=retrieval_model.get('weights', None),
                                                           )
 

api/core/tools/tool/dataset_retriever/dataset_multi_retriever_tool.py

@@ -181,7 +181,8 @@ class DatasetMultiRetrieverTool(DatasetRetrieverBaseTool):
                                                           if retrieval_model['score_threshold_enabled'] else None,
                                                           reranking_model=retrieval_model.get('reranking_model', None)
                                                           if retrieval_model['reranking_enable'] else None,
-                                                          reranking_mode=retrieval_model.get('reranking_mode', 'reranking_model'),
+                                                          reranking_mode=retrieval_model.get('reranking_mode')
+                                                          if retrieval_model.get('reranking_mode') else 'reranking_model',
                                                           weights=retrieval_model.get('weights', None),
                                                           )
 

api/core/tools/tool/dataset_retriever/dataset_retriever_tool.py

@@ -79,7 +79,8 @@ class DatasetRetrieverTool(DatasetRetrieverBaseTool):
                                                       score_threshold=retrieval_model.get('score_threshold', .0)
                                                       if retrieval_model['score_threshold_enabled'] else None,
                                                       reranking_model=retrieval_model.get('reranking_model', None),
-                                                      reranking_mode=retrieval_model.get('reranking_mode', 'reranking_model'),
+                                                      reranking_mode=retrieval_model.get('reranking_mode')
+                                                      if retrieval_model.get('reranking_mode') else 'reranking_model',
                                                       weights=retrieval_model.get('weights', None),
                                                       )
             else:

api/services/hit_testing_service.py

@@ -45,7 +45,8 @@ class HitTestingService:
                                                   score_threshold=retrieval_model.get('score_threshold', .0)
                                                   if retrieval_model['score_threshold_enabled'] else None,
                                                   reranking_model=retrieval_model.get('reranking_model', None),
-                                                  reranking_mode=retrieval_model.get('reranking_mode', 'reranking_model'),
+                                                  reranking_mode=retrieval_model.get('reranking_mode')
+                                                  if retrieval_model.get('reranking_mode') else 'reranking_model',
                                                   weights=retrieval_model.get('weights', None),
                                                   )