File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
backend/onyx/agents/agent_search/deep_search/shared/expanded_retrieval/nodes Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change 2424from onyx .context .search .models import RerankingDetails
2525from onyx .context .search .postprocessing .postprocessing import rerank_sections
2626from 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
2931def 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 (
You can’t perform that action at this time.
0 commit comments