From 9dc734eee56366b6542ddfb6caade2676e6c7420 Mon Sep 17 00:00:00 2001 From: Santosh Bhavani Date: Sun, 19 Oct 2025 14:52:24 -0500 Subject: [PATCH] Add NVIDIA_API_KEY support and update ollama to v0.12.6 --- nvidia/txt2kg/assets/deploy/compose/docker-compose.yml | 1 + nvidia/txt2kg/assets/deploy/services/ollama/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/nvidia/txt2kg/assets/deploy/compose/docker-compose.yml b/nvidia/txt2kg/assets/deploy/compose/docker-compose.yml index 5916938..ea8a9ec 100644 --- a/nvidia/txt2kg/assets/deploy/compose/docker-compose.yml +++ b/nvidia/txt2kg/assets/deploy/compose/docker-compose.yml @@ -14,6 +14,7 @@ services: - OLLAMA_BASE_URL=http://ollama:11434/v1 - OLLAMA_MODEL=llama3.1:8b - REMOTE_WEBGPU_SERVICE_URL=http://txt2kg-remote-webgpu:8083 + - NVIDIA_API_KEY=${NVIDIA_API_KEY:-} # Node.js timeout configurations for large model processing - NODE_OPTIONS=--max-http-header-size=80000 - UV_THREADPOOL_SIZE=128 diff --git a/nvidia/txt2kg/assets/deploy/services/ollama/Dockerfile b/nvidia/txt2kg/assets/deploy/services/ollama/Dockerfile index e3e76d8..efdad5f 100644 --- a/nvidia/txt2kg/assets/deploy/services/ollama/Dockerfile +++ b/nvidia/txt2kg/assets/deploy/services/ollama/Dockerfile @@ -1,4 +1,4 @@ -FROM ollama/ollama:latest +FROM ollama/ollama:0.12.6 # Copy the entrypoint script COPY entrypoint.sh /entrypoint.sh