109 lines
3.3 KiB
Bash
109 lines
3.3 KiB
Bash
|
#!/bin/bash
|
||
|
|
# Virtual Environment Health Check and Bootstrap Script
|
||
|
|
# Checks if Python venvs are compatible with current Python version
|
||
|
|
# Rebuilds venvs if needed
|
||
|
|
|
||
|
|
set -e
|
||
|
|
|
||
|
|
echo "=== Python Virtual Environment Health Check ==="
|
||
|
|
|
||
|
|
# Get current system Python version
|
||
|
|
SYSTEM_PYTHON=$(python3 --version | awk '{print $2}')
|
||
|
|
SYSTEM_PYTHON_MAJOR_MINOR=$(echo "$SYSTEM_PYTHON" | cut -d'.' -f1,2)
|
||
|
|
|
||
|
|
echo "System Python: $SYSTEM_PYTHON ($SYSTEM_PYTHON_MAJOR_MINOR)"
|
||
|
|
|
||
|
|
# List of venvs to check
|
||
|
|
VENVS=(
|
||
|
|
"/workspace/ai/vllm/venv"
|
||
|
|
"/workspace/ai/webdav-sync/venv"
|
||
|
|
"/workspace/ComfyUI/venv"
|
||
|
|
)
|
||
|
|
|
||
|
|
REBUILD_NEEDED=0
|
||
|
|
|
||
|
|
# Check each venv
|
||
|
|
for VENV_PATH in "${VENVS[@]}"; do
|
||
|
|
if [ ! -d "$VENV_PATH" ]; then
|
||
|
|
echo "⚠ venv not found: $VENV_PATH (will be created on first service start)"
|
||
|
|
continue
|
||
|
|
fi
|
||
|
|
|
||
|
|
VENV_NAME=$(basename $(dirname "$VENV_PATH"))
|
||
|
|
echo ""
|
||
|
|
echo "Checking venv: $VENV_NAME ($VENV_PATH)"
|
||
|
|
|
||
|
|
# Check if venv Python executable works
|
||
|
|
if ! "$VENV_PATH/bin/python" --version >/dev/null 2>&1; then
|
||
|
|
echo " ❌ BROKEN - Python executable not working"
|
||
|
|
REBUILD_NEEDED=1
|
||
|
|
continue
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Get venv Python version
|
||
|
|
VENV_PYTHON=$("$VENV_PATH/bin/python" --version 2>&1 | awk '{print $2}')
|
||
|
|
VENV_PYTHON_MAJOR_MINOR=$(echo "$VENV_PYTHON" | cut -d'.' -f1,2)
|
||
|
|
|
||
|
|
echo " venv Python: $VENV_PYTHON ($VENV_PYTHON_MAJOR_MINOR)"
|
||
|
|
|
||
|
|
# Compare major.minor versions
|
||
|
|
if [ "$SYSTEM_PYTHON_MAJOR_MINOR" != "$VENV_PYTHON_MAJOR_MINOR" ]; then
|
||
|
|
echo " ⚠ VERSION MISMATCH - System is $SYSTEM_PYTHON_MAJOR_MINOR, venv is $VENV_PYTHON_MAJOR_MINOR"
|
||
|
|
REBUILD_NEEDED=1
|
||
|
|
else
|
||
|
|
# Check if pip works
|
||
|
|
if ! "$VENV_PATH/bin/pip" --version >/dev/null 2>&1; then
|
||
|
|
echo " ❌ BROKEN - pip not working"
|
||
|
|
REBUILD_NEEDED=1
|
||
|
|
else
|
||
|
|
echo " ✓ HEALTHY"
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
|
||
|
|
# If any venv needs rebuild, warn the user
|
||
|
|
if [ $REBUILD_NEEDED -eq 1 ]; then
|
||
|
|
echo ""
|
||
|
|
echo "========================================"
|
||
|
|
echo " ⚠ WARNING: Some venvs need rebuilding"
|
||
|
|
echo "========================================"
|
||
|
|
echo ""
|
||
|
|
echo "One or more Python virtual environments are incompatible with the current"
|
||
|
|
echo "Python version or are broken. This can happen when:"
|
||
|
|
echo " - Docker image Python version changed"
|
||
|
|
echo " - venv files were corrupted"
|
||
|
|
echo " - Binary dependencies are incompatible"
|
||
|
|
echo ""
|
||
|
|
echo "RECOMMENDED ACTIONS:"
|
||
|
|
echo ""
|
||
|
|
echo "1. vLLM venv rebuild:"
|
||
|
|
echo " cd /workspace/ai/vllm"
|
||
|
|
echo " rm -rf venv"
|
||
|
|
echo " python3 -m venv venv"
|
||
|
|
echo " source venv/bin/activate"
|
||
|
|
echo " pip install -r requirements.txt"
|
||
|
|
echo ""
|
||
|
|
echo "2. ComfyUI venv rebuild:"
|
||
|
|
echo " cd /workspace/ComfyUI"
|
||
|
|
echo " rm -rf venv"
|
||
|
|
echo " python3 -m venv venv"
|
||
|
|
echo " source venv/bin/activate"
|
||
|
|
echo " pip install -r requirements.txt"
|
||
|
|
echo ""
|
||
|
|
echo "3. WebDAV sync venv rebuild (if used):"
|
||
|
|
echo " cd /workspace/ai/webdav-sync"
|
||
|
|
echo " rm -rf venv"
|
||
|
|
echo " python3 -m venv venv"
|
||
|
|
echo " source venv/bin/activate"
|
||
|
|
echo " pip install -r requirements.txt"
|
||
|
|
echo ""
|
||
|
|
echo "Services may fail to start until venvs are rebuilt!"
|
||
|
|
echo "========================================"
|
||
|
|
echo ""
|
||
|
|
else
|
||
|
|
echo ""
|
||
|
|
echo "✓ All virtual environments are healthy"
|
||
|
|
fi
|
||
|
|
|
||
|
|
exit 0
|