101 lines
3.8 KiB
Python
101 lines
3.8 KiB
Python
"""圆桌后台作业真网关 —— headless 验证脚本（Phase 3b）。
|
||
|
||
用法（在能起完整后端栈 + 配好真模型的环境）：
|
||
|
||
1) 用真网关启动后端：
|
||
ROUNDTABLE_JOB_GATEWAY=inprocess make gateway
|
||
（或 docker 里给 gateway 容器加该环境变量）
|
||
|
||
2) 跑本脚本，它会：POST 启动一个作业 → 轮询 GET 打印进度 → 跑到 done 打印报告长度。
|
||
PYTHONPATH=. uv run python scripts/probe_roundtable_job.py
|
||
|
||
环境变量：
|
||
GATEWAY_URL 默认 http://127.0.0.1:8001
|
||
AUTH_COOKIE 若后端开启鉴权，传 "session=xxx" 之类的 Cookie 头
|
||
PROBE_INTENT 自定义任务意图
|
||
PROBE_MODE recommend（默认）| chain
|
||
|
||
说明：鉴权关闭的 dev 模式无需 AUTH_COOKIE。真网关会进程内驱动 roundtable-coordinator /
|
||
各席位 / roundtable-report 真模型，耗时取决于模型，可能数分钟。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import asyncio
|
||
import os
|
||
import time
|
||
|
||
import httpx
|
||
|
||
GATEWAY_URL = os.getenv("GATEWAY_URL", "http://127.0.0.1:8001").rstrip("/")
|
||
AUTH_COOKIE = os.getenv("AUTH_COOKIE", "")
|
||
INTENT = os.getenv("PROBE_INTENT", "为一家中型制造企业设计一套数字化转型的整体方案")
|
||
MODE = os.getenv("PROBE_MODE", "recommend")
|
||
|
||
# 用 3 个内置圆桌席位（确保可见、已 seed）。
|
||
AGENTS = [
|
||
{"agentId": "roundtable-intelligence", "name": "情报收集"},
|
||
{"agentId": "roundtable-solution-design", "name": "方案设计"},
|
||
{"agentId": "roundtable-summary", "name": "总结陈述"},
|
||
]
|
||
|
||
TERMINAL = {"done", "error", "cancelled"}
|
||
|
||
|
||
def _headers() -> dict[str, str]:
|
||
h = {"Content-Type": "application/json"}
|
||
if AUTH_COOKIE:
|
||
h["Cookie"] = AUTH_COOKIE
|
||
return h
|
||
|
||
|
||
async def main() -> None:
|
||
async with httpx.AsyncClient(timeout=30.0, trust_env=False) as client:
|
||
print(f"[probe] POST {GATEWAY_URL}/api/roundtable-jobs mode={MODE}")
|
||
resp = await client.post(
|
||
f"{GATEWAY_URL}/api/roundtable-jobs",
|
||
headers=_headers(),
|
||
json={"intent": INTENT, "agents": AGENTS, "orchestrationMode": MODE},
|
||
)
|
||
if resp.status_code != 201:
|
||
print(f"[probe] start failed: HTTP {resp.status_code}: {resp.text[:500]}")
|
||
return
|
||
job = resp.json()
|
||
job_id = job["id"]
|
||
print(f"[probe] started job {job_id}")
|
||
|
||
last = ""
|
||
t0 = time.perf_counter()
|
||
while True:
|
||
await asyncio.sleep(2.0)
|
||
r = await client.get(f"{GATEWAY_URL}/api/roundtable-jobs/{job_id}", headers=_headers())
|
||
if r.status_code != 200:
|
||
print(f"[probe] get failed: HTTP {r.status_code}: {r.text[:300]}")
|
||
return
|
||
j = r.json()
|
||
line = (
|
||
f"status={j['status']} phase={j['phase']} cycle={j['cycle']} "
|
||
f"active={j.get('activeAgentId')} consensus={j['consensusPercentage']}%"
|
||
)
|
||
if line != last:
|
||
print(f"[probe] +{time.perf_counter() - t0:5.0f}s {line}")
|
||
last = line
|
||
if j["status"] in TERMINAL:
|
||
print(f"[probe] terminal: {j['status']}")
|
||
if j["status"] == "error":
|
||
print(f"[probe] error detail: {j.get('error')}")
|
||
# 读草稿 step3 验证报告已落库
|
||
if j.get("draftId"):
|
||
dr = await client.get(
|
||
f"{GATEWAY_URL}/api/roundtable-drafts/{j['draftId']}", headers=_headers()
|
||
)
|
||
if dr.status_code == 200:
|
||
step3 = dr.json().get("step3")
|
||
html_len = len((step3 or {}).get("html") or "")
|
||
print(f"[probe] draft.step3 html length = {html_len}")
|
||
return
|
||
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main())
|