Update agent.py

agent.py

@@ -52,10 +52,4 @@ def simple_graph():
     
     graph = builder.compile()
 
-    return graph
-
-
-
-if __name__ == "__main__":
-    # Build the graph
-    graph = simple_graph()
+    return graph