diff --git a/pyproject.toml b/pyproject.toml
index 8ce9c280a..4f718a9ed 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,6 +26,7 @@ click = "*"
 jinja2 = "^3.1.2"
 aiohttp = "*"
 requests = "^2.31.0"
+pydantic = "^2.0.0"
 
 uvicorn = {version = "^0.22.0", optional = true}
 gunicorn = {version = "^22.0.0", optional = true}
diff --git a/requirements.txt b/requirements.txt
index 07dc7e347..4b4ba17a4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -43,5 +43,5 @@ certifi==2023.7.22 # [TODO] clean up
 aiohttp==3.9.0 # for network calls
 aioboto3==12.3.0 # for async sagemaker calls
 tenacity==8.2.3  # for retrying requests, when litellm.num_retries set
-pydantic==2.7.1 # openai req. 
+pydantic==2.7.1 # proxy + openai req.
 ####
\ No newline at end of file