diff --git a/agent/architect/graph.py b/agent/architect/graph.py index 1cbdf94..5bc63fd 100644 --- a/agent/architect/graph.py +++ b/agent/architect/graph.py @@ -126,12 +126,6 @@ def should_conduct_research(state: SoftwareArchitectState): else: return "plan_is_not_valid" -def call_model(state: SoftwareArchitectState): - response = plan_next_step_runnable.invoke({"atomic_implementation_research":state.implementation_research_scratchpad, - "codebase_structure": get_files_structure.invoke({"directory": "./workspace_repo"}), - "historical_actions": "No historical actions"}) - return {"implementation_research_scratchpad": [response]} - class SoftwareArchitectInput(TypedDict): implementation_research_scratchpad: List[AnyMessage] @@ -152,14 +146,13 @@ class SoftwareArchitectOutput(TypedDict): workflow.add_edge(START, "come_up_with_research_next_step") workflow.add_edge("come_up_with_research_next_step", "check_research_step") workflow.add_conditional_edges( - "check_research_step", + "check_research_step", should_conduct_research, { "plan_is_valid": "conduct_research", "plan_is_not_valid": "come_up_with_research_next_step" } ) -workflow.add_edge("check_research_step", "conduct_research") workflow.add_conditional_edges( "conduct_research", should_call_tool,