From 7a65daf13de6a689acd953ed0b8bcc92086431c9 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 10 Jan 2024 15:15:49 +0530
Subject: [PATCH] (ci/cd) set litellm as entrypoint

---
 Dockerfile.database | 5 ++++-
 entrypoint.sh       | 2 --
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/Dockerfile.database b/Dockerfile.database
index 6791dc1a2..48b45ccdd 100644
--- a/Dockerfile.database
+++ b/Dockerfile.database
@@ -54,4 +54,7 @@ RUN chmod +x entrypoint.sh
 EXPOSE 4000/tcp
 
 # Set your entrypoint and command
-CMD ["./entrypoint.sh"]
\ No newline at end of file
+CMD ["./entrypoint.sh"]
+
+# this allows accepting litellm args
+ENTRYPOINT ["litellm", "--port", "4000"]
diff --git a/entrypoint.sh b/entrypoint.sh
index 9071a5249..80adf8d07 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -46,5 +46,3 @@ fi
 
 echo "Database push successful!"
 
-# Start server
-litellm --port 4000 --num_workers 8
\ No newline at end of file