Files
esh-pfi-infrastructure/stacks/litellm/conf/config.yaml
T
vh 9a49963d07 feat(mistral-small-4): pin v0.22.0 for working VISION baseline + reasoning entry
Operator needs a verified-working vision tower as the abliteration/tuning
baseline. vLLM 0.23.0 crashes Mistral multimodal at startup (#44911
fetch_images regression, ~0.22.1+). Pinned the Mistral container to
v0.22.0 — the last pre-regression release — which loads the NVFP4
(compressed-tensors) AND serves vision: verified a half-blue/half-red
image read correctly ('left blue, right red'). Dropped --limit-mm
(vision re-enabled). qwen36 stays on 0.23.0 (separate container; needs it
for its ModelOpt NVFP4).

- gateway: add mistral-small-4-reasoning. Operator asked for effort=medium
  but Mistral's reasoning_effort is BINARY (none/high only — medium 400s);
  set to 'high' (sole reasoning-ON level). NOTE: reasoning fires but
  reasoning_content-splitting is unreliable on v0.22.0 (lands in content);
  clean split would need 0.23.0, which breaks vision — vision prioritized.
- mistral-small-4 (instant) + mistral-small-4-reasoning both gateway-live.
2026-06-15 17:56:37 -07:00

194 lines
8.4 KiB
YAML

# LiteLLM gateway config — fronts the vLLM services on ana-ml2
# (10.250.50.54) and logs every request + response so they're
# inspectable in the Logs UI at http://10.250.50.70:4000/ui.
#
# Deploys to /opt/docker/conf/litellm/config.yaml (mounted read-only
# into the container at /app/config.yaml).
#
# Model-name → upstream mapping:
# phi4-mini → vLLM :8004 (generative chat)
# qwen3-embedding → vLLM :8001 (/v1/embeddings)
# qwen3-reranker → vLLM :8002 (/rerank)
# * (wildcard) → llama-swap :9292 (the swappable generative zoo)
#
# The wildcard fronts llama-swap so its whole model zoo logs through the
# gateway without per-model registration. The vllm-reward classifier
# (:8003) is a pooling /classify endpoint with no first-class LiteLLM
# route — left direct; see README.
model_list:
# --- Granite 4.1 8B (generative chat) — production summarizer + dreaming
# agent. Replaced phi4-mini 2026-06-05 (beat it on precision in brokkr's
# R15 P03 eval). vLLM on ana-ml2 GPU 1, official FP8, 50K ctx. Explicit
# entry shadows the "*" wildcard's llama-swap route for this name. Full
# prompt + completion captured per call. ---
- model_name: granite-4.1-8b
litellm_params:
model: hosted_vllm/granite-4.1-8b
api_base: http://10.250.50.54:8004/v1
api_key: os.environ/VLLM_API_KEY
model_info:
mode: chat
# --- Qwen3.6-35B-A3B vision-language MoE (official FP8) — vision + chat. vLLM
# on ana-ml2 GPU 1, :8007. Explicit entry shadows the "*" wildcard llama-swap
# route. REPLACED qwen3.5-9b-fp8 2026-06-14 (the 9B is retired; this is a
# 35B-A3B MoE — served under its TRUE name, never aliased under the old one).
#
# THINKING SPLIT (2026-06-15, operator call — mirrors the glm-5.1 pattern
# above). One hybrid checkpoint; the per-request `enable_thinking` switch
# picks the mode. LiteLLM forwards extra_body verbatim to vLLM, where
# chat_template_kwargs lands in the chat template. vLLM runs
# --reasoning-parser qwen3 so reasoning surfaces as reasoning_content. ---
# qwen3.6-35b-a3b: thinking DISABLED by default. The checkpoint defaults
# thinking ON; enable_thinking=false forces the empty <think></think> block.
# Reasoning is opt-in via qwen3.6-35b-a3b-thinking below.
- model_name: qwen3.6-35b-a3b
litellm_params:
model: hosted_vllm/qwen3.6-35b-a3b
api_base: http://10.250.50.54:8007/v1
api_key: os.environ/VLLM_API_KEY
extra_body:
chat_template_kwargs:
enable_thinking: false
model_info:
mode: chat
# qwen3.6-35b-a3b-thinking: identical upstream checkpoint, thinking ENABLED
# (opt-in reasoning). The qwen3 reasoning-parser splits <think>…</think> into
# reasoning_content; content holds just the answer.
- model_name: qwen3.6-35b-a3b-thinking
litellm_params:
model: hosted_vllm/qwen3.6-35b-a3b
api_base: http://10.250.50.54:8007/v1
api_key: os.environ/VLLM_API_KEY
extra_body:
chat_template_kwargs:
enable_thinking: true
model_info:
mode: chat
# --- Mistral Small 4 (official NVFP4) — creative-writing / general text. 119B
# MoE (6.5B active), vLLM on ana-ml2 GPU 0 (dedicated 96 GB Blackwell), :8010.
# Explicit entry shadows the "*" wildcard. TEXT-ONLY for now — vLLM 0.23.0's
# Mistral multimodal processor crashes at startup (loaded with image/video
# limit 0); vision returns when vLLM patches it. Deployed 2026-06-15. ---
- model_name: mistral-small-4
litellm_params:
model: hosted_vllm/mistral-small-4
api_base: http://10.250.50.54:8010/v1
api_key: os.environ/VLLM_API_KEY
model_info:
mode: chat
# mistral-small-4-reasoning: same upstream checkpoint, reasoning ON (2026-06-15,
# operator wanted "medium" — but Mistral's reasoning_effort is BINARY, only
# 'none' or 'high' (a medium/low request 400s). 'high' is the sole reasoning-ON
# level, so it carries the -reasoning intent. LiteLLM forwards extra_body to
# vLLM; the mistral reasoning-parser splits [THINK]…[/THINK] into reasoning_content.
- model_name: mistral-small-4-reasoning
litellm_params:
model: hosted_vllm/mistral-small-4
api_base: http://10.250.50.54:8010/v1
api_key: os.environ/VLLM_API_KEY
extra_body:
reasoning_effort: high
model_info:
mode: chat
# --- Qwen3 embeddings ---
- model_name: qwen3-embedding
litellm_params:
model: hosted_vllm/Qwen/Qwen3-Embedding-0.6B
api_base: http://10.250.50.54:8001/v1
api_key: os.environ/VLLM_API_KEY
model_info:
mode: embedding
# --- Qwen3 reranker (proxy /rerank route) ---
- model_name: qwen3-reranker
litellm_params:
model: hosted_vllm/Qwen/Qwen3-Reranker-0.6B
api_base: http://10.250.50.54:8002/v1
api_key: os.environ/VLLM_API_KEY
model_info:
mode: rerank
# --- z.ai GLM (cloud API) — fronted for unified logging across local
# + cloud inference. Explicit entries, so they win over the "*"
# wildcard below (no collision with llama-swap's glm4.7-flash etc.
# — different model IDs). NOTE: paid API; only gateway-keyed callers
# can reach these, but they DO spend z.ai credits. Key in .env. ---
# glm-5.1: thinking DISABLED by default (2026-06-11, operator call). LiteLLM
# strips a top-level `thinking` param (drop_params), but forwards `extra_body`
# verbatim to z.ai, where the native thinking:{type:disabled} control lands —
# verified reasoning_tokens→0. Reasoning is opt-in via glm-5.1-reasoning below.
- model_name: glm-5.1
litellm_params:
model: openai/glm-5.1
api_base: https://api.z.ai/api/coding/paas/v4
api_key: os.environ/Z_AI_API_KEY
extra_body:
thinking:
type: disabled
# glm-5.1-reasoning: identical upstream, thinking ENABLED (opt-in reasoning).
- model_name: glm-5.1-reasoning
litellm_params:
model: openai/glm-5.1
api_base: https://api.z.ai/api/coding/paas/v4
api_key: os.environ/Z_AI_API_KEY
extra_body:
thinking:
type: enabled
- model_name: glm-5-turbo
litellm_params:
model: openai/glm-5-turbo
api_base: https://api.z.ai/api/coding/paas/v4
api_key: os.environ/Z_AI_API_KEY
- model_name: glm-4.7
litellm_params:
model: openai/glm-4.7
api_base: https://api.z.ai/api/coding/paas/v4
api_key: os.environ/Z_AI_API_KEY
- model_name: glm-4.5-air
litellm_params:
model: openai/glm-4.5-air
api_base: https://api.z.ai/api/coding/paas/v4
api_key: os.environ/Z_AI_API_KEY
# --- llama-swap passthrough (the swappable generative LLM zoo on
# ana-ml2:9292) ---
# Wildcard: any model name NOT matched by an exact entry above routes to
# llama-swap, which swaps the requested model into GPU on demand. This lets
# the gateway front the WHOLE swappable zoo (artemis / selene / qwen3.x / …)
# for logging + auth WITHOUT registering each model here — keep adding and
# swapping models in llama-swap freely; litellm logs them all. litellm does
# no inference; llama-swap still does all the model loading + serving.
# Exact matches above (phi4-mini / qwen3-embedding / qwen3-reranker) win;
# this only catches everything else. `openai/*` forwards the requested model
# name verbatim to llama-swap's OpenAI-compatible endpoint.
- model_name: "*"
litellm_params:
model: openai/*
api_base: http://10.250.50.54:9292/v1
api_key: "noauth" # llama-swap takes no auth; placeholder bearer
general_settings:
master_key: os.environ/LITELLM_MASTER_KEY
database_url: os.environ/DATABASE_URL
store_model_in_db: true
# THE log switch: persists full request messages + response bodies into
# SpendLogs so they render in the Logs UI. Without this you get metadata
# (tokens, latency, model) but not the prompt/completion text.
store_prompts_in_spend_logs: true
litellm_settings:
# vLLM rejects some OpenAI params other backends accept; drop silently
# rather than 400 the caller.
drop_params: true
# --- Langfuse trace export (live 2026-06-05). Full prompt/completion +
# reasoning + tok-derivable latency traces ship to the Langfuse stack on
# ana-docker (project "gateway"). Keys + host in .env. The gateway and
# every consumer stay pointed here — this callback is the whole upgrade. ---
success_callback: ["langfuse"]
failure_callback: ["langfuse"]