From d4d6e9dc746b7997e4f85acdb3adefeda7d20228 Mon Sep 17 00:00:00 2001
From: AG2AI-Admin <AG2AI-Admin@users.noreply.github.com>
Date: Fri, 18 Jul 2025 12:36:19 -0400
Subject: [PATCH] Migrate from pyautogen to ag2 library

---
 poetry.lock    | 6 +++---
 pyproject.toml | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index 939147d..e2bc414 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1171,14 +1171,14 @@ files = [
 ]
 
 [[package]]
-name = "pyautogen"
+name = "ag2"
 version = "0.1.7"
 description = "Enabling Next-Gen LLM Applications via Multi-Agent Conversation Framework"
 optional = false
 python-versions = ">=3.8"
 files = [
-    {file = "pyautogen-0.1.7-py3-none-any.whl", hash = "sha256:4b8ffbb995b6adbaad51245017199c2140b5517e9bbd292f8c064610e9302614"},
-    {file = "pyautogen-0.1.7.tar.gz", hash = "sha256:08bd461c24a431303c51c1d141a044c9654ce2ba4c36543d0e7bba5df32487ad"},
+    {file = "ag2-0.1.7-py3-none-any.whl", hash = "sha256:4b8ffbb995b6adbaad51245017199c2140b5517e9bbd292f8c064610e9302614"},
+    {file = "ag2-0.1.7.tar.gz", hash = "sha256:08bd461c24a431303c51c1d141a044c9654ce2ba4c36543d0e7bba5df32487ad"},
 ]
 
 [package.dependencies]
diff --git a/pyproject.toml b/pyproject.toml
index 1fbd10b..8431a36 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,7 +12,7 @@ openai = "^1.2.3"
 psycopg2-binary = "^2.9.8"
 argparse = "^1.4.0"
 python-dotenv = "^1.0.0"
-pyautogen = "^0.1.7"
+ag2 = "^0.1.7"
 transformers = "^4.34.1"
 torch = "^2.1.0"
 scikit-learn = "^1.3.1"
@@ -25,7 +25,7 @@ requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.scripts]
-# Not available in current version due to pyautogen not supporting openai ^1.2.3
+# Not available in current version due to ag2 not supporting openai ^1.2.3
 # old_start = "postgres_da_ai_agent.main:main"
 start = "postgres_da_ai_agent.turbo_main:main"
 turbo = "postgres_da_ai_agent.turbo_main:main"