NodeMemoryGraph/Dockerfile at main · wefio/NodeMemoryGraph · 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
# syntax=docker/dockerfile:1.7

ARG NODE_VERSION=22.19.0

FROM node:${NODE_VERSION}-bookworm-slim AS build
WORKDIR /app

COPY package.json package-lock.json ./
RUN npm ci

COPY tsconfig.json tsconfig.build.json ./
COPY src ./src
RUN npm run build


FROM node:${NODE_VERSION}-bookworm-slim AS nmg-runtime
ENV DEBIAN_FRONTEND=noninteractive \
    NMG_DATA_DIR=/data \
    NMG_DB_PATH=/data/nmg.sqlite \
    NMG_DAEMON_IDLE_TIMEOUT_MS=0 \
    NMG_EMBED_AUTO_SYNC=1 \
    NMG_EMBED_LOCAL_SERVER=0

RUN apt-get -o Acquire::Retries=5 update \
    && apt-get -o Acquire::Retries=5 install -y --no-install-recommends \
       ca-certificates \
       tini \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /app
COPY package.json package-lock.json ./
RUN npm ci --omit=dev --omit=optional \
    && npm cache clean --force

COPY --from=build /app/dist ./dist
COPY bin ./bin
COPY docker/entrypoint.sh ./docker/entrypoint.sh

RUN chmod +x ./docker/entrypoint.sh \
    && mkdir -p /data

VOLUME ["/data"]

HEALTHCHECK --interval=30s --timeout=5s --start-period=90s --retries=3 \
  CMD node bin/nmg.mjs daemon status --db "$NMG_DB_PATH" --json \
      | grep -q '"running": true'

ENTRYPOINT ["/usr/bin/tini", "--", "/app/docker/entrypoint.sh"]


# Lightweight runtime for FTS-only use or an external OpenAI-compatible
# embedding provider. It intentionally contains no Python, PyTorch, CUDA, or
# model weights.
FROM nmg-runtime AS external


# Self-contained convenience image. This remains the final/default target so
# existing `docker build .` commands keep producing the local-BGE image.
FROM nmg-runtime AS bge

ARG TORCH_INDEX_URL=https://download.pytorch.org/whl/cu126
ARG BGE_SOURCE_MODEL=BAAI/bge-small-en-v1.5
ARG BGE_REVISION=5c38ec7c405ec4b44b94cc5a9bb96e735b38267a

ENV NMG_EMBED_BASE_URL=http://127.0.0.1:8000/v1 \
    NMG_EMBED_MODEL=BAAI/bge-small-en-v1.5 \
    NMG_EMBED_PROFILE=bge-en \
    NMG_EMBED_LOCAL_SERVER=1 \
    BGE_MODEL=/opt/models/bge-small-en-v1.5 \
    BGE_PORT=8000 \
    VIRTUAL_ENV=/opt/nmg-embed \
    PATH=/opt/nmg-embed/bin:$PATH

RUN apt-get -o Acquire::Retries=5 update \
    && apt-get -o Acquire::Retries=5 install -y --no-install-recommends \
       python3 \
       python3-pip \
       python3-venv \
    && rm -rf /var/lib/apt/lists/*

RUN python3 -m venv "$VIRTUAL_ENV" \
    && pip install --no-cache-dir --upgrade pip setuptools wheel \
    && pip install --no-cache-dir --index-url "$TORCH_INDEX_URL" torch

RUN pip install --no-cache-dir sentence-transformers fastapi "uvicorn[standard]"

RUN BGE_SOURCE_MODEL="$BGE_SOURCE_MODEL" BGE_SOURCE_REVISION="$BGE_REVISION" \
    python -c 'import os; from sentence_transformers import SentenceTransformer; model = SentenceTransformer(os.environ["BGE_SOURCE_MODEL"], revision=os.environ["BGE_SOURCE_REVISION"]); model.save_pretrained("/opt/models/bge-small-en-v1.5")'

COPY evals/omnimemeval/bge_server.py ./evals/omnimemeval/bge_server.py
COPY evals/omnimemeval/embedding_batcher.py ./evals/omnimemeval/embedding_batcher.py

ENV HF_HUB_OFFLINE=1 \
    TRANSFORMERS_OFFLINE=1

HEALTHCHECK --interval=30s --timeout=5s --start-period=90s --retries=3 \
  CMD python -c "import urllib.request; urllib.request.urlopen('http://127.0.0.1:8000/health', timeout=3)" \
      && node bin/nmg.mjs daemon status --db "$NMG_DB_PATH" --json \
      | grep -q '"running": true'