From 0a6808c4c1f5d08487fd6760bf1feb01dbf84335 Mon Sep 17 00:00:00 2001 From: joachim-danswer Date: Thu, 30 Jan 2025 11:41:43 -0800 Subject: [PATCH] rename initial_sub_question_creation --- .../consolidate_sub_answers/graph_builder.py | 14 +++++++------- ...decomposition.py => decompose_orig_question.py} | 2 +- 2 files changed, 8 insertions(+), 8 deletions(-) rename backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/nodes/{initial_decomposition.py => decompose_orig_question.py} (99%) diff --git a/backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/graph_builder.py b/backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/graph_builder.py index 45e8c4827..55691facb 100644 --- a/backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/graph_builder.py +++ b/backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/graph_builder.py @@ -5,12 +5,12 @@ from langgraph.graph import StateGraph from onyx.agents.agent_search.deep_search_a.initial.consolidate_sub_answers.edges import ( parallelize_initial_sub_question_answering, ) +from onyx.agents.agent_search.deep_search_a.initial.consolidate_sub_answers.nodes.decompose_orig_question import ( + decompose_orig_question, +) from onyx.agents.agent_search.deep_search_a.initial.consolidate_sub_answers.nodes.ingest_initial_sub_answers import ( ingest_initial_sub_answers, ) -from onyx.agents.agent_search.deep_search_a.initial.consolidate_sub_answers.nodes.initial_decomposition import ( - initial_sub_question_creation, -) from onyx.agents.agent_search.deep_search_a.initial.consolidate_sub_answers.states import ( SQInput, ) @@ -34,8 +34,8 @@ def initial_sq_subgraph_builder(test_mode: bool = False) -> StateGraph: ) graph.add_node( - node="initial_sub_question_creation", - action=initial_sub_question_creation, + node="decompose_orig_question", + action=decompose_orig_question, ) answer_query_subgraph = answer_query_graph_builder().compile() graph.add_node( @@ -59,7 +59,7 @@ def initial_sq_subgraph_builder(test_mode: bool = False) -> StateGraph: graph.add_edge( start_key=START, - end_key="initial_sub_question_creation", + end_key="decompose_orig_question", ) # graph.add_edge( @@ -73,7 +73,7 @@ def initial_sq_subgraph_builder(test_mode: bool = False) -> StateGraph: # ) graph.add_conditional_edges( - source="initial_sub_question_creation", + source="decompose_orig_question", path=parallelize_initial_sub_question_answering, path_map=["answer_query_subgraph"], ) diff --git a/backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/nodes/initial_decomposition.py b/backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/nodes/decompose_orig_question.py similarity index 99% rename from backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/nodes/initial_decomposition.py rename to backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/nodes/decompose_orig_question.py index fcbe72f7d..bfb059508 100644 --- a/backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/nodes/initial_decomposition.py +++ b/backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/nodes/decompose_orig_question.py @@ -34,7 +34,7 @@ from onyx.chat.models import SubQuestionPiece from onyx.configs.agent_configs import AGENT_NUM_DOCS_FOR_DECOMPOSITION -def initial_sub_question_creation( +def decompose_orig_question( state: SearchSQState, config: RunnableConfig ) -> BaseDecompUpdate: now_start = datetime.now()