From 70d3b827487fb99d5719e43e1d5edaf3ca203cc9 Mon Sep 17 00:00:00 2001
From: Eric Zhu <ekzhu@users.noreply.github.com>
Date: Sat, 18 Jan 2025 00:59:03 -0800
Subject: [PATCH] Fixed autogen package

---
 conda/conda-reqs-pip.txt | 2 +-
 requirements-all.txt     | 4 ++--
 setup.py                 | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/conda/conda-reqs-pip.txt b/conda/conda-reqs-pip.txt
index 9aefd408..832713c5 100644
--- a/conda/conda-reqs-pip.txt
+++ b/conda/conda-reqs-pip.txt
@@ -8,5 +8,5 @@ sumologic-sdk>=0.1.11
 splunk-sdk>=1.6.0,!=2.0.0
 packaging>=24.0
 requests>=2.31.0
-pyautogen[retrievechat]>=0.2.28
+autogen-agentchat[retrievechat]~=0.2.0
 importlib-resources >= 6.4.0; python_version <= "3.8"
diff --git a/requirements-all.txt b/requirements-all.txt
index 83e721a4..58d4417e 100644
--- a/requirements-all.txt
+++ b/requirements-all.txt
@@ -44,7 +44,7 @@ packaging>=24.0
 pandas>=1.4.0, <3.0.0
 panel>=0.14.4
 passivetotal>=2.5.3
-pyautogen[retrievechat]>=0.2.28
+autogen-agentchat[retrievechat]~=0.2.0
 pydantic>=1.8.0, <3.0.0
 pygments>=2.0.0
 pyjwt>=2.3.0
@@ -63,4 +63,4 @@ typing-extensions>=4.2.0
 urllib3>=1.23
 vt-py>=0.18.0
 vt-graph-api>=2.0
-requests>=2.31.0
\ No newline at end of file
+requests>=2.31.0
diff --git a/setup.py b/setup.py
index a5af0011..993ac249 100644
--- a/setup.py
+++ b/setup.py
@@ -53,7 +53,7 @@ def _combine_extras(extras: list) -> list:
     "sql2kql": ["mo-sql-parsing>=8, <9.0.0"],
     "riskiq": ["passivetotal>=2.5.3", "requests>=2.31.0"],
     "panel": ["panel>=0.14.4"],
-    "aiagents": ["pyautogen[retrievechat]>=0.2.28"],
+    "aiagents": ["autogen-agentchat[retrievechat]~=0.2.0"],
 }
 extras_all = [
     extra for name, extras in EXTRAS.items() for extra in extras if name != "dev"