diff --git a/Dockerfile b/Dockerfile
index ee825353d66..47d625e7687 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -13,7 +13,7 @@ WORKDIR /app
 USER root
 
 # Install build dependencies
-RUN apk add --no-cache bash gcc openssl openssl-dev
+RUN apk add --no-cache bash gcc git openssl openssl-dev
 
 RUN python -m pip install --upgrade pip setuptools && \
     python -m pip install build wheel cmake setuptools_rust maturin
diff --git a/docker/Dockerfile.database b/docker/Dockerfile.database
index a6fcd98ab6d..4aaaed3f212 100644
--- a/docker/Dockerfile.database
+++ b/docker/Dockerfile.database
@@ -1,8 +1,8 @@
 # Base image for building
-ARG LITELLM_BUILD_IMAGE=cgr.dev/chainguard/wolfi-base
+ARG LITELLM_BUILD_IMAGE=cgr.dev/chainguard/python:latest-dev
 
 # Runtime image
-ARG LITELLM_RUNTIME_IMAGE=cgr.dev/chainguard/wolfi-base
+ARG LITELLM_RUNTIME_IMAGE=cgr.dev/chainguard/python:latest-dev
 # Builder stage
 FROM $LITELLM_BUILD_IMAGE AS builder
 
@@ -15,9 +15,7 @@ USER root
 RUN apk add --no-cache \
     bash \
     gcc \
-    py3-pip \
-    python3 \
-    python3-dev \
+    git \
     openssl \
     openssl-dev