Update agents.py

agents.py

@@ -132,4 +132,15 @@ def create_agent_flow(provider: str = "groq"):
     graph.add_conditional_edges("assistant", tools_condition)
     graph.add_edge("tools", "assistant")
 
-    return graph.compile()
+    return graph.compile()
+
+
+if __name__ == "__main__":
+    question = "What is the capital of France?"
+    # Build the graph
+    graph = create_agent_flow(provider="groq")
+    # Run the graph
+    messages = [HumanMessage(content=question)]
+    messages = graph.invoke({"messages": messages})
+    for m in messages["messages"]:
+        m.pretty_print()