Udate app.py

app.py

@@ -549,8 +549,31 @@ def tool_node(state: AgentState) -> AgentState:
     print(f"DEBUG: Exiting tool_node. Tool output added to history. New history length: {len(state['history'])}")
     return state
 
+
 # ====== Agent Graph ======
-# ... (no changes needed here)
+def create_agent_workflow(tools: List[BaseTool]): # Use BaseTool for consistency
+    workflow = StateGraph(AgentState)
+    
+    workflow.add_node("reason", reasoning_node)
+    workflow.add_node("action", tool_node)
+    
+    workflow.set_entry_point("reason")
+    
+    workflow.add_conditional_edges(
+        "reason",
+        should_continue,
+        {
+            "action": "action",
+            "reason": "reason",
+            "end": END
+        }
+    )
+    
+    workflow.add_edge("action", "reason")
+    
+    app = workflow.compile()
+    return app
+
 
 # ====== Agent Interface ======
 class BasicAgent: