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..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
@@ -104,7 +104,7 @@
             "packaging",
         ],
         "ray": [
-            "ray[tune]~=1.13",
+            "ray[tune]>=1.13",
         ],
         "azureml": [
             "azureml-mlflow",