SIGMUS/compose.yaml at main · nesl/SIGMUS · 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
x-app: &app
  build:
    context: .
    dockerfile: Dockerfile
    additional_contexts:
      observation_processing: ../urban-observation-processing
  image: sigmus:local
  restart: unless-stopped
  volumes:
    - ./.runtime/config.docker.json:/app/config.json:ro

services:
  timescaledb:
    image: timescale/timescaledb:latest-pg16
    restart: unless-stopped
    environment:
      POSTGRES_DB: ${SIGMUS_POSTGRES_DB:?Run ./docker-start instead of docker compose directly}
      POSTGRES_USER: ${SIGMUS_POSTGRES_USER:?Run ./docker-start instead of docker compose directly}
      POSTGRES_PASSWORD: ${SIGMUS_POSTGRES_PASSWORD:?Run ./docker-start instead of docker compose directly}
    volumes:
      - timescale-data:/var/lib/postgresql/data
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U $$POSTGRES_USER -d $$POSTGRES_DB"]
      interval: 5s
      timeout: 5s
      retries: 20

  neo4j:
    image: neo4j:5
    restart: unless-stopped
    environment:
      NEO4J_AUTH: neo4j/${SIGMUS_NEO4J_PASSWORD:?Run ./docker-start instead of docker compose directly}
      # Local demo instance only; Neo4j defaults to a minimum of 8 characters.
      NEO4J_dbms_security_auth__minimum__password__length: 6
    ports:
      - "127.0.0.1:${SIGMUS_NEO4J_BROWSER_PORT:-7474}:7474"
      - "127.0.0.1:${SIGMUS_NEO4J_BOLT_PORT:-7687}:7687"
    volumes:
      - neo4j-data:/data
    healthcheck:
      test: ["CMD-SHELL", "cypher-shell -u neo4j -p \"$${NEO4J_AUTH#*/}\" 'RETURN 1' || exit 1"]
      interval: 5s
      timeout: 5s
      retries: 30
      start_period: 20s

  marqo:
    image: marqoai/marqo:latest
    restart: unless-stopped
    volumes:
      - marqo-data:/opt/vespa/var
    healthcheck:
      test: ["CMD-SHELL", "curl -fsS http://localhost:8882/health >/dev/null || exit 1"]
      interval: 10s
      timeout: 5s
      retries: 30
      start_period: 30s

  ingestion:
    <<: *app
    depends_on:
      timescaledb: {condition: service_healthy}
      neo4j: {condition: service_healthy}
    volumes:
      - ./.runtime/config.docker.json:/app/config.json:ro
      - type: bind
        source: ${SIGMUS_STREAM_FOLDER:?Run ./docker-start instead of docker compose directly}
        target: /streams
        read_only: true
      - ingestion-state:/state
    command: ["python", "-m", "database_storage.worker", "--input", "/streams/${SIGMUS_STREAM_FILE:-observations.jsonl}"]
    healthcheck:
      test: ["CMD-SHELL", "test -f /state/healthy"]
      interval: 10s
      timeout: 3s
      retries: 6

  mcp:
    <<: *app
    depends_on:
      timescaledb: {condition: service_healthy}
      neo4j: {condition: service_healthy}
    ports:
      - "127.0.0.1:${SIGMUS_MCP_PORT:-8006}:8006"
    environment:
      SIGMUS_QA_HOST: 0.0.0.0
      SIGMUS_QA_PORT: 8006
    command: ["python", "-m", "qa.mcp_server"]
    healthcheck:
      test: ["CMD", "python", "-c", "import socket; socket.create_connection(('127.0.0.1',8006),2).close()"]
      interval: 10s
      timeout: 3s
      retries: 6

volumes:
  timescale-data:
  neo4j-data:
  marqo-data:
  ingestion-state: