version: '3.8'

services:
  neo4j:
    image: neo4j:5.15.0
    container_name: graphrag_neo4j
    ports:
      - "7474:7474"
      - "7687:7687"
    environment:
      - NEO4J_AUTH=neo4j/password123

  python-service:
    build: ./PythonProject
    container_name: graphrag_python
    ports:
      - "8000:8000"

  ollama:
    image: ollama/ollama:latest
    container_name: graphrag_ollama
    ports:
      - "11434:11434"
    volumes:
      - ollama_data:/root/.ollama
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
  
  ollama-init:
    image: ollama/ollama:latest
    container_name: graphrag_ollama_init
    depends_on:
      - ollama # Ждем, пока запустится основной сервер
    environment:
      # Указываем этому контейнеру стучаться в основной сервер
      - OLLAMA_HOST=ollama:11434 
    # Команда на скачивание (замени qwen2.5-coder на нужную, если передумаешь)
    command: pull qwen2.5-coder:7b

volumes:
  ollama_data: