Update app.py

app.py

@@ -118,6 +118,7 @@ class BasicAgent:
         except Exception as e:
             print(f"Agent error: {e}", flush=True)
             return ""
+            
     def assistant(self, state: AgentState):
         sys_msg = SystemMessage(content="""
         You are a precise question-answering agent.
@@ -260,6 +261,13 @@ class BasicAgent:
     
         return "final_process"
     
+    def route_after_assistant(self, state: AgentState):
+        last_message = state["messages"][-1]
+    
+        if getattr(last_message, "tool_calls", None):
+            return "tools"
+    
+        return "final_process"
     
     def retry_with_feedback(self, state: AgentState):
         notes = state.get("verification_notes", "")