feat(embed-rerank): TEI is the fleet embed/rerank engine; esh-ml1 sole backend; retire fv-ml1 seats
Prime, 2026-09-25: TEI serves embedding + reranking for esh-ml1 and the fleet from now on; fv-ml1 retires both once esh-ml1 is up. - stacks/embed-rerank: vLLM -> TEI 1.9.4 (89- Ada build), same ports 8001/8013, fail-closed truncation (--auto-truncate false; embed --max-batch-tokens 32768). - litellm: qwen3-embedding -> esh-ml1 only (hosted_vllm/, unchanged address); reranker -> huggingface/ provider at :8013 (hosted_vllm/ 422s on TEI's `texts` body). DB alias reranker-a3-bge-v2-m3 patched to the same target. - Verified via the gateway against the retiring fv-ml1 seats: embed cosine median 0.999927 (n=203); rerank top-1/top-3 29/30. - stacks/vllm: vllm-embed and vllm-rerank-a3 removed (containers retired on fv-ml1, GPU 1 freed ~6.1 GB); reward + coder unchanged. - Bake-off record moved to docs/pfi/embed-rerank-tei-vs-vllm-bakeoff.md; CLAUDE.md gains the TEI convention.
This commit is contained in:
@@ -1,63 +1,66 @@
|
||||
# embed-rerank — the fleet's embedding + reranking models, served locally at ESH
|
||||
# on esh-ml1 (CT 110 on esh-pve, RTX 2000E Ada, 16 GB).
|
||||
# embed-rerank — THE fleet's embedding + reranking service, on esh-ml1 (CT 110 on
|
||||
# esh-pve, RTX 2000E Ada, 16 GB). Served by Hugging Face Text Embeddings
|
||||
# Inference (TEI).
|
||||
#
|
||||
# WHY: Prime's decision 2026-09-24 — the Ada card offloads embedding and
|
||||
# reranking so ESH consumers (Open WebUI RAG, Paperless) are not tied to one
|
||||
# seat on fv-ml1. It serves the SAME models as fv-ml1's `vllm` stack, with the
|
||||
# SAME vLLM version and flags, because embedding vectors are model-specific:
|
||||
# a different embedding model here would silently poison every index built
|
||||
# against fv-ml1's. Mirror stacks/vllm/compose.yaml when that changes.
|
||||
# Prime, 2026-09-25: "TEI is embed/reranker server for esh-ml1 and the FLEET in
|
||||
# general, in future." It replaced vLLM here the same day, after a side-by-side
|
||||
# bake-off on this card (docs/pfi/embed-rerank-tei-vs-vllm-bakeoff.md). TEI
|
||||
# gives the same vectors as the old vLLM seats (no re-embedding), but it is
|
||||
# ~1.3x slower on bulk work on this card. It is much lighter (2.6 GB VRAM for
|
||||
# both, 8 GB image, ~4 s restart). fv-ml1's vLLM embed/rerank seats were
|
||||
# retired after this went live.
|
||||
#
|
||||
# vllm-embed Qwen/Qwen3-Embedding-0.6B → /v1/embeddings :8001
|
||||
# vllm-rerank-bge BAAI/bge-reranker-v2-m3 → /rerank, /score :8013
|
||||
# tei-embed Qwen/Qwen3-Embedding-0.6B → /v1/embeddings (OpenAI), /embed :8001
|
||||
# tei-rerank BAAI/bge-reranker-v2-m3 → /rerank (body: query + texts) :8013
|
||||
#
|
||||
# Ports match fv-ml1 on purpose, so a LiteLLM deployment for either site
|
||||
# differs only in the host part of api_base.
|
||||
# Ports kept from the vLLM era (and fv-ml1), so the embedding gateway entry did
|
||||
# not change address. ⚠ The RERANK gateway entry must use LiteLLM's
|
||||
# `huggingface/` provider: `hosted_vllm/` sends `documents` and TEI answers 422
|
||||
# "missing field texts".
|
||||
#
|
||||
# `vllm-rerank-bge`, not fv-ml1's `vllm-rerank-a3`: that name carries R43
|
||||
# bake-off provenance for THAT container; and plain `vllm-rerank` was the
|
||||
# retired Qwen3-Reranker that measured harmful. Name the model instead.
|
||||
# ⚠ Embedding vectors are model-specific. Never change EMBED_MODEL without a
|
||||
# re-embedding plan for every index built on it (Worldtree, nevermore, Open WebUI).
|
||||
#
|
||||
# NO `tnet`/traefik-net: esh-ml1 runs no traefik, and an external network
|
||||
# that does not exist would stop the stack from starting. Consumers reach the
|
||||
# published ports directly.
|
||||
# FAIL-CLOSED truncation (--auto-truncate false). TEI's default silently
|
||||
# embedded the first 16,384 tokens of a ~40k-token input and returned 200.
|
||||
# Turning it off requires --max-batch-tokens >= the model's max input (32,768
|
||||
# for Qwen3-Embedding), or TEI refuses to start.
|
||||
#
|
||||
# Host setup (driver, LXC, docker, toolkit): playbooks/esh-pve-nvidia-host.yaml
|
||||
# then playbooks/esh-ml1-lxc.yaml. Tunables live in .env — edit that, not this.
|
||||
# NO `tnet`/traefik-net: esh-ml1 runs no traefik; consumers reach the published
|
||||
# ports, and in practice only the LiteLLM gateway does (verified from the seats'
|
||||
# logs 2026-09-25: every request matched a gateway spend-log row).
|
||||
#
|
||||
# Host setup: playbooks/esh-pve-nvidia-host.yaml, then playbooks/esh-ml1-lxc.yaml.
|
||||
# Tunables live in .env.
|
||||
|
||||
name: embed-rerank
|
||||
|
||||
services:
|
||||
vllm-embed:
|
||||
image: vllm/vllm-openai:${VLLM_VERSION}
|
||||
container_name: vllm-embed
|
||||
tei-embed:
|
||||
image: ghcr.io/huggingface/text-embeddings-inference:${TEI_TAG}
|
||||
container_name: tei-embed
|
||||
restart: unless-stopped
|
||||
ipc: host
|
||||
ports:
|
||||
- "${EMBED_PORT}:8000"
|
||||
- "${EMBED_PORT}:80"
|
||||
volumes:
|
||||
- /opt/aimodels/huggingface:/hfcache
|
||||
- /opt/aimodels/tei-cache:/data
|
||||
environment:
|
||||
- HF_HOME=/hfcache
|
||||
- HF_HUB_CACHE=/hfcache/hub
|
||||
- HUGGING_FACE_HUB_TOKEN=${HF_TOKEN:-}
|
||||
- VLLM_API_KEY=${API_KEY:-}
|
||||
- HF_TOKEN=${HF_TOKEN:-}
|
||||
command:
|
||||
- --model-id
|
||||
- ${EMBED_MODEL}
|
||||
- --served-model-name
|
||||
- ${EMBED_MODEL}
|
||||
- --runner
|
||||
- pooling
|
||||
- --host
|
||||
- 0.0.0.0
|
||||
- --port
|
||||
- "8000"
|
||||
- --gpu-memory-utilization
|
||||
- ${EMBED_GPU_MEM_UTIL}
|
||||
- --max-model-len
|
||||
- ${EMBED_MAX_MODEL_LEN}
|
||||
# TEI on CUDA is float16-only; parity vs the bf16 vLLM vectors was measured.
|
||||
- --dtype
|
||||
- auto
|
||||
- float16
|
||||
# Default 32; vLLM had no cap and callers batch 64.
|
||||
- --max-client-batch-size
|
||||
- "${MAX_CLIENT_BATCH_SIZE}"
|
||||
- --auto-truncate
|
||||
- "false"
|
||||
- --max-batch-tokens
|
||||
- "32768"
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
@@ -66,49 +69,39 @@ services:
|
||||
device_ids: ["0"]
|
||||
capabilities: [gpu]
|
||||
healthcheck:
|
||||
test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
|
||||
test: ["CMD", "curl", "-fsS", "http://localhost:80/health"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 180s
|
||||
start_period: 60s
|
||||
labels:
|
||||
- homepage.group=AI - Eval & Retrieval
|
||||
- homepage.name=vLLM Embed (esh-ml1)
|
||||
- homepage.name=Embed — Qwen3 0.6B (TEI, esh-ml1)
|
||||
- homepage.icon=mdi-vector-arrange-below
|
||||
- homepage.description=Qwen3 Embedding 0.6B via vLLM (esh-ml1, RTX 2000E Ada)
|
||||
- homepage.description=Fleet embeddings (qwen3-embedding) via TEI on esh-ml1
|
||||
- homepage.href=http://10.0.50.80:${EMBED_PORT}/docs
|
||||
|
||||
vllm-rerank-bge:
|
||||
image: vllm/vllm-openai:${VLLM_VERSION}
|
||||
container_name: vllm-rerank-bge
|
||||
tei-rerank:
|
||||
image: ghcr.io/huggingface/text-embeddings-inference:${TEI_TAG}
|
||||
container_name: tei-rerank
|
||||
restart: unless-stopped
|
||||
ipc: host
|
||||
ports:
|
||||
- "${RERANK_PORT}:8000"
|
||||
- "${RERANK_PORT}:80"
|
||||
volumes:
|
||||
- /opt/aimodels/huggingface:/hfcache
|
||||
- /opt/aimodels/tei-cache:/data
|
||||
environment:
|
||||
- HF_HOME=/hfcache
|
||||
- HF_HUB_CACHE=/hfcache/hub
|
||||
- HUGGING_FACE_HUB_TOKEN=${HF_TOKEN:-}
|
||||
- VLLM_API_KEY=${API_KEY:-}
|
||||
- HF_TOKEN=${HF_TOKEN:-}
|
||||
command:
|
||||
- --model-id
|
||||
- ${RERANK_MODEL}
|
||||
- --served-model-name
|
||||
- ${RERANK_MODEL}
|
||||
- --runner
|
||||
- pooling
|
||||
# bge-reranker-v2-m3 is natively a cross-encoder; no --hf-overrides.
|
||||
- --host
|
||||
- 0.0.0.0
|
||||
- --port
|
||||
- "8000"
|
||||
- --gpu-memory-utilization
|
||||
- ${RERANK_GPU_MEM_UTIL}
|
||||
- --max-model-len
|
||||
- ${RERANK_MAX_MODEL_LEN}
|
||||
- --dtype
|
||||
- auto
|
||||
- float16
|
||||
- --max-client-batch-size
|
||||
- "${MAX_CLIENT_BATCH_SIZE}"
|
||||
# Fail-closed; bge-reranker-v2-m3's max input (8,192) fits the default
|
||||
# max-batch-tokens (16,384).
|
||||
- --auto-truncate
|
||||
- "false"
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
@@ -117,14 +110,14 @@ services:
|
||||
device_ids: ["0"]
|
||||
capabilities: [gpu]
|
||||
healthcheck:
|
||||
test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
|
||||
test: ["CMD", "curl", "-fsS", "http://localhost:80/health"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 180s
|
||||
start_period: 60s
|
||||
labels:
|
||||
- homepage.group=AI - Eval & Retrieval
|
||||
- homepage.name=vLLM Rerank bge-v2-m3 (esh-ml1)
|
||||
- homepage.name=Rerank — bge-v2-m3 (TEI, esh-ml1)
|
||||
- homepage.icon=mdi-sort-variant
|
||||
- homepage.description=bge-reranker-v2-m3 via vLLM (esh-ml1, RTX 2000E Ada)
|
||||
- homepage.description=Fleet reranker (reranker) via TEI on esh-ml1
|
||||
- homepage.href=http://10.0.50.80:${RERANK_PORT}/docs
|
||||
|
||||
Reference in New Issue
Block a user