112 lines
3.4 KiB
Python
112 lines
3.4 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
E2E: 演示 Agent「演示-主链路委派invoke」执行成功，并校验父子执行链。
|
|||
|
|
|
|||
|
|
前置:
|
|||
|
|
- 已运行 create_router_invoke_demo_agent.py
|
|||
|
|
- 平台 API 可访问
|
|||
|
|
|
|||
|
|
用法:
|
|||
|
|
cd backend && .\\venv\\Scripts\\python.exe scripts/e2e_router_invoke_demo.py
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import time
|
|||
|
|
from typing import Any, Dict, Optional
|
|||
|
|
|
|||
|
|
import requests
|
|||
|
|
|
|||
|
|
BASE = os.getenv("PLATFORM_BASE_URL", "http://127.0.0.1:8037").rstrip("/")
|
|||
|
|
USER = os.getenv("PLATFORM_USERNAME", "admin")
|
|||
|
|
PWD = os.getenv("PLATFORM_PASSWORD", "123456")
|
|||
|
|
DEMO_AGENT_NAME = os.getenv("DEMO_AGENT_NAME", "演示-主链路委派invoke")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _login_headers() -> Dict[str, str]:
|
|||
|
|
r = requests.post(
|
|||
|
|
f"{BASE}/api/v1/auth/login",
|
|||
|
|
data={"username": USER, "password": PWD},
|
|||
|
|
headers={"Content-Type": "application/x-www-form-urlencoded"},
|
|||
|
|
timeout=15,
|
|||
|
|
)
|
|||
|
|
r.raise_for_status()
|
|||
|
|
token = r.json().get("access_token")
|
|||
|
|
if not token:
|
|||
|
|
raise RuntimeError("无 access_token")
|
|||
|
|
return {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _find_demo_id(h: Dict[str, str]) -> Optional[str]:
|
|||
|
|
r = requests.get(
|
|||
|
|
f"{BASE}/api/v1/agents", params={"search": DEMO_AGENT_NAME, "limit": 50}, headers=h, timeout=30
|
|||
|
|
)
|
|||
|
|
r.raise_for_status()
|
|||
|
|
for a in r.json() or []:
|
|||
|
|
if a.get("name") == DEMO_AGENT_NAME:
|
|||
|
|
return a.get("id")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _wait_done(h: Dict[str, str], eid: str, timeout_s: float = 180.0) -> Dict[str, Any]:
|
|||
|
|
t0 = time.time()
|
|||
|
|
while time.time() - t0 < timeout_s:
|
|||
|
|
r = requests.get(f"{BASE}/api/v1/executions/{eid}", headers=h, timeout=30)
|
|||
|
|
r.raise_for_status()
|
|||
|
|
d = r.json()
|
|||
|
|
st = d.get("status")
|
|||
|
|
if st in ("completed", "failed"):
|
|||
|
|
return d
|
|||
|
|
time.sleep(1.2)
|
|||
|
|
raise TimeoutError(eid)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> int:
|
|||
|
|
h = _login_headers()
|
|||
|
|
aid = _find_demo_id(h)
|
|||
|
|
if not aid:
|
|||
|
|
print(f"未找到 Agent: {DEMO_AGENT_NAME}，请先运行 create_router_invoke_demo_agent.py", file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
|
|||
|
|
body = {
|
|||
|
|
"agent_id": aid,
|
|||
|
|
"input_data": {"query": "E2E 委派演示：请简短回复收到。", "user_id": "e2e_router_user"},
|
|||
|
|
}
|
|||
|
|
cr = requests.post(f"{BASE}/api/v1/executions", headers=h, json=body, timeout=30)
|
|||
|
|
if cr.status_code >= 400:
|
|||
|
|
print(cr.status_code, cr.text[:1200], file=sys.stderr)
|
|||
|
|
return 2
|
|||
|
|
eid = cr.json()["id"]
|
|||
|
|
print(f"execution={eid}")
|
|||
|
|
|
|||
|
|
done = _wait_done(h, eid)
|
|||
|
|
if done.get("status") != "completed":
|
|||
|
|
print("failed:", done.get("error_message"), file=sys.stderr)
|
|||
|
|
return 3
|
|||
|
|
|
|||
|
|
sm = requests.get(
|
|||
|
|
f"{BASE}/api/v1/execution-logs/executions/{eid}/chain/summary", headers=h, timeout=30
|
|||
|
|
)
|
|||
|
|
if sm.status_code == 200:
|
|||
|
|
s = sm.json()
|
|||
|
|
print(
|
|||
|
|
"chain_summary:",
|
|||
|
|
s.get("total_executions"),
|
|||
|
|
s.get("status_count"),
|
|||
|
|
s.get("total_execution_time_ms"),
|
|||
|
|
)
|
|||
|
|
if (s.get("total_executions") or 0) < 2:
|
|||
|
|
print("警告: 期望至少 2 条执行记录（父+子），请确认 API 已重启到最新代码", file=sys.stderr)
|
|||
|
|
elif sm.status_code == 404:
|
|||
|
|
print("chain/summary 404：当前 API 进程可能未加载最新路由，跳过汇总校验")
|
|||
|
|
else:
|
|||
|
|
print("chain/summary", sm.status_code, sm.text[:300])
|
|||
|
|
|
|||
|
|
print("E2E 通过")
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
raise SystemExit(main())
|