From afc1b2209ba80040c810540bd6f8c6693e9e304d Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 30 Nov 2023 14:49:21 -0800
Subject: [PATCH] (feat) docker - allow user to specify base image

---
 Dockerfile | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 480ee680e..90edfcdf7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,8 @@
 # Base image
-FROM python:3.9-slim
+ARG LITELLM_BASE_IMAGE=python:3.9-slim
+
+# allow users to specify, else use python 3.9-slim
+FROM $LITELLM_BASE_IMAGE
 
 # Set the working directory to /app
 WORKDIR /app