Skip to content

Commit 5ba5b43

Browse files
Updated reranking override
1 parent b5b8278 commit 5ba5b43

File tree

1 file changed

+8
-0
lines changed
  • backend/onyx/agents/agent_search/deep_search/shared/expanded_retrieval/nodes

1 file changed

+8
-0
lines changed

backend/onyx/agents/agent_search/deep_search/shared/expanded_retrieval/nodes/rerank_documents.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@
2424
from onyx.context.search.models import RerankingDetails
2525
from onyx.context.search.postprocessing.postprocessing import rerank_sections
2626
from onyx.context.search.postprocessing.postprocessing import reranking_is_runnable
27+
from onyx.db.engine import get_session_context_manager
28+
from onyx.db.search_settings import get_current_search_settings
2729

2830

2931
def rerank_documents(
@@ -51,6 +53,12 @@ def rerank_documents(
5153

5254
rerank_settings = graph_config.inputs.search_request.rerank_settings
5355

56+
with get_session_context_manager() as db_session:
57+
if rerank_settings is None:
58+
search_settings = get_current_search_settings(db_session)
59+
if not search_settings.disable_rerank_for_streaming:
60+
rerank_settings = RerankingDetails.from_db_model(search_settings)
61+
5462
if reranking_is_runnable(rerank_settings) and len(verified_documents) > 0:
5563
if len(verified_documents) > 1:
5664
reranked_documents = rerank_sections(

0 commit comments

Comments
 (0)