From ee61e5af65dfd92a5e44ba53511a8789c927a7d3 Mon Sep 17 00:00:00 2001 From: Jessica Singh Date: Thu, 11 Sep 2025 18:52:05 -0700 Subject: [PATCH] add query for fed slack --- .../dr/sub_agents/basic_search/dr_basic_search_2_act.py | 1 + backend/onyx/tools/tool_implementations/search/search_tool.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/backend/onyx/agents/agent_search/dr/sub_agents/basic_search/dr_basic_search_2_act.py b/backend/onyx/agents/agent_search/dr/sub_agents/basic_search/dr_basic_search_2_act.py index 08192936842..bf067cac36a 100644 --- a/backend/onyx/agents/agent_search/dr/sub_agents/basic_search/dr_basic_search_2_act.py +++ b/backend/onyx/agents/agent_search/dr/sub_agents/basic_search/dr_basic_search_2_act.py @@ -152,6 +152,7 @@ def basic_search( alternate_db_session=search_db_session, retrieved_sections_callback=callback_container.append, skip_query_analysis=True, + original_query=rewritten_query, ), ): # get retrieved docs to send to the rest of the graph diff --git a/backend/onyx/tools/tool_implementations/search/search_tool.py b/backend/onyx/tools/tool_implementations/search/search_tool.py index e275592b6eb..a0b84df8660 100644 --- a/backend/onyx/tools/tool_implementations/search/search_tool.py +++ b/backend/onyx/tools/tool_implementations/search/search_tool.py @@ -293,7 +293,7 @@ def run( self, override_kwargs: SearchToolOverrideKwargs | None = None, **llm_kwargs: Any ) -> Generator[ToolResponse, None, None]: query = cast(str, llm_kwargs[QUERY_FIELD]) - original_query = None + original_query = query precomputed_query_embedding = None precomputed_is_keyword = None precomputed_keywords = None @@ -312,7 +312,7 @@ def run( kg_sources = None kg_chunk_id_zero_only = False if override_kwargs: - original_query = override_kwargs.original_query + original_query = override_kwargs.original_query or query precomputed_is_keyword = override_kwargs.precomputed_is_keyword precomputed_keywords = override_kwargs.precomputed_keywords precomputed_query_embedding = override_kwargs.precomputed_query_embedding