ai-engine / Dockerfile
Severian's picture
Update Dockerfile
dac5c20 verified
raw
history blame
4.85 kB
# Base Python image with correct version
FROM python:3.12-slim-bookworm AS base
# Set shared environment variables
ENV POETRY_VERSION=1.8.4 \
POETRY_NO_INTERACTION=1 \
POETRY_VIRTUALENVS_CREATE=true \
POETRY_VIRTUALENVS_IN_PROJECT=true \
POETRY_CACHE_DIR=/tmp/poetry_cache \
PYTHONDONTWRITEBYTECODE=1
# Create directories with correct permissions first (as root)
RUN mkdir -p /app/api /app/web /data/storage /var/lib/postgresql/data && \
chown -R user:user /app /data && \
chmod 777 /data /app && \
chown -R postgres:postgres /var/lib/postgresql/data && \
chmod 700 /var/lib/postgresql/data
# Install system dependencies and create user
RUN apt-get update && apt-get install -y \
sudo \
curl \
git \
gcc \
python3-dev \
libgmp-dev \
libmpfr-dev \
libmpc-dev \
nodejs \
npm \
postgresql \
postgresql-contrib \
&& rm -rf /var/lib/apt/lists/* \
&& pip install --no-cache-dir "poetry==${POETRY_VERSION}" \
&& useradd -m -u 1000 user && \
usermod -aG sudo user && \
echo "user ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers && \
chown -R postgres:postgres /var/lib/postgresql
# Initialize PostgreSQL database as postgres user
USER postgres
RUN /usr/lib/postgresql/15/bin/initdb -D /var/lib/postgresql/data && \
echo "host all all 0.0.0.0/0 md5" >> /var/lib/postgresql/data/pg_hba.conf && \
echo "listen_addresses='*'" >> /var/lib/postgresql/data/postgresql.conf
# Switch to user for remaining operations
USER user
# Set environment for user
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
# Pull official images
FROM langgenius/dify-web:latest AS web
FROM langgenius/dify-api:latest AS api
# Final stage (continuing from line 32)
FROM base
# Set up directory structure
WORKDIR /app
RUN mkdir -p api web /data/storage && \
chown -R user:user /app /data
# Copy from official images with correct ownership
COPY --from=web --chown=user:user /app/web /app/web/
COPY --from=api --chown=user:user /app/api /app/api/
# Install API dependencies using Poetry
WORKDIR /app/api
COPY --from=api --chown=user /app/api/pyproject.toml /app/api/poetry.lock /app/api/poetry.toml ./
RUN poetry install --no-root --no-dev
# Create symlink for persistent storage
RUN ln -s /data/storage /app/api/storage
# Set environment variables
ENV FLASK_APP=app.py \
EDITION=SELF_HOSTED \
DEPLOY_ENV=PRODUCTION \
MODE=api \
LOG_LEVEL=INFO \
DEBUG=false \
FLASK_DEBUG=false \
SECRET_KEY=sk-9f73s3ljTXVcMT3Blb3ljTqtsKiGHXVcMT3BlbkFJLK7U \
CONSOLE_API_URL=http://127.0.0.1:7860 \
CONSOLE_WEB_URL=http://127.0.0.1:3000 \
SERVICE_API_URL=http://127.0.0.1:7860 \
APP_WEB_URL=http://127.0.0.1:3000 \
DIFY_PORT=7860 \
DIFY_BIND_ADDRESS=0.0.0.0 \
DB_USERNAME=postgres \
DB_PASSWORD=difyai123456 \
DB_HOST=localhost \
DB_PORT=5432 \
DB_DATABASE=dify \
REDIS_HOST=localhost \
REDIS_PORT=6379 \
REDIS_PASSWORD=difyai123456 \
CELERY_BROKER_URL=amqp://guest:guest@localhost:5672// \
CELERY_RESULT_BACKEND=redis://localhost:6379/0 \
PYTHONPATH=/app/api \
STORAGE_PATH=/data/storage
EXPOSE 7860 3000
# Create startup script with connection retries
RUN echo '#!/bin/bash\n\
echo "===== Application Startup at $(date "+%Y-%m-%d %H:%M:%S") ====="\n\
\n\
# Start PostgreSQL\n\
sudo -u postgres /usr/lib/postgresql/15/bin/pg_ctl -D /var/lib/postgresql/data start\n\
\n\
# Wait for PostgreSQL to start\n\
max_tries=30\n\
count=0\n\
echo "Checking database connection..."\n\
until PGPASSWORD=$DB_PASSWORD psql -h localhost -p 5432 -U postgres -c "SELECT 1" > /dev/null 2>&1; do\n\
echo "Waiting for database connection..."\n\
sleep 2\n\
count=$((count+1))\n\
if [ $count -gt $max_tries ]; then\n\
echo "Failed to connect to database after $max_tries attempts"\n\
exit 1\n\
fi\n\
done\n\
\n\
# Create database and user if they dont exist\n\
PGPASSWORD=$DB_PASSWORD psql -h localhost -U postgres -c "CREATE DATABASE $DB_DATABASE;" || true\n\
PGPASSWORD=$DB_PASSWORD psql -h localhost -U postgres -c "CREATE USER $DB_USERNAME WITH PASSWORD '\''$DB_PASSWORD'\'';" || true\n\
PGPASSWORD=$DB_PASSWORD psql -h localhost -U postgres -c "GRANT ALL PRIVILEGES ON DATABASE $DB_DATABASE TO $DB_USERNAME;" || true\n\
\n\
echo "Database connection successful"\n\
\n\
# Run database migrations\n\
cd /app/api && poetry run python -m flask db upgrade\n\
\n\
# Start API server\n\
cd /app/api && poetry run python -m gunicorn app:app \
--bind ${DIFY_BIND_ADDRESS:-0.0.0.0}:${DIFY_PORT:-7860} \
--worker-class gevent \
--workers 1 \
--timeout 300 \
--preload &\n\
\n\
# Start web server\n\
cd /app/web && node server.js &\n\
\n\
wait' > /app/entrypoint.sh && \
chmod +x /app/entrypoint.sh
WORKDIR /app
CMD ["./entrypoint.sh"]