feat: include model

2026-03-17 17:18:40 -04:00
parent eeae628a1a
commit d5415da5d9
1 changed files with 3 additions and 0 deletions
--- a/pkg/Dockerfile
+++ b/pkg/Dockerfile
@@ -14,6 +14,9 @@ COPY requirements.txt .
 # Install dependencies using buildkit cache
 RUN --mount=type=cache,target=/root/.cache/pip pip install -r requirements.txt

+# Pre-download the embedding model into the image to avoid runtime downloads
+RUN python -c "from fastembed import TextEmbedding; TextEmbedding(model_name='BAAI/bge-small-en-v1.5')"
+
 # Copy the source code
 COPY src/ ./src/