fix edit docker file ui base path

build_admin_ui.sh

@@ -7,56 +7,68 @@
 echo
 pwd
 
+# if UI_BASE_PATH env is set 
+if [ -z "$UI_BASE_PATH" ]; then
 
-# only run this step for litellm enterprise, we run this if enterprise/enterprise_ui/_enterprise.json exists
-if [ ! -f "enterprise/enterprise_ui/enterprise_colors.json" ]; then
+
+# only run this step for litellm enterprise, we run this if enterprise/enterprise_ui/_enterprise.json exists or env var UI_BASE_PATH is set
+if [ -f "enterprise/enterprise_ui/enterprise_colors.json" ] || [ -n "${UI_BASE_PATH:-}" ]; then
+    echo "Building Admin UI..."
+
+    # Install dependencies
+    # Check if we are on macOS
+    if [[ "$(uname)" == "Darwin" ]]; then
+        # Install dependencies using Homebrew
+        if ! command -v brew &> /dev/null; then
+            echo "Error: Homebrew not found. Please install Homebrew and try again."
+            exit 1
+        fi
+        brew update
+        brew install curl
+    else
+        # Assume Linux, try using apt-get
+        if command -v apt-get &> /dev/null; then
+            apt-get update
+            apt-get install -y curl
+        elif command -v apk &> /dev/null; then
+            # Try using apk if apt-get is not available
+            apk update
+            apk add curl
+        else
+            echo "Error: Unsupported package manager. Cannot install dependencies."
+            exit 1
+        fi
+    fi
+    curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.38.0/install.sh | bash
+    source ~/.nvm/nvm.sh
+    nvm install v18.17.0
+    nvm use v18.17.0
+    npm install -g npm
+
+    if [ -n "${UI_BASE_PATH:-}" ]; then
+        echo "Using UI_BASE_PATH: $UI_BASE_PATH"
+        
+        # make a file call .env in ui/litellm-dashboard and store the UI_BASE_PATH in it
+        echo "UI_BASE_PATH=$UI_BASE_PATH" > ui/litellm-dashboard/.env
+
+    fi
+
+    # copy _enterprise.json from this directory to /ui/litellm-dashboard, and rename it to ui_colors.json
+    cp enterprise/enterprise_ui/enterprise_colors.json ui/litellm-dashboard/ui_colors.json
+
+    # cd in to /ui/litellm-dashboard
+    cd ui/litellm-dashboard
+
+    # ensure have access to build_ui.sh
+    chmod +x ./build_ui.sh
+
+    # run ./build_ui.sh
+    ./build_ui.sh
+
+    # return to root directory
+    cd ../..
+
+else
     echo "Admin UI - using default LiteLLM UI"
     exit 0
 fi
-
-echo "Building Custom Admin UI..."
-
-# Install dependencies
-# Check if we are on macOS
-if [[ "$(uname)" == "Darwin" ]]; then
-    # Install dependencies using Homebrew
-    if ! command -v brew &> /dev/null; then
-        echo "Error: Homebrew not found. Please install Homebrew and try again."
-        exit 1
-    fi
-    brew update
-    brew install curl
-else
-    # Assume Linux, try using apt-get
-    if command -v apt-get &> /dev/null; then
-        apt-get update
-        apt-get install -y curl
-    elif command -v apk &> /dev/null; then
-        # Try using apk if apt-get is not available
-        apk update
-        apk add curl
-    else
-        echo "Error: Unsupported package manager. Cannot install dependencies."
-        exit 1
-    fi
-fi
-curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.38.0/install.sh | bash
-source ~/.nvm/nvm.sh
-nvm install v18.17.0
-nvm use v18.17.0
-npm install -g npm
-
-# copy _enterprise.json from this directory to /ui/litellm-dashboard, and rename it to ui_colors.json
-cp enterprise/enterprise_ui/enterprise_colors.json ui/litellm-dashboard/ui_colors.json
-
-# cd in to /ui/litellm-dashboard
-cd ui/litellm-dashboard
-
-# ensure have access to build_ui.sh
-chmod +x ./build_ui.sh
-
-# run ./build_ui.sh
-./build_ui.sh
-
-# return to root directory
-cd ../..