89 lines
3.2 KiB
Python
89 lines
3.2 KiB
Python
|
"""临时 E2E：对比 → 采用 → 存档 → 回滚。跑完可删。"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import sys
|
|||
|
|
import urllib.request
|
|||
|
|
|
|||
|
|
# Windows 控制台避免 ✅ 等字符触发 GBK 编码错误
|
|||
|
|
sys.stdout.reconfigure(encoding="utf-8", errors="replace")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def chat(text: str, sid: str | None = None) -> list[dict]:
|
|||
|
|
body = json.dumps({"text": text, "sessionId": sid}).encode()
|
|||
|
|
req = urllib.request.Request(
|
|||
|
|
"http://localhost:8000/api/chat",
|
|||
|
|
data=body,
|
|||
|
|
headers={"Content-Type": "application/json"},
|
|||
|
|
)
|
|||
|
|
events: list[dict] = []
|
|||
|
|
with urllib.request.urlopen(req, timeout=60) as resp:
|
|||
|
|
buf = ""
|
|||
|
|
for chunk in resp:
|
|||
|
|
buf += chunk.decode()
|
|||
|
|
while "\n\n" in buf:
|
|||
|
|
frame, buf = buf.split("\n\n", 1)
|
|||
|
|
if frame.startswith("data: "):
|
|||
|
|
events.append(json.loads(frame[6:]))
|
|||
|
|
return events
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get(url: str) -> dict:
|
|||
|
|
with urllib.request.urlopen(url, timeout=30) as r:
|
|||
|
|
return json.loads(r.read())
|
|||
|
|
|
|||
|
|
|
|||
|
|
def post(url: str, data: dict) -> dict:
|
|||
|
|
body = json.dumps(data).encode()
|
|||
|
|
req = urllib.request.Request(url, data=body, headers={"Content-Type": "application/json"})
|
|||
|
|
with urllib.request.urlopen(req, timeout=60) as r:
|
|||
|
|
return json.loads(r.read())
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> None:
|
|||
|
|
# 1) compare
|
|||
|
|
ev = chat("对比几种策略", "e2e-m2")
|
|||
|
|
blocks = [e for e in ev if e.get("type") == "block"]
|
|||
|
|
assert any(b["block"]["type"] == "scenario-cards" for b in blocks), blocks
|
|||
|
|
cards = next(b["block"]["props"]["cards"] for b in blocks if b["block"]["type"] == "scenario-cards")
|
|||
|
|
print("compare OK", len(cards), "cards:", [c["label"] for c in cards])
|
|||
|
|
tl0 = get("http://localhost:8000/api/timeline")
|
|||
|
|
v0 = len(tl0["versions"])
|
|||
|
|
|
|||
|
|
# 2) apply best
|
|||
|
|
best = min(cards, key=lambda c: (c["kpi"]["totalTardiness"], c["kpi"]["conflictCount"]))
|
|||
|
|
r = post(
|
|||
|
|
"http://localhost:8000/api/actions/scenario/apply",
|
|||
|
|
{"sessionId": "e2e-m2", "strategy": best["strategy"], "engine": best["engine"]},
|
|||
|
|
)
|
|||
|
|
print("apply OK", (r.get("message") or "")[:80])
|
|||
|
|
tl1 = get("http://localhost:8000/api/timeline")
|
|||
|
|
assert len(tl1["versions"]) >= v0, (v0, len(tl1["versions"]))
|
|||
|
|
print("versions", v0, "->", len(tl1["versions"]))
|
|||
|
|
|
|||
|
|
# 3) checkpoint create
|
|||
|
|
ev2 = chat("存个档 叫 e2e基线", "e2e-m2")
|
|||
|
|
text = "".join(e.get("text", "") for e in ev2 if e.get("type") == "token")
|
|||
|
|
print("checkpoint create:", text[:100])
|
|||
|
|
tl2 = get("http://localhost:8000/api/timeline")
|
|||
|
|
assert tl2["checkpoints"], tl2
|
|||
|
|
pair = tl2["checkpoints"][-1]
|
|||
|
|
print("pair", pair["pairId"], pair["label"])
|
|||
|
|
|
|||
|
|
# 4) rollback stage + approve
|
|||
|
|
ev3 = chat(f"回滚到检查点 {pair['pairId']}", "e2e-m2")
|
|||
|
|
blocks3 = [e for e in ev3 if e.get("type") == "block"]
|
|||
|
|
assert any(b["block"]["type"] == "confirm-card" for b in blocks3), blocks3
|
|||
|
|
cid = next(b["block"]["props"]["confirmId"] for b in blocks3 if b["block"]["type"] == "confirm-card")
|
|||
|
|
print("rollback staged", cid)
|
|||
|
|
r2 = post(
|
|||
|
|
"http://localhost:8000/api/actions/confirm",
|
|||
|
|
{"sessionId": "e2e-m2", "confirmId": cid, "approve": True},
|
|||
|
|
)
|
|||
|
|
print("rollback OK", (r2.get("message") or "")[:100])
|
|||
|
|
print("E2E M2 PASS")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|