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

rename initial_sub_question_creation

Browse Source
This commit is contained in:
joachim-danswer 2025-01-30 11:41:43 -08:00 committed by Evan Lohn
parent 6442c56d82
commit 0a6808c4c1
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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"],
)

2
backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/nodes/initial_decomposition.py → backend/onyx/agents/agent_search/deep_search_a/initial/consolidate_sub_answers/nodes/decompose_orig_question.py
View File

@ -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()