Update agents.py

agents.py

@@ -137,11 +137,16 @@ def create_agent_flow(provider: str = "groq"):
 
 if __name__ == "__main__":
     question = "What is the capital of France?"
+
     # Build the graph
-    graph = create_agent_flow(provider="groq")
+    compiled_graph = create_agent_flow(provider="groq")
+
+    # Prepare input messages
+    messages = [system_message, HumanMessage(content=question)]
+
     # Run the graph
-    messages = [HumanMessage(content=question)]
-    messages = graph.invoke({"messages": messages})
-    for m in messages["messages"]:
-        m.pretty_print()
+    output_state = compiled_graph.invoke({"messages": messages})
 
+    # Print the final output
+    for m in output_state["messages"]:
+        print(m.content)