From 4a0b2a6c09d1ab4490b5dac7c2bc3932388f3ca0 Mon Sep 17 00:00:00 2001
From: joachim-danswer <joachim@danswer.ai>
Date: Thu, 30 Jan 2025 16:33:04 -0800
Subject: [PATCH] additional naming fixes

---
 .../agent_search/deep_search_a/main/edges.py       |  2 +-
 .../deep_search_a/main/graph_builder.py            | 14 +++++++-------
 ...refined_answer.py => decide_refinement_need.py} |  2 +-
 3 files changed, 9 insertions(+), 9 deletions(-)
 rename backend/onyx/agents/agent_search/deep_search_a/main/nodes/{validate_refined_answer.py => decide_refinement_need.py} (97%)

diff --git a/backend/onyx/agents/agent_search/deep_search_a/main/edges.py b/backend/onyx/agents/agent_search/deep_search_a/main/edges.py
index fed2c0127..70995ea46 100644
--- a/backend/onyx/agents/agent_search/deep_search_a/main/edges.py
+++ b/backend/onyx/agents/agent_search/deep_search_a/main/edges.py
@@ -97,7 +97,7 @@ def parallelize_refined_sub_question_answering(
     if len(state.refined_sub_questions) > 0:
         return [
             Send(
-                "answer_refined_question",
+                "answer_refined_question_subgraphs",
                 AnswerQuestionInput(
                     question=question_data.sub_question,
                     question_id=make_question_id(1, question_nr),
diff --git a/backend/onyx/agents/agent_search/deep_search_a/main/graph_builder.py b/backend/onyx/agents/agent_search/deep_search_a/main/graph_builder.py
index 8b4a08146..a121025bf 100644
--- a/backend/onyx/agents/agent_search/deep_search_a/main/graph_builder.py
+++ b/backend/onyx/agents/agent_search/deep_search_a/main/graph_builder.py
@@ -20,6 +20,9 @@ from onyx.agents.agent_search.deep_search_a.main.nodes.compare_answers import (
 from onyx.agents.agent_search.deep_search_a.main.nodes.create_refined_sub_questions import (
     create_refined_sub_questions,
 )
+from onyx.agents.agent_search.deep_search_a.main.nodes.decide_refinement_need import (
+    decide_refinement_need,
+)
 from onyx.agents.agent_search.deep_search_a.main.nodes.extract_entity_term import (
     extract_entity_term,
 )
@@ -35,9 +38,6 @@ from onyx.agents.agent_search.deep_search_a.main.nodes.persist_agent_results imp
 from onyx.agents.agent_search.deep_search_a.main.nodes.start_agent_search import (
     start_agent_search,
 )
-from onyx.agents.agent_search.deep_search_a.main.nodes.validate_refined_answer import (
-    validate_refined_answer,
-)
 from onyx.agents.agent_search.deep_search_a.main.states import MainInput
 from onyx.agents.agent_search.deep_search_a.main.states import MainState
 from onyx.agents.agent_search.deep_search_a.refinement.consolidate_sub_answers.graph_builder import (
@@ -119,8 +119,8 @@ def main_graph_builder(test_mode: bool = False) -> StateGraph:
         action=extract_entity_term,
     )
     graph.add_node(
-        node="validate_refined_answer",
-        action=validate_refined_answer,
+        node="decide_refinement_need",
+        action=decide_refinement_need,
     )
     graph.add_node(
         node="compare_answers",
@@ -167,11 +167,11 @@ def main_graph_builder(test_mode: bool = False) -> StateGraph:
 
     graph.add_edge(
         start_key=["generate_initial_answer_subgraph", "extract_entity_term"],
-        end_key="validate_refined_answer",
+        end_key="decide_refinement_need",
     )
 
     graph.add_conditional_edges(
-        source="validate_refined_answer",
+        source="decide_refinement_need",
         path=continue_to_refined_answer_or_end,
         path_map=["create_refined_sub_questions", "logging_node"],
     )
diff --git a/backend/onyx/agents/agent_search/deep_search_a/main/nodes/validate_refined_answer.py b/backend/onyx/agents/agent_search/deep_search_a/main/nodes/decide_refinement_need.py
similarity index 97%
rename from backend/onyx/agents/agent_search/deep_search_a/main/nodes/validate_refined_answer.py
rename to backend/onyx/agents/agent_search/deep_search_a/main/nodes/decide_refinement_need.py
index 9e8bec287..568815071 100644
--- a/backend/onyx/agents/agent_search/deep_search_a/main/nodes/validate_refined_answer.py
+++ b/backend/onyx/agents/agent_search/deep_search_a/main/nodes/decide_refinement_need.py
@@ -11,7 +11,7 @@ from onyx.agents.agent_search.deep_search_a.main.states import (
 from onyx.agents.agent_search.models import AgentSearchConfig
 
 
-def validate_refined_answer(
+def decide_refinement_need(
     state: MainState, config: RunnableConfig
 ) -> RequireRefinedAnswerUpdate:
     now_start = datetime.now()