Update agents.py

agents.py

@@ -125,14 +125,14 @@ def create_agent_flow(provider: str = "groq"):
 
     
     # Build the conversational graph
-    graph = StateGraph(MessagesState)
-    graph.add_node("assistant", assistant_node)
-    graph.add_node("tools", ToolNode(toolset))
-    graph.add_edge(START, "assistant")
-    graph.add_conditional_edges("assistant", tools_condition)
-    graph.add_edge("tools", "assistant")
+    graph01 = StateGraph(MessagesState)
+    graph01.add_node("assistant", assistant_node)
+    graph01.add_node("tools", ToolNode(toolset))
+    graph01.add_edge(START, "assistant")
+    graph01.add_conditional_edges("assistant", tools_condition)
+    graph01.add_edge("tools", "assistant")
 
-    return graph.compile()
+    return graph01.compile()
 
 
 if __name__ == "__main__":
@@ -144,3 +144,4 @@ if __name__ == "__main__":
     messages = graph.invoke({"messages": messages})
     for m in messages["messages"]:
         m.pretty_print()
+