diff --git a/pyproject.toml b/pyproject.toml
index ebaea52..b1315bd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,7 +19,7 @@ dependencies = [
   "traitlets>=5.14.1",
   "ipdb>=0.13.13",
   "ipython>=8.18.1",
-  "litellm>=1.37.9",
+  "litellm==1.55.9",
   "PyYAML>=6.0.1", 
   "ipyflow>=0.0.130",
   "numpy>=1.26.3",