highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-05-20 16:51:06 +02:00
Code Issues Packages Projects Releases Wiki Activity

additional naming fixes

Browse Source
This commit is contained in:
joachim-danswer 2025-01-30 16:33:04 -08:00 committed by Evan Lohn
parent 756a1cbf8f
commit 4a0b2a6c09
3 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/onyx/agents/agent_search/deep_search_a/main/edges.py
View File

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

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

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

2
backend/onyx/agents/agent_search/deep_search_a/main/nodes/validate_refined_answer.py → backend/onyx/agents/agent_search/deep_search_a/main/nodes/decide_refinement_need.py
View File

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