highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-19 12:03:54 +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
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: if len(state.refined_sub_questions) > 0:
return [ return [
Send( Send(
"answer_refined_question", "answer_refined_question_subgraphs",
AnswerQuestionInput( AnswerQuestionInput(
question=question_data.sub_question, question=question_data.sub_question,
question_id=make_question_id(1, question_nr), 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 ( from onyx.agents.agent_search.deep_search_a.main.nodes.create_refined_sub_questions import (
create_refined_sub_questions, 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 ( from onyx.agents.agent_search.deep_search_a.main.nodes.extract_entity_term import (
extract_entity_term, 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 ( from onyx.agents.agent_search.deep_search_a.main.nodes.start_agent_search import (
start_agent_search, 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 MainInput
from onyx.agents.agent_search.deep_search_a.main.states import MainState 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 ( 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, action=extract_entity_term,
) )
graph.add_node( graph.add_node(
node="validate_refined_answer", node="decide_refinement_need",
action=validate_refined_answer, action=decide_refinement_need,
) )
graph.add_node( graph.add_node(
node="compare_answers", node="compare_answers",
@@ -167,11 +167,11 @@ def main_graph_builder(test_mode: bool = False) -> StateGraph:
graph.add_edge( graph.add_edge(
start_key=["generate_initial_answer_subgraph", "extract_entity_term"], start_key=["generate_initial_answer_subgraph", "extract_entity_term"],
end_key="validate_refined_answer", end_key="decide_refinement_need",
) )
graph.add_conditional_edges( graph.add_conditional_edges(
source="validate_refined_answer", source="decide_refinement_need",
path=continue_to_refined_answer_or_end, path=continue_to_refined_answer_or_end,
path_map=["create_refined_sub_questions", "logging_node"], 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 from onyx.agents.agent_search.models import AgentSearchConfig
def validate_refined_answer( def decide_refinement_need(
state: MainState, config: RunnableConfig state: MainState, config: RunnableConfig
) -> RequireRefinedAnswerUpdate: ) -> RequireRefinedAnswerUpdate:
now_start = datetime.now() now_start = datetime.now()