node renaming

backend/onyx/agents/agent_search/deep_search_a/shared/expanded_retrieval/edges.py

@@ -23,7 +23,7 @@ def parallel_retrieval_edge(
 
     return [
         Send(
-            "doc_retrieval",
+            "retrieve_documents",
             RetrievalInput(
                 query_to_retrieve=query,
                 question=question,

backend/onyx/agents/agent_search/deep_search_a/shared/expanded_retrieval/graph_builder.py

@@ -5,26 +5,26 @@ from langgraph.graph import StateGraph
 from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.edges import (
     parallel_retrieval_edge,
 )
-from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.doc_reranking import (
-    doc_reranking,
-)
-from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.doc_retrieval import (
-    doc_retrieval,
-)
-from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.doc_verification import (
-    doc_verification,
-)
-from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.dummy import (
-    dummy,
-)
 from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.expand_queries import (
     expand_queries,
 )
+from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.format_queries import (
+    format_queries,
+)
 from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.format_results import (
     format_results,
 )
-from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.verification_kickoff import (
+from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.kickoff_verification import (
-    verification_kickoff,
+    kickoff_verification,
+)
+from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.rerank_documents import (
+    rerank_documents,
+)
+from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.retrieve_documents import (
+    retrieve_documents,
+)
+from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.nodes.verify_documents import (
+    verify_documents,
 )
 from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.states import (
     ExpandedRetrievalInput,
@@ -57,24 +57,24 @@ def expanded_retrieval_graph_builder() -> StateGraph:
 
     graph.add_node(
         node="dummy",
-        action=dummy,
+        action=format_queries,
     )
 
     graph.add_node(
-        node="doc_retrieval",
+        node="retrieve_documents",
-        action=doc_retrieval,
+        action=retrieve_documents,
     )
     graph.add_node(
-        node="verification_kickoff",
+        node="kickoff_verification",
-        action=verification_kickoff,
+        action=kickoff_verification,
     )
     graph.add_node(
-        node="doc_verification",
+        node="verify_documents",
-        action=doc_verification,
+        action=verify_documents,
     )
     graph.add_node(
-        node="doc_reranking",
+        node="rerank_documents",
-        action=doc_reranking,
+        action=rerank_documents,
     )
     graph.add_node(
         node="format_results",
@@ -94,18 +94,18 @@ def expanded_retrieval_graph_builder() -> StateGraph:
     graph.add_conditional_edges(
         source="dummy",
         path=parallel_retrieval_edge,
-        path_map=["doc_retrieval"],
+        path_map=["retrieve_documents"],
     )
     graph.add_edge(
-        start_key="doc_retrieval",
+        start_key="retrieve_documents",
-        end_key="verification_kickoff",
+        end_key="kickoff_verification",
     )
     graph.add_edge(
-        start_key="doc_verification",
+        start_key="verify_documents",
-        end_key="doc_reranking",
+        end_key="rerank_documents",
     )
     graph.add_edge(
-        start_key="doc_reranking",
+        start_key="rerank_documents",
         end_key="format_results",
     )
     graph.add_edge(

backend/onyx/agents/agent_search/deep_search_a/shared/expanded_retrieval/nodes/format_queries.py

@@ -8,7 +8,7 @@ from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.states imp
 )
 
 
-def dummy(
+def format_queries(
     state: ExpandedRetrievalState, config: RunnableConfig
 ) -> QueryExpansionUpdate:
     return QueryExpansionUpdate(

backend/onyx/agents/agent_search/deep_search_a/shared/expanded_retrieval/nodes/kickoff_verification.py

@@ -12,10 +12,10 @@ from onyx.agents.agent_search.deep_search_a.shared.expanded_retrieval.states imp
 )
 
 
-def verification_kickoff(
+def kickoff_verification(
     state: ExpandedRetrievalState,
     config: RunnableConfig,
-) -> Command[Literal["doc_verification"]]:
+) -> Command[Literal["verify_documents"]]:
     documents = state.retrieved_documents
     verification_question = state.question
 
@@ -24,7 +24,7 @@ def verification_kickoff(
         update={},
         goto=[
             Send(
-                node="doc_verification",
+                node="verify_documents",
                 arg=DocVerificationInput(
                     doc_to_verify=doc,
                     question=verification_question,

backend/onyx/agents/agent_search/deep_search_a/shared/expanded_retrieval/nodes/rerank_documents.py

@@ -24,7 +24,7 @@ from onyx.context.search.postprocessing.postprocessing import rerank_sections
 from onyx.db.engine import get_session_context_manager
 
 
-def doc_reranking(
+def rerank_documents(
     state: ExpandedRetrievalState, config: RunnableConfig
 ) -> DocRerankingUpdate:
     now_start = datetime.now()

backend/onyx/agents/agent_search/deep_search_a/shared/expanded_retrieval/nodes/retrieve_documents.py

@@ -26,7 +26,9 @@ from onyx.tools.tool_implementations.search.search_tool import (
 from onyx.tools.tool_implementations.search.search_tool import SearchResponseSummary
 
 
-def doc_retrieval(state: RetrievalInput, config: RunnableConfig) -> DocRetrievalUpdate:
+def retrieve_documents(
+    state: RetrievalInput, config: RunnableConfig
+) -> DocRetrievalUpdate:
     """
     Retrieve documents
 

backend/onyx/agents/agent_search/deep_search_a/shared/expanded_retrieval/nodes/verify_documents.py

@@ -16,7 +16,7 @@ from onyx.agents.agent_search.shared_graph_utils.agent_prompt_ops import (
 from onyx.agents.agent_search.shared_graph_utils.prompts import VERIFIER_PROMPT
 
 
-def doc_verification(
+def verify_documents(
     state: DocVerificationInput, config: RunnableConfig
 ) -> DocVerificationUpdate:
     """