services:
  postgres:
    image: postgres:16-alpine
    container_name: litellm-postgres
    environment:
      POSTGRES_DB: litellm
      POSTGRES_USER: litellm
      POSTGRES_PASSWORD: litellm
    ports:
      - "5432:5432"
    volumes:
      - pgdata:/var/lib/postgresql/data

  litellm:
    image: litellm/litellm:main-latest
    container_name: litellm-proxy
    depends_on:
      - postgres
    ports:
      - "4000:4000"
    env_file: .env
    environment:
      DATABASE_URL: postgresql://litellm:litellm@postgres:5432/litellm
      PYTHONPATH: /app
    volumes:
      - ./config/litellm_config.yaml:/app/config.yaml:ro
      - ./config/genai.pem:/app/config/genai.pem:ro
      - ./src/custom_callbacks.py:/app/custom_callbacks.py:ro
      - ./src/ledger.py:/app/ledger.py:ro
      - ./storage:/app/storage      
    command: ["--config", "/app/config.yaml", "--port", "4000", "--host", "0.0.0.0"]

volumes:
  pgdata: