diff --git a/pyproject.toml b/pyproject.toml
index 2b237d75426..403d7e97b97 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -65,7 +65,7 @@ litellm = ["litellm>=1.36.1"]
 llamaindex = ["llama-index>=0.10.35"]
 mistral0 = ["mistralai>=0.1.8,<1.0.0"]
 mistral1 = ["mistralai>=1.0.0"]
-notdiamond = ["notdiamond>=0.3.21"]
+notdiamond = ["notdiamond>=0.3.21", "litellm<=1.49.1"]
 openai = ["openai>=1.0.0"]
 modal = ["modal", "python-dotenv"]
 test = [