highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-07-15 07:33:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

more renamings

Browse Source
This commit is contained in:
joachim-danswer
2025-01-30 13:56:33 -08:00
committed by Evan Lohn
parent ceaaa05af0
commit 2517aa39b2
3 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/graph_builder.py
View File

@ -27,7 +27,7 @@ logger = setup_logger()
test_mode = False test_mode = False
def initial_sq_subgraph_builder(test_mode: bool = False) -> StateGraph: def consolidate_sub_answers_graph_builder() -> StateGraph:
graph = StateGraph( graph = StateGraph(
state_schema=SQState, state_schema=SQState,
input=SQInput, input=SQInput,

14
backend/onyx/agents/agent_search/deep_search_a/initial/generate_initial_answer/graph_builder.py
View File

@ -3,7 +3,7 @@ from langgraph.graph import START
from langgraph.graph import StateGraph from langgraph.graph import StateGraph
from onyx.agents.agent_search.deep_search_a.initial.consolidate_sub_answers.graph_builder import ( from onyx.agents.agent_search.deep_search_a.initial.consolidate_sub_answers.graph_builder import (
initial_sq_subgraph_builder, consolidate_sub_answers_graph_builder,
) )
from onyx.agents.agent_search.deep_search_a.initial.generate_initial_answer.nodes.consolidate_retrieved_documents import ( from onyx.agents.agent_search.deep_search_a.initial.generate_initial_answer.nodes.consolidate_retrieved_documents import (
consolidate_retrieved_documents, consolidate_retrieved_documents,
@ -28,7 +28,7 @@ from onyx.utils.logger import setup_logger
logger = setup_logger() logger = setup_logger()
def initial_search_sq_subgraph_builder(test_mode: bool = False) -> StateGraph: def generate_initial_answer_graph_builder(test_mode: bool = False) -> StateGraph:
graph = StateGraph( graph = StateGraph(
state_schema=SearchSQState, state_schema=SearchSQState,
input=SearchSQInput, input=SearchSQInput,
@ -39,10 +39,10 @@ def initial_search_sq_subgraph_builder(test_mode: bool = False) -> StateGraph:
# action=initial_sub_question_creation, # action=initial_sub_question_creation,
# ) # )
sub_question_answering_subgraph = initial_sq_subgraph_builder().compile() consolidate_sub_answers = consolidate_sub_answers_graph_builder().compile()
graph.add_node( graph.add_node(
node="sub_question_answering_subgraph", node="consolidate_sub_answers",
action=sub_question_answering_subgraph, action=consolidate_sub_answers,
) )
# answer_query_subgraph = answer_query_graph_builder().compile() # answer_query_subgraph = answer_query_graph_builder().compile()
@ -88,11 +88,11 @@ def initial_search_sq_subgraph_builder(test_mode: bool = False) -> StateGraph:
graph.add_edge( graph.add_edge(
start_key=START, start_key=START,
end_key="sub_question_answering_subgraph", end_key="consolidate_sub_answers",
) )
graph.add_edge( graph.add_edge(
start_key=["base_raw_search_subgraph", "sub_question_answering_subgraph"], start_key=["base_raw_search_subgraph", "consolidate_sub_answers"],
end_key="retrieval_consolidation", end_key="retrieval_consolidation",
) )

12
backend/onyx/agents/agent_search/deep_search_a/main/graph_builder.py
View File

@ -3,7 +3,7 @@ from langgraph.graph import START
from langgraph.graph import StateGraph from langgraph.graph import StateGraph
from onyx.agents.agent_search.deep_search_a.initial.generate_initial_answer.graph_builder import ( from onyx.agents.agent_search.deep_search_a.initial.generate_initial_answer.graph_builder import (
initial_search_sq_subgraph_builder, generate_initial_answer_graph_builder,
) )
from onyx.agents.agent_search.deep_search_a.main.edges import ( from onyx.agents.agent_search.deep_search_a.main.edges import (
continue_to_refined_answer_or_end, continue_to_refined_answer_or_end,
@ -106,10 +106,10 @@ def main_graph_builder(test_mode: bool = False) -> StateGraph:
# action=initial_sub_question_creation, # action=initial_sub_question_creation,
# ) # )
initial_search_sq_subgraph = initial_search_sq_subgraph_builder().compile() generate_initial_answer_graph = generate_initial_answer_graph_builder().compile()
graph.add_node( graph.add_node(
node="initial_search_sq_subgraph", node="generate_initial_answer_graph",
action=initial_search_sq_subgraph, action=generate_initial_answer_graph,
) )
# answer_query_subgraph = answer_query_graph_builder().compile() # answer_query_subgraph = answer_query_graph_builder().compile()
@ -239,7 +239,7 @@ def main_graph_builder(test_mode: bool = False) -> StateGraph:
graph.add_edge( graph.add_edge(
start_key="agent_search_start", start_key="agent_search_start",
end_key="initial_search_sq_subgraph", end_key="generate_initial_answer_graph",
) )
# graph.add_edge( # graph.add_edge(
# start_key="agent_search_start", # start_key="agent_search_start",
@ -316,7 +316,7 @@ def main_graph_builder(test_mode: bool = False) -> StateGraph:
# ) # )
graph.add_edge( graph.add_edge(
start_key=["initial_search_sq_subgraph", "entity_term_extraction_llm"], start_key=["generate_initial_answer_graph", "entity_term_extraction_llm"],
end_key="refined_answer_decision", end_key="refined_answer_decision",
) )