From 7780e34535ca86ae3dd2f8bfef3efe60d7f07a69 Mon Sep 17 00:00:00 2001
From: Shicheng Liu <liushicheng_george@outlook.com>
Date: Wed, 10 Apr 2024 21:18:10 +0000
Subject: [PATCH] add `litellm` dependency

---
 conda_env.yml | 3 ++-
 setup.py      | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

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