diff --git a/setup.py b/setup.py
index 74fda36..2a55124 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
     long_description = f.read()
 
 install_requires = [
-    "openai>=1.3", "diskcache", "termcolor", "pyautogen>=0.2.0", "eventlet",
+    "openai>=1.3", "diskcache", "termcolor", "ag2>=0.2.0", "eventlet",
     "gurobipy"
 ]