docker-compose.yml

volumes:
  postgres_storage:
  ollama_storage:
  qdrant_storage:

networks:
  poc:

services:
  # ollama-nvidia:
  #   image: ollama/ollama:latest
  #   container_name: ollama
  #   networks: ['poc']
  #   restart: unless-stopped
  #   ports:
  #     - 11434:11434
  #   volumes:
  #     - ollama_storage:/root/.ollama
  #   deploy:
  #     resources:
  #       reservations:
  #         devices:
  #           - driver: nvidia
  #             count: 1
  #             capabilities: [gpu]

  ollama:
    image: ollama/ollama:latest
    container_name: ollama
    networks: ['poc']
    restart: unless-stopped
    ports:
      - 11434:11434
    volumes:
      - ollama_storage:/root/.ollama

  # init-ollama-text-model:
  #   image: ollama/ollama:latest
  #   networks: ['poc']
  #   container_name: init-ollama-text-model
  #   volumes:
  #     - ollama_storage:/root/.ollama
  #   entrypoint: /bin/sh
  #   command:
  #     - "-c"
  #     - "sleep 3; OLLAMA_HOST=ollama:11434 ollama pull llama3.1:8b"

  # init-ollama-text-model:
  #   image: ollama/ollama:latest
  #   networks: ['poc']
  #   container_name: init-ollama-text-model
  #   volumes:
  #     - ollama_storage:/root/.ollama
  #   entrypoint: /bin/sh
  #   command:
  #     - "-c"
  #     - "sleep 3; OLLAMA_HOST=ollama:11434 ollama pull llama3.2:3b"

  init-ollama-text-model:
    image: ollama/ollama:latest
    networks: ['poc']
    container_name: init-ollama-text-model
    volumes:
      - ollama_storage:/root/.ollama
    entrypoint: /bin/sh
    command:
      - "-c"
      - "sleep 3; OLLAMA_HOST=ollama:11434 ollama pull mistral-nemo:12b"

  # init-ollama-embedding-model:
  #   image: ollama/ollama:latest
  #   networks: ['poc']
  #   container_name: init-ollama-embedding-model
  #   volumes:
  #     - ollama_storage:/root/.ollama
  #   entrypoint: /bin/sh
  #   command:
  #     - "-c"
  #     - "sleep 3; OLLAMA_HOST=ollama:11434 ollama pull nomic-embed-text"

  init-ollama-embedding-model:
    image: ollama/ollama:latest
    networks: ['poc']
    container_name: init-ollama-embedding-model
    volumes:
      - ollama_storage:/root/.ollama
    entrypoint: /bin/sh
    command:
      - "-c"
      - "sleep 3; OLLAMA_HOST=ollama:11434 ollama pull snowflake-arctic-embed:335m"
  
  qdrant:
    image: qdrant/qdrant
    container_name: qdrant
    networks: ['poc']
    restart: unless-stopped
    ports:
      - 6333:6333
      - 6334:6334
    volumes:
      - qdrant_storage:/qdrant/storage