version: "3.8"
services:
  cognita-db:
    image: postgres:13
    container_name: cognita-postgres
    restart: unless-stopped
    ports:
      - "${POSTGRES_PORT}:5432"
    volumes:
      - ./volumes/pgdata:/var/lib/postgresql/data
    environment:
      POSTGRES_USER: ${POSTGRES_USER}
      POSTGRES_PASSWORD: ${POSTGRES_PASSWORD}
      POSTGRES_DB: cognita-config
    healthcheck:
      test:
        [
          "CMD-SHELL",
          "sh -c 'pg_isready -U ${POSTGRES_USER} -d cognita-config'",
        ]
      interval: 30s
      timeout: 60s
      retries: 5
      start_period: 15s
    networks:
      - cognita-docker

  ollama-server:
    image: ollama/ollama:0.1.42
    pull_policy: if_not_present
    restart: unless-stopped
    container_name: ollama
    profiles:
      - ollama
    volumes:
      - ./volumes/ollama:/root/.ollama
    ports:
      - 11434:11434
    healthcheck:
      test: ["CMD", "ollama", "list"]
      interval: 30s
      timeout: 5s
      retries: 3
      start_period: 60s
    environment:
      - OLLAMA_KEEP_ALIVE=24h
      - OLLAMA_HOST=0.0.0.0
      - OLLAMA_PORT=11434
    entrypoint: /bin/bash
    command: -c "set -e; ollama serve & sleep 10 && ollama pull ${OLLAMA_MODEL} && sleep infinity"
    networks:
      - cognita-docker

  infinity-server:
    image: michaelf34/infinity:0.0.63
    pull_policy: if_not_present
    restart: unless-stopped
    container_name: infinity
    profiles:
      - infinity
    ports:
      - "7997:7997"
    volumes:
      - ./volumes/infinity:/app/.cache
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:7997/health"]
      interval: 30s
      timeout: 5s
      retries: 10
      start_period: 30s
    environment:
      - INFINITY_MODEL_ID=${INFINITY_EMBEDDING_MODEL};${INFINITY_RERANKING_MODEL}
      - INFINITY_BATCH_SIZE=8
      - INFINITY_API_KEY=${INFINITY_API_KEY}
    command: v2
    networks:
      - cognita-docker

  qdrant-server:
    image: qdrant/qdrant:v1.8.4
    pull_policy: if_not_present
    restart: unless-stopped
    container_name: qdrant
    ports:
      - 6333:6333
      - 6334:6334
    expose:
      - 6333
      - 6334
      - 6335
    healthcheck:
      test: ["CMD-SHELL", "/bin/bash -c ':> /dev/tcp/0.0.0.0/6333'"]
      interval: 30s
      timeout: 5s
      retries: 3
      start_period: 10s
    volumes:
      - ./volumes/qdrant_storage:/qdrant/storage:z
    networks:
      - cognita-docker

  unstructured-io-parsers:
    # Docs: http://localhost:9500/general/docs
    image: downloads.unstructured.io/unstructured-io/unstructured-api:0.0.73
    pull_policy: if_not_present
    restart: unless-stopped
    container_name: unstructured-api
    ports:
      - 9500:9500
    expose:
      - 9500
    environment:
      - PORT=9500
    healthcheck:
      test:
        [
          "CMD",
          "wget",
          "-O",
          "/dev/null",
          "-o",
          "/dev/null",
          "http://localhost:9500/healthcheck",
        ]
      interval: 30s
      timeout: 5s
      retries: 3
      start_period: 5s
    networks:
      - cognita-docker

  # https://github.com/fedirz/faster-whisper-server/tree/master
  faster-whisper:
    image: fedirz/faster-whisper-server:latest-cpu
    pull_policy: if_not_present
    container_name: faster-whisper
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
      interval: 30s
      timeout: 5s
      retries: 3
      start_period: 10s
    environment:
      - WHISPER__MODEL=${WHISPER_MODEL}
      - WHISPER__INFERENCE_DEVICE=auto
    ports:
      - ${WHISPER_PORT}:8000
    volumes:
      - ./volumes/hugging_face_cache:/root/.cache/huggingface
    networks:
      - cognita-docker

  cognita-backend:
    build:
      context: .
      dockerfile: ./backend/Dockerfile
    restart: unless-stopped
    container_name: cognita-backend
    ports:
      - "${COGNITA_BACKEND_PORT}:8000"
    depends_on:
      cognita-db:
        condition: service_healthy
        restart: true
      qdrant-server:
        condition: service_healthy
        restart: true
      unstructured-io-parsers:
        condition: service_healthy
        restart: true
    volumes:
      - .:/app
      - ./volumes/pgdata:/var/lib/postgresql/data
      - ./volumes/user_data:/app/user_data
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8000/health-check"]
      interval: 30s
      timeout: 5s
      retries: 3
      start_period: 10s
    environment:
      - DEBUG_MODE=true
      - LOCAL=${LOCAL}
      - PROCESS_POOL_WORKERS=${PROCESS_POOL_WORKERS}
      - LOG_LEVEL=DEBUG
      - ML_REPO_NAME=${ML_REPO_NAME}
      - DATABASE_URL=postgresql://${POSTGRES_USER}:${POSTGRES_PASSWORD}@cognita-db:5432/cognita-config
      - METADATA_STORE_CONFIG=${METADATA_STORE_CONFIG}
      - VECTOR_DB_CONFIG=${VECTOR_DB_CONFIG}
      - LOCAL_DATA_DIRECTORY=/app/user_data
      - UNSTRUCTURED_IO_URL=${UNSTRUCTURED_IO_URL}
      - UNSTRUCTURED_IO_API_KEY=${UNSTRUCTURED_IO_API_KEY}
      - INFINITY_API_KEY=${INFINITY_API_KEY}
      - MODELS_CONFIG_PATH=${MODELS_CONFIG_PATH}
      - TFY_HOST=${TFY_HOST}
      - TFY_API_KEY=${TFY_API_KEY}
      - OPENAI_API_KEY=${OPENAI_API_KEY}
      - BRAVE_API_KEY=${BRAVE_API_KEY}
    entrypoint: /bin/bash
    security_opt:
      - seccomp:./seccomp.json
    command: -c "set -e; prisma db push --schema ./backend/database/schema.prisma && uvicorn --host 0.0.0.0 --port 8000 backend.server.app:app --reload"
    networks:
      - cognita-docker

  cognita-frontend:
    build:
      context: ./frontend
      dockerfile: ./Dockerfile.dev
      args:
        - VITE_QA_FOUNDRY_URL=${VITE_QA_FOUNDRY_URL}
        - VITE_DOCS_QA_DELETE_COLLECTIONS=${VITE_DOCS_QA_DELETE_COLLECTIONS}
        - VITE_DOCS_QA_STANDALONE_PATH=${VITE_DOCS_QA_STANDALONE_PATH}
        - VITE_DOCS_QA_ENABLE_REDIRECT=${VITE_DOCS_QA_ENABLE_REDIRECT}
        - VITE_DOCS_QA_MAX_UPLOAD_SIZE_MB=${VITE_DOCS_QA_MAX_UPLOAD_SIZE_MB}
        - VITE_USE_LOCAL=${LOCAL}
    restart: unless-stopped
    container_name: cognita-frontend
    ports:
      - "${COGNITA_FRONTEND_PORT}:5001"
    volumes:
      - ./frontend:/app
      - /app/node_modules
    depends_on:
      cognita-backend:
        condition: service_healthy
        restart: true
    entrypoint: /bin/bash
    command: -c "set -e; yarn install --frozen-lockfile && yarn dev --host 0.0.0.0"
    networks:
      - cognita-docker

networks:
  cognita-docker:
    external: false