diff --git a/conda_env.yml b/conda_env.yml
index 4a18a2a..1b4e64c 100644
--- a/conda_env.yml
+++ b/conda_env.yml
@@ -15,4 +15,5 @@ dependencies:
     - tiktoken==0.4.0
     - psycopg2-binary==2.9.7  # you can also install from source if it works
     - pglast==5.3
-    - FlagEmbedding==1.2.5
\ No newline at end of file
+    - FlagEmbedding==1.2.5
+    - litellm==1.34.34
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 19005a0..7313b8c 100644
--- a/setup.py
+++ b/setup.py
@@ -24,6 +24,7 @@
     'psycopg2-binary==2.9.7',
     'pglast==5.3',
     'FlagEmbedding==1.2.5',
+    'litellm==1.34.34'
 ]
 
 # Additional package information