highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-08 11:58:34 +02:00
Code Issues Packages Projects Releases Wiki Activity

use reranking settings and persona during preprocessing in reranker

Browse Source
This commit is contained in:
Evan Lohn 2025-01-28 15:44:57 -08:00
parent 9bad12968f
commit efa32a8c04
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
backend/onyx/agents/agent_search/deep_search_a/expanded_retrieval/nodes/doc_reranking.py
View File

@ -36,8 +36,17 @@ def doc_reranking(
if agent_a_config.search_tool is None:
raise ValueError("search_tool must be provided for agentic search")
with get_session_context_manager() as db_session:
# we ignore some of the user specified fields since this search is
# internal to agentic search, but we still want to pass through
# persona (for stuff like document sets) and rerank settings
# (to not make an unnecessary db call).
search_request = SearchRequest(
query=question,
persona=agent_a_config.search_request.persona,
rerank_settings=agent_a_config.search_request.rerank_settings,
)
_search_query = retrieval_preprocessing(
search_request=SearchRequest(query=question),
search_request=search_request,
user=agent_a_config.search_tool.user, # bit of a hack
llm=agent_a_config.fast_llm,
db_session=db_session,