init

2026-03-27 10:49:34 +08:00
commit 22a5f31847
56 changed files with 5145 additions and 0 deletions
@@ -0,0 +1,64 @@
+from __future__ import annotations
+
+import argparse
+import json
+import ssl
+import time
+import urllib.error
+import urllib.request
+
+
+def _json_request(url: str, method: str = "GET", payload: dict | None = None, timeout: int = 20) -> dict:
+    body = None if payload is None else json.dumps(payload, ensure_ascii=False).encode("utf-8")
+    req = urllib.request.Request(url=url, method=method, data=body)
+    if payload is not None:
+        req.add_header("Content-Type", "application/json")
+    ctx = ssl._create_unverified_context() if url.startswith("https://") else None
+    with urllib.request.urlopen(req, timeout=timeout, context=ctx) as resp:
+        return json.loads(resp.read().decode("utf-8"))
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="Quick QA for text pipeline and metrics")
+    parser.add_argument("--base-url", default="https://127.0.0.1:8080")
+    parser.add_argument("--rounds", type=int, default=5)
+    args = parser.parse_args()
+
+    health = _json_request(f"{args.base_url}/health")
+    print("health.ok:", bool(health.get("llm")))
+    latencies = []
+    for i in range(args.rounds):
+        t0 = time.perf_counter()
+        data = _json_request(
+            f"{args.base_url}/chat/text",
+            method="POST",
+            payload={"text": f"第{i+1}轮压测，请简短回复。"},
+            timeout=60,
+        )
+        dt = int((time.perf_counter() - t0) * 1000)
+        latencies.append(dt)
+        print(
+            f"round={i+1} ok={data.get('ok')} llm={data.get('llm_latency_ms')} "
+            f"tts={data.get('tts_latency_ms')} total={dt}"
+        )
+
+    p50 = sorted(latencies)[len(latencies) // 2] if latencies else 0
+    print("rounds:", args.rounds, "p50_ms:", p50, "max_ms:", max(latencies) if latencies else 0)
+
+    health2 = _json_request(f"{args.base_url}/health")
+    print(
+        "final.health:",
+        {
+            "pipeline_runs": health2.get("pipeline_runs"),
+            "last_latency_ms": health2.get("last_latency_ms"),
+            "last_llm_latency_ms": health2.get("last_llm_latency_ms"),
+            "last_tts_latency_ms": health2.get("last_tts_latency_ms"),
+        },
+    )
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except urllib.error.URLError as exc:
+        raise SystemExit(f"request failed: {exc}")