diff --git a/pyproject.toml b/pyproject.toml
index a894fa5..5fcf6a2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,7 +9,7 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = ">=3.11,<3.12"
 chromadb = "^0.5.0"
-pyautogen = "^0.2.28"
+ag2 = "^0.2.28"
 python-dotenv = "^1.0.1"
 xgboost = "^2.0.3"
 markdownify = "^0.12.1"