version: '3.8'

services:
  ollama:
    image: ollama/ollama:latest
    container_name: ollama
    ports:
      - "11434:11434"
    volumes:
      - ./data:/root/.ollama
    environment:
      - OLLAMA_HOST=0.0.0.0
      - OLLAMA_NOPRUNE=1  # 保留部分下载，重试时更可能断点续传
      - OLLAMA_MAX_LOADED_MODELS=1  # 仅保留 1 个模型在内存，其他小模型不常驻，节省内存
    restart: unless-stopped

  olares-ollama:
    build: .
    container_name: olares-ollama
    ports:
      - "8080:8080"
    environment:
      - OLLAMA_MODEL=llama2
      - OLLAMA_URL=http://ollama:11434
      - PORT=8080
      - DOWNLOAD_TIMEOUT=120  # 2 hours download timeout
      - APP_URL=https://xx.userid.olares.cn/  # API access URL (optional)
    depends_on:
      - ollama
    restart: unless-stopped

# Using local data directory, no need to define volumes