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'",