Update app.py

app.py

@@ -76,14 +76,9 @@ class BasicAgent:
             answer = response['messages'][-1].content # First pass 
             #return answer.split("FINAL ANSWER:")[-1].strip() # Return for single pass
 
-            # Structure the messages
-            messages = [
-                {"role": "system", "content": sys_msg},
-                {"role": "user", "content": response}
-            ]
-            
             # Call the agent again
-            response = self.graph.invoke({"messages": messages})
+            print({"messages": [sys_msg] + [SystemMessage(content=response)]})
+            response = self.graph.invoke({"messages": [sys_msg] + [SystemMessage(content=response)]})
             return answer.split("FINAL ANSWER:")[-1].strip()