From cbad5c948fbf659cb1f48e653e508d91eef04dfa Mon Sep 17 00:00:00 2001
From: OlivierBinette-Upstart <olivier.binette@upstart.com>
Date: Thu, 31 Jul 2025 19:01:28 +0000
Subject: [PATCH 1/4] unpin pydantic

---
 flaml/autogen/agentchat/contrib/math_user_proxy_agent.py | 8 +++++++-
 setup.py                                                 | 2 +-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/flaml/autogen/agentchat/contrib/math_user_proxy_agent.py b/flaml/autogen/agentchat/contrib/math_user_proxy_agent.py
index e90bb1f533..5d243aa17e 100644
--- a/flaml/autogen/agentchat/contrib/math_user_proxy_agent.py
+++ b/flaml/autogen/agentchat/contrib/math_user_proxy_agent.py
@@ -3,7 +3,13 @@
 from time import sleep
 from typing import Any, Callable, Dict, List, Optional, Union
 
-from pydantic import BaseModel, Extra, root_validator
+import pydantic
+from packaging import version
+
+if version(pydantic.__version__) < version.parse("2.0.0"):
+    from pydantic import BaseModel, Extra, root_validator
+else:
+    from pydantic.v1 import BaseModel, Extra, root_validator
 
 from flaml.autogen.agentchat import Agent, UserProxyAgent
 from flaml.autogen.code_utils import UNKNOWN, execute_code, extract_code, infer_lang
diff --git a/setup.py b/setup.py
index 31cc563726..d1b58e0022 100644
--- a/setup.py
+++ b/setup.py
@@ -90,7 +90,7 @@
             "tensorboardX==2.6",  # test_forecast_panel
             "requests<2.29.0",  # https://github.com/docker/docker-py/issues/3113
             "packaging",
-            "pydantic==1.10.9",
+            "pydantic",
             "sympy",
             "wolframalpha",
             "dill",  # a drop in replacement of pickle

From ff4acdb4605c857621ea375099248b8d0a655431 Mon Sep 17 00:00:00 2001
From: Olivier Binette <olivier.binette@upstart.com>
Date: Mon, 4 Aug 2025 12:19:19 -0400
Subject: [PATCH 2/4] Update setup.py

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index d1b58e0022..5c41a63018 100644
--- a/setup.py
+++ b/setup.py
@@ -104,7 +104,7 @@
             "packaging",
         ],
         "ray": [
-            "ray[tune]~=1.13",
+            "ray[tune]>=2.9",
         ],
         "azureml": [
             "azureml-mlflow",

From 5627ef517708daff7a0c9863a5a4e5ffb6d0d8c5 Mon Sep 17 00:00:00 2001
From: Olivier Binette <olivier.binette@upstart.com>
Date: Tue, 5 Aug 2025 10:11:29 -0400
Subject: [PATCH 3/4] Update setup.py

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 5c41a63018..8a911b2ab2 100644
--- a/setup.py
+++ b/setup.py
@@ -90,7 +90,7 @@
             "tensorboardX==2.6",  # test_forecast_panel
             "requests<2.29.0",  # https://github.com/docker/docker-py/issues/3113
             "packaging",
-            "pydantic",
+            "pydantic==1.10.9",
             "sympy",
             "wolframalpha",
             "dill",  # a drop in replacement of pickle
@@ -104,7 +104,7 @@
             "packaging",
         ],
         "ray": [
-            "ray[tune]>=2.9",
+            "ray[tune]>=1.13",
         ],
         "azureml": [
             "azureml-mlflow",

From 7019bfad068d5bf6c4c0f62b90781ac85d34300c Mon Sep 17 00:00:00 2001
From: Olivier Binette <olivier.binette@upstart.com>
Date: Tue, 5 Aug 2025 10:17:28 -0400
Subject: [PATCH 4/4] Update setup.py

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 8a911b2ab2..ad6e1977dd 100644
--- a/setup.py
+++ b/setup.py
@@ -90,7 +90,7 @@
             "tensorboardX==2.6",  # test_forecast_panel
             "requests<2.29.0",  # https://github.com/docker/docker-py/issues/3113
             "packaging",
-            "pydantic==1.10.9",
+            "pydantic>=1.10.9",
             "sympy",
             "wolframalpha",
             "dill",  # a drop in replacement of pickle