diff --git a/backend/onyx/agents/agent_search/deep_search_a/initial_search_sq_subgraph/nodes/generate_initial_answer.py b/backend/onyx/agents/agent_search/deep_search_a/initial_search_sq_subgraph/nodes/generate_initial_answer.py index 38f10f79f..d40829142 100644 --- a/backend/onyx/agents/agent_search/deep_search_a/initial_search_sq_subgraph/nodes/generate_initial_answer.py +++ b/backend/onyx/agents/agent_search/deep_search_a/initial_search_sq_subgraph/nodes/generate_initial_answer.py @@ -79,6 +79,28 @@ def generate_initial_answer( ) decomp_questions = [] + # Use the query info from the base document retrieval + query_info = get_query_info(state.original_question_retrieval_results) + if agent_a_config.search_tool is None: + raise ValueError("search_tool must be provided for agentic search") + for tool_response in yield_search_responses( + query=question, + reranked_sections=relevant_docs, + final_context_sections=relevant_docs, + search_query_info=query_info, + get_section_relevance=lambda: None, # TODO: add relevance + search_tool=agent_a_config.search_tool, + ): + dispatch_custom_event( + "tool_response", + ExtendedToolResponse( + id=tool_response.id, + response=tool_response.response, + level=0, + level_question_nr=0, # 0, 0 is the base question + ), + ) + if len(relevant_docs) == 0: dispatch_custom_event( "initial_agent_answer", @@ -99,28 +121,6 @@ def generate_initial_answer( ) else: - # Use the query info from the base document retrieval - query_info = get_query_info(state.original_question_retrieval_results) - if agent_a_config.search_tool is None: - raise ValueError("search_tool must be provided for agentic search") - for tool_response in yield_search_responses( - query=question, - reranked_sections=relevant_docs, - final_context_sections=relevant_docs, - search_query_info=query_info, - get_section_relevance=lambda: None, # TODO: add relevance - search_tool=agent_a_config.search_tool, - ): - dispatch_custom_event( - "tool_response", - ExtendedToolResponse( - id=tool_response.id, - response=tool_response.response, - level=0, - level_question_nr=0, # 0, 0 is the base question - ), - ) - net_new_original_question_docs = [] for all_original_question_doc in all_original_question_documents: if all_original_question_doc not in sub_question_docs: