#!/usr/bin/env bash
set -euo pipefail

# 1) Embedding / Rerank 还在不在
curl -s -m 15 -H 'Authorization: Bearer xf-...' http://127.0.0.1:9997/v1/models

# 2) Rerank 能不能打分（应在 1 秒内）
curl -s -m 30 -H 'Authorization: Bearer xf-...' \
  -H 'Content-Type: application/json' \
  -d '{"model":"bge-reranker-v2-m3","query":"管理人员需要钉钉打卡吗","documents":["管理人员应通过钉钉打卡。","班车时间以通知为准。"]}' \
  http://127.0.0.1:9997/v1/rerank

# 3) 知识库命中测试（含 Rerank 的完整检索）
# 用 Dify 控制台或 dify_api.py POST /datasets/<id>/hit-testing
# 本环境该问句约 2.4s HTTP 200

# 4) vLLM 还在不在（在 207 上测跨机）
curl -s -m 10 -H 'Authorization: Bearer 123456' \
  http://116.62.10.183:8000/v1/models

# 5) 最短对话（不走知识库）
curl -s -m 60 -X POST http://116.62.10.183:8000/v1/chat/completions \
  -H 'Authorization: Bearer 123456' -H 'Content-Type: application/json' \
  -d '{"model":"qwen3.6-27b","messages":[{"role":"user","content":"只回：pong"}],"max_tokens":16}'