aleksey
83d516afa1
FastAPI gateway with A/B/C lane orchestration, LiteLLM proxy config, Swarm stack (postgres, redis, VPN off-by-default), deploy/smoke/audit scripts.
54 lines
1.7 KiB
Bash
54 lines
1.7 KiB
Bash
#!/usr/bin/env bash
|
|
# Smoke tests — classify (no LLM) + minimal chat
|
|
set -euo pipefail
|
|
|
|
STACK_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
cd "$STACK_DIR"
|
|
|
|
set -a
|
|
# shellcheck disable=SC1091
|
|
source .env 2>/dev/null || true
|
|
set +a
|
|
|
|
BASE="${TEST_BASE_URL:-http://127.0.0.1:8000}"
|
|
LITELLM_BASE="${TEST_LITELLM_URL:-http://127.0.0.1:4000}"
|
|
ROUTER_KEY="${ROUTER_API_KEY:?ROUTER_API_KEY required}"
|
|
|
|
CURL=(curl -sf)
|
|
if [[ "$BASE" == https:* ]]; then
|
|
CURL+=( -k )
|
|
fi
|
|
|
|
echo "== Router health =="
|
|
"${CURL[@]}" "${BASE}/health" | jq .
|
|
|
|
echo "== Classify SIMPLE =="
|
|
"${CURL[@]}" "${BASE}/classify" \
|
|
-H "Authorization: Bearer ${ROUTER_KEY}" \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"messages":[{"role":"user","content":"Привет, что такое Docker?"}]}' | jq .
|
|
|
|
echo "== Classify COMPLEX =="
|
|
"${CURL[@]}" "${BASE}/classify" \
|
|
-H "Authorization: Bearer ${ROUTER_KEY}" \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"messages":[{"role":"user","content":"Спроектируй архитектуру microservices"}]}' | jq .
|
|
|
|
echo "== Chat smart-router max_tokens=16 =="
|
|
"${CURL[@]}" "${BASE}/v1/chat/completions" \
|
|
-H "Authorization: Bearer ${ROUTER_KEY}" \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"model":"smart-router","max_tokens":16,"messages":[{"role":"user","content":"bash docker service ls"}]}' \
|
|
| jq '.choices[0].message.content, .x_router_meta // empty'
|
|
|
|
if [[ -n "${SKIP_LITELLM_SMOKE:-}" ]]; then
|
|
echo "SKIP_LITELLM_SMOKE set — skipping LiteLLM checks"
|
|
else
|
|
echo "== LiteLLM liveliness =="
|
|
"${CURL[@]}" "${LITELLM_BASE}/health/liveliness" && echo
|
|
echo "== LiteLLM UI =="
|
|
curl -sfI -k "${LITELLM_BASE}/ui" | head -3 || true
|
|
fi
|
|
|
|
echo "All smoke checks passed."
|