Plasmod/docker-compose.test.yml at main · CodeSoul-co/Plasmod · 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
# Test-mode override for docker-compose.yml.
#
# CPU-only (default):
#   APP_MODE=test docker compose -f docker-compose.yml -f docker-compose.test.yml \
#     up -d --build --force-recreate
#
# GPU / TensorRT (stack all three overrides):
#   APP_MODE=test docker compose \
#     -f docker-compose.yml \
#     -f docker-compose.test.yml \
#     -f docker-compose.gpu.yml \
#     -f docker-compose.test-gpu.yml \
#     up -d --build --force-recreate
#
# Enables:
#   - APP_MODE=test  (debug endpoints, relaxed visibility, /v1/debug/echo)
#   - ONNX CPU embedder by default (override via PLASMOD_EMBEDDER env var)
#   - Read-only bind-mounts for models/ and database/ (testQuery10K.fbin)

services:
  andb:
    environment:
      APP_MODE: test
      # Hardcoded ONNX CPU embedder — overrides .env tfidf default
      PLASMOD_EMBEDDER: onnx
      PLASMOD_EMBEDDER_MODEL_PATH: /models/minilm-l6-v2.onnx
      PLASMOD_ONNX_VOCAB_PATH: /models/minilm-l6-v2-vocab.txt
      PLASMOD_EMBEDDER_DIM: "384"
      PLASMOD_EMBEDDER_BATCH_SIZE: "32"
      PLASMOD_EMBEDDER_DEVICE: cpu
      PLASMOD_HTTP_ADDR: "0.0.0.0:8080"
    volumes:
      - ./models:/models:ro
      - /home/duanzhenke/database:/database:ro