You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.3 KiB
50 lines
1.3 KiB
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
BASE_URL="${BASE_URL:-http://127.0.0.1:8000}"
|
|
USERNAME="${USERNAME:-admin}"
|
|
PASSWORD="${PASSWORD:-admin123}"
|
|
SESSION_ID="${SESSION_ID:-curl-sse-$(date +%s)}"
|
|
MESSAGE="${MESSAGE:-杀戮尖塔的观者怎么玩}"
|
|
|
|
TOKEN="$(
|
|
curl -sS "${BASE_URL}/api/v1/user/login" \
|
|
-H 'Content-Type: application/json' \
|
|
-d "{\"username\":\"${USERNAME}\",\"password\":\"${PASSWORD}\"}" \
|
|
| python3 -c 'import sys, json; print(json.load(sys.stdin)["token"])'
|
|
)"
|
|
|
|
TMP_OUT="$(mktemp)"
|
|
cleanup() { rm -f "${TMP_OUT}"; }
|
|
trap cleanup EXIT
|
|
|
|
curl -N -sS "${BASE_URL}/api/v1/ai/chat" \
|
|
-H "Authorization: Bearer ${TOKEN}" \
|
|
-H "Content-Type: application/json" \
|
|
-H "Accept: text/event-stream" \
|
|
-d "$(python3 - <<'PY'
|
|
import json, os
|
|
payload = {
|
|
"sessionId": os.environ["SESSION_ID"],
|
|
"message": os.environ["MESSAGE"],
|
|
"stream": True,
|
|
"context": {
|
|
"webSearch": True,
|
|
"agent": "Hadoop助手. You MUST use the web_search tool before answering."
|
|
}
|
|
}
|
|
print(json.dumps(payload, ensure_ascii=False))
|
|
PY
|
|
)" | tee "${TMP_OUT}"
|
|
|
|
python3 - <<'PY'
|
|
import sys, re, pathlib
|
|
p = pathlib.Path(sys.argv[1])
|
|
s = p.read_text(encoding="utf-8", errors="ignore")
|
|
n = len(re.findall(r"^data: ", s, flags=re.M))
|
|
if n <= 0:
|
|
raise SystemExit("未收到任何 SSE data 行，测试失败")
|
|
print(f"OK: 收到 {n} 条 SSE data 行")
|
|
PY "${TMP_OUT}"
|
|
|