add litellm dependency

stanford-oval · Apr 10, 2024 · 7780e34 · 7780e34
1 parent c4e1b66
commit 7780e34
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
+    - FlagEmbedding==1.2.5
+    - litellm==1.34.34
diff --git 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