# Atlas Stack - generated stack definition
# Copy .env.example to .env before first run.
name: atlasstack

services:
  ollama:
    image: ollama/ollama:${OLLAMA_TAG:-latest}
    container_name: atlas-ollama
    ports: ["11434:11434"]
    volumes: ["ollama_data:/root/.ollama"]
    restart: unless-stopped
    healthcheck:
      test: ["CMD-SHELL", "ollama list >/dev/null 2>&1 || exit 1"]
      interval: 30s
      timeout: 10s
      retries: 5

  tgi:
    image: ghcr.io/huggingface/text-generation-inference:${TGI_TAG:-latest}
    container_name: atlas-tgi
    ports: ["8080:80"]
    volumes: ["tgi_data:/data"]
    restart: unless-stopped
    command: ["--model-id", "${TGI_MODEL:-Qwen/Qwen2.5-7B-Instruct}"]
    shm_size: "1g"

  vllm:
    image: vllm/vllm-openai:${VLLM_TAG:-latest}
    container_name: atlas-vllm
    ports: ["8000:8000"]
    volumes: ["vllm_data:/root/.cache/huggingface"]
    restart: unless-stopped
    command: ["--model", "${VLLM_MODEL:-Qwen/Qwen2.5-7B-Instruct}"]
    shm_size: "1g"

  localai:
    image: localai/localai:${LOCALAI_TAG:-latest}
    container_name: atlas-localai
    ports: ["8081:8080"]
    volumes: ["localai_data:/models"]
    restart: unless-stopped

  qdrant:
    image: qdrant/qdrant:${QDRANT_TAG:-latest}
    container_name: atlas-qdrant
    ports: ["6333:6333"]
    volumes: ["qdrant_data:/qdrant/storage"]
    restart: unless-stopped
    healthcheck:
      test: ["CMD-SHELL", "curl -fsS localhost:6333/ >/dev/null || exit 1"
      interval: 30s
      timeout: 10s
      retries: 5

  chroma:
    image: chromadb/chroma:${CHROMA_TAG:-latest}
    container_name: atlas-chroma
    ports: ["8002:8000"]
    volumes: ["chroma_data:/chroma/chroma"]
    restart: unless-stopped
    healthcheck:
      test: ["CMD-SHELL", "curl -fsS localhost:8000/ >/dev/null || exit 1"
      interval: 30s
      timeout: 10s
      retries: 5

  milvus:
    image: milvusdb/milvus:${MILVUS_TAG:-latest}
    container_name: atlas-milvus
    ports: ["19530:19530"]
    volumes: ["milvus_data:/var/lib/milvus"]
    restart: unless-stopped
    environment:
      ETCD_USE_EMBED: "true"
      COMMON_STORAGETYPE: local

volumes:
  ollama_data:
  tgi_data:
  vllm_data:
  localai_data:
  qdrant_data:
  chroma_data:
  milvus_data: