Nexarag/docker/prod/gpu/docker-compose.yml at main · KevinMoonLab/Nexarag · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
services:
  nexarag.neo4j:
    image: neo4j:latest
    container_name: nexarag.neo4j
    restart: unless-stopped
    environment:
      - NEO4J_AUTH=${NEO4J_USERNAME:-neo4j}/${NEO4J_PASSWORD:-password}
      - NEO4J_apoc_export_file_enabled=true
      - NEO4J_apoc_import_file_enabled=true
      - NEO4J_apoc_import_file_use__neo4j__config=true
      - NEO4J_dbms_security_procedures_unrestricted=apoc.meta.data
      - NEO4J_PLUGINS=["apoc"]
    volumes:
      - neo4j_data:/data
      - ./kg_dumps:/dumps
      - ./kg_dumps:/var/lib/neo4j/import
    healthcheck:
      test: ["CMD", "cypher-shell", "-u", "neo4j", "-p", "password", "MATCH (n) RETURN COUNT(n);"]
      interval: 10s
      timeout: 5s
      retries: 5

  nexarag.rabbitmq:
    image: rabbitmq:3.11-management
    container_name: nexarag.rabbitmq
    restart: unless-stopped
    environment:
      RABBITMQ_DEFAULT_USER: guest
      RABBITMQ_DEFAULT_PASS: guest
    volumes:
      - rabbit_data:/var/lib/rabbitmq
    healthcheck:
      test: ["CMD", "rabbitmqctl", "status"]
      interval: 10s
      timeout: 5s
      retries: 5

  nexarag.api:
    image: ghcr.io/kevinmoonlab/nexarag/nexarag.api:latest
    container_name: nexarag.api
    restart: unless-stopped
    ports:
      - "127.0.0.1:${API_PORT:-8000}:8000"
    environment:
      NEO4J_URI: bolt://nexarag.neo4j:7687
      NEO4J_USERNAME: ${NEO4J_USERNAME:-neo4j}
      NEO4J_PASSWORD: ${NEO4J_PASSWORD:-password}
      NEO4J_DATABASE: neo4j
      RABBITMQ_PASSWORD: ${RABBITMQ_PASSWORD:-guest}
      RABBITMQ_USERNAME: ${RABBITMQ_USERNAME:-guest}
      OLLAMA_BASE_URL: http://nexarag.ollama:11434
      EMBEDDING_CHUNK_SIZE: ${EMBEDDING_CHUNK_SIZE:-500}
      EMBEDDING_CHUNK_OVERLAP: ${EMBEDDING_CHUNK_OVERLAP:-100}
      EMBEDDING_MODEL: ${EMBEDDING_MODEL:-nomic-embed-text:v1.5}
      DEFAULT_MODEL: ${DEFAULT_MODEL:-gemma3:1b}
    depends_on:
      nexarag.neo4j:
        condition: service_healthy
      nexarag.rabbitmq:
        condition: service_healthy
    volumes:
      - docs_data:/docs
      - ./kg_dumps:/dumps

  nexarag.frontend:
    image: ghcr.io/kevinmoonlab/nexarag/nexarag.frontend:latest
    container_name: nexarag.frontend
    restart: unless-stopped
    ports:
      - "127.0.0.1:${FRONTEND_PORT:-5000}:80"
    depends_on:
      - nexarag.api

  nexarag.mcp:
    image: ghcr.io/kevinmoonlab/nexarag/nexarag.mcp:latest
    container_name: nexarag.mcp
    ports:
      - "127.0.0.1:${MCP_PORT:-9000}:9000"
    environment:
      NEO4J_URI: bolt://nexarag.neo4j:7687
      NEO4J_USERNAME: ${NEO4J_USERNAME:-neo4j}
      NEO4J_PASSWORD: ${NEO4J_PASSWORD:-password}
      NEO4J_DATABASE: neo4j
      RABBITMQ_PASSWORD: ${RABBITMQ_PASSWORD:-guest}
      RABBITMQ_USERNAME: ${RABBITMQ_USERNAME:-guest}
      OLLAMA_BASE_URL: http://nexarag.ollama:11434
      PYTHONUNBUFFERED: 1
      MCP_PORT: ${MCP_PORT:-9000}
      DEFAULT_MODEL: ${DEFAULT_MODEL:-qwen3:8b}
    restart: unless-stopped
    depends_on:
      nexarag.neo4j:
        condition: service_healthy

  nexarag.kg:
    image: ghcr.io/kevinmoonlab/nexarag/nexarag.kg:latest
    container_name: nexarag.kg
    restart: unless-stopped
    environment:
      NEO4J_URI: bolt://nexarag.neo4j:7687
      NEO4J_USERNAME: ${NEO4J_USERNAME:-neo4j}
      NEO4J_PASSWORD: ${NEO4J_PASSWORD:-password}
      NEO4J_DATABASE: neo4j
      RABBITMQ_PASSWORD: ${RABBITMQ_PASSWORD:-guest}
      RABBITMQ_USERNAME: ${RABBITMQ_USERNAME:-guest}
      OLLAMA_BASE_URL: http://nexarag.ollama:11434
      DEFAULT_MODEL: ${DEFAULT_MODEL:-gemma3:1b}
    volumes:
      - docs_data:/docs
      - conv_history_data:/data
      - ./kg_dumps:/dumps
    depends_on:
      nexarag.neo4j:
        condition: service_healthy
      nexarag.rabbitmq:
        condition: service_healthy

  nexarag.ollama:
    image: ollama/ollama
    container_name: nexarag.ollama
    ports:
      - "127.0.0.1:${OLLAMA_PORT:-11434}:11434"
    volumes:
      - ollama_data:/root/.ollama
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

volumes:
  neo4j_data:
  rabbit_data:
  ollama_data:
  docs_data:
  conv_history_data: