version: "3.8"

services:
  litellm:
    image: ghcr.io/berriai/litellm:main-latest
    container_name: litellm
    restart: unless-stopped
    ports:
      - "${LITELLM_PORT:-4000}:4000"
    volumes:
      - litellm_config:/app/config
    environment:
      - LITELLM_MASTER_KEY=${LITELLM_MASTER_KEY:-sk-master-key}
      - LITELLM_LOG_LEVEL=${LOG_LEVEL:-INFO}
      - DATABASE_URL=postgresql://${PG_USER:-litellm}:${PG_PASSWORD:-litellm}@litellm-db:5432/${PG_DB:-litellm}
    command: --config /app/config/litellm_config.yaml --port 4000
    depends_on:
      - litellm-db

  litellm-db:
    image: postgres:16-alpine
    container_name: litellm-db
    restart: unless-stopped
    environment:
      - POSTGRES_USER=${PG_USER:-litellm}
      - POSTGRES_PASSWORD=${PG_PASSWORD:-litellm}
      - POSTGRES_DB=${PG_DB:-litellm}
    volumes:
      - litellm_pg_data:/var/lib/postgresql/data

volumes:
  litellm_config:
  litellm_pg_data: