From 861b4c667ee952927204bda997d374ba8e68b8f8 Mon Sep 17 00:00:00 2001
From: user <70670632+stuxf@users.noreply.github.com>
Date: Fri, 3 Apr 2026 05:45:01 +0000
Subject: [PATCH] fix: restore proxy extras parity and venv migrations

---
 docker/entrypoint.sh | 19 +++++++++++--------
 pyproject.toml       |  2 +-
 2 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
index a028e542629..003d9b21db8 100755
--- a/docker/entrypoint.sh
+++ b/docker/entrypoint.sh
@@ -1,13 +1,16 @@
 #!/bin/bash
-echo $(pwd)
+set -euo pipefail
 
-# Run the Python migration script
-python3 litellm/proxy/prisma_migration.py
+REPO_ROOT="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")/.." && pwd)"
+VENV_PYTHON="$REPO_ROOT/.venv/bin/python"
+MIGRATION_SCRIPT="$REPO_ROOT/litellm/proxy/prisma_migration.py"
 
-# Check if the Python script executed successfully
-if [ $? -eq 0 ]; then
-    echo "Migration script ran successfully!"
+if [ -x "$VENV_PYTHON" ]; then
+    "$VENV_PYTHON" "$MIGRATION_SCRIPT"
+elif command -v uv >/dev/null 2>&1; then
+    (cd "$REPO_ROOT" && uv run --no-sync python "$MIGRATION_SCRIPT")
 else
-    echo "Migration script failed!"
-    exit 1
+    python3 "$MIGRATION_SCRIPT"
 fi
+
+echo "Migration script ran successfully!"
diff --git a/pyproject.toml b/pyproject.toml
index f24fcca9f51..6062ba3a9a5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -53,7 +53,7 @@ proxy = [
     "azure-identity==1.25.2; python_version >= '3.9'",
     "azure-storage-blob==12.28.0",
     "mcp==1.26.0; python_version >= '3.10'",
-    "litellm-proxy-extras==0.4.62",
+    "litellm-proxy-extras==0.4.63",
     "litellm-enterprise==0.1.35",
     "rich==13.9.4",
     "polars==1.38.1; python_version >= '3.10'",