Skip to content

Commit 6f7bd1b

Browse files
committed
updating dockerfiles for heimdall plugins
1 parent 4d86c0b commit 6f7bd1b

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

docker/Dockerfile.amd64-cuda-heimdall

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,12 @@ RUN echo "Building APOC plugin..." && \
6868
cd apoc/plugin-src/apoc && go build -buildmode=plugin -o ../../../apoc/built-plugins/apoc.so apoc_plugin.go && \
6969
echo "✓ Built plugin:" && ls -lh /build/apoc/built-plugins/*.so
7070

71+
# Build Heimdall plugin
72+
RUN echo "Building Heimdall plugin..." && \
73+
mkdir -p plugins/heimdall/built-plugins && \
74+
cd plugins/heimdall/plugin-src/watcher && go build -buildmode=plugin -o ../../built-plugins/watcher.so watcher_plugin.go && \
75+
echo "✓ Built Heimdall plugin:" && ls -lh /build/plugins/heimdall/built-plugins/*.so
76+
7177
# =============================================================================
7278
# Stage 4: Runtime
7379
# =============================================================================
@@ -81,6 +87,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
8187

8288
COPY --from=builder /build/nornicdb /app/
8389
COPY --from=builder /build/apoc/built-plugins /app/plugins/
90+
COPY --from=builder /build/plugins/heimdall/built-plugins /app/plugins/heimdall/
8491
COPY docker/entrypoint.sh /app/entrypoint-real.sh
8592
COPY docker/cuda-fallback-wrapper.sh /app/entrypoint.sh
8693
RUN chmod +x /app/entrypoint.sh /app/entrypoint-real.sh
@@ -121,6 +128,7 @@ ENV NORNICDB_DATA_DIR=/data \
121128
NORNICDB_PLUGINS_DIR=/app/plugins \
122129
NORNICDB_HEIMDALL_ENABLED=true \
123130
NORNICDB_HEIMDALL_MODEL=qwen2.5-0.5b-instruct \
131+
NORNICDB_HEIMDALL_PLUGINS_DIR=/app/plugins/heimdall \
124132
NVIDIA_VISIBLE_DEVICES=all \
125133
NVIDIA_DRIVER_CAPABILITIES=compute,utility \
126134
LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/nvidia/lib64:${LD_LIBRARY_PATH}

docker/Dockerfile.arm64-metal-heimdall

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,12 @@ RUN echo "Building APOC plugin..." && \
8787
cd apoc/plugin-src/apoc && go build -buildmode=plugin -o ../../../apoc/built-plugins/apoc.so apoc_plugin.go && \
8888
echo "✓ Built plugin:" && ls -lh /build/apoc/built-plugins/*.so
8989

90+
# Build Heimdall plugin
91+
RUN echo "Building Heimdall plugin..." && \
92+
mkdir -p plugins/heimdall/built-plugins && \
93+
cd plugins/heimdall/plugin-src/watcher && go build -buildmode=plugin -o ../../built-plugins/watcher.so watcher_plugin.go && \
94+
echo "✓ Built Heimdall plugin:" && ls -lh /build/plugins/heimdall/built-plugins/*.so
95+
9096
# =============================================================================
9197
# Stage 4: Runtime with embedded models
9298
# =============================================================================
@@ -99,6 +105,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
99105

100106
COPY --from=builder /build/nornicdb /app/
101107
COPY --from=builder /build/apoc/built-plugins /app/plugins/
108+
COPY --from=builder /build/plugins/heimdall/built-plugins /app/plugins/heimdall/
102109
COPY docker/entrypoint.sh /app/
103110
RUN chmod +x /app/entrypoint.sh
104111

@@ -147,6 +154,7 @@ ENV NORNICDB_DATA_DIR=/data \
147154
NORNICDB_HEIMDALL_ANOMALY_DETECTION=true \
148155
NORNICDB_HEIMDALL_RUNTIME_DIAGNOSIS=true \
149156
NORNICDB_HEIMDALL_MEMORY_CURATION=false \
157+
NORNICDB_HEIMDALL_PLUGINS_DIR=/app/plugins/heimdall \
150158
# General settings
151159
NORNICDB_NO_AUTH=true \
152160
NORNICDB_PLUGINS_DIR=/app/plugins

0 commit comments

Comments
 (0)