Update agent.py

agent.py

@@ -209,4 +209,5 @@ if __name__ == "__main__":
     # Run the graph
     messages = [HumanMessage(content=question)]
     messages = graph.invoke({"messages": messages})
-    for m in messages["me
+     for m in messages["messages"]:
+        m.pretty_print()