0608161c82
补齐平台模板与场景 DSL、预算控制、执行看板和企业场景脚本，增强 Windows 启动/迁移与前端代理和聊天会话记忆，修复执行创建阶段 500 与异步链路排障体验。 Made-with: Cursor
93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
与 e2e_platform_capability_smoke.py 相同业务验证，但走 FastAPI TestClient（不经过 TCP 8037）。
|
||
|
||
适用：本机存在多个进程占用 8037、HTTP 命中旧实例导致 /platform 路由 404 时。
|
||
|
||
用法:
|
||
cd backend && .\\venv\\Scripts\\python.exe scripts/e2e_platform_capability_smoke_testclient.py
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
import sys
|
||
import time
|
||
|
||
from fastapi.testclient import TestClient
|
||
|
||
# 确保加载 backend 目录
|
||
def main() -> int:
|
||
os.environ.setdefault("PLATFORM_BASE_URL", "http://testserver")
|
||
from app.main import app
|
||
|
||
c = TestClient(app)
|
||
r = c.post(
|
||
"/api/v1/auth/login",
|
||
data={"username": os.getenv("PLATFORM_USERNAME", "admin"), "password": os.getenv("PLATFORM_PASSWORD", "123456")},
|
||
headers={"Content-Type": "application/x-www-form-urlencoded"},
|
||
)
|
||
if r.status_code != 200:
|
||
print("login:", r.status_code, r.text[:500], file=sys.stderr)
|
||
return 1
|
||
token = r.json().get("access_token")
|
||
h = {"Authorization": f"Bearer {token}"}
|
||
|
||
g = c.get("/api/v1/platform/scene-templates", headers=h)
|
||
if g.status_code != 200:
|
||
print("scene-templates:", g.status_code, g.text[:500], file=sys.stderr)
|
||
return 2
|
||
templates = g.json() or []
|
||
tid = "template_customer_service"
|
||
ids = [x.get("id") for x in templates if isinstance(x, dict)]
|
||
if tid not in ids:
|
||
tid = ids[0] if ids else None
|
||
if not tid:
|
||
print("无模板", file=sys.stderr)
|
||
return 3
|
||
|
||
name = os.getenv("SMOKE_AGENT_NAME") or f"冒烟_客服DSL_TC_{int(time.time())}"
|
||
cr = c.post(
|
||
"/api/v1/platform/agents/from-template",
|
||
headers=h,
|
||
json={
|
||
"template_id": tid,
|
||
"name": name,
|
||
"description": "e2e testclient",
|
||
"parameters": {"temperature": 0.35},
|
||
"budget_config": {"max_steps": 50, "max_llm_invocations": 5, "max_tool_calls": 20},
|
||
},
|
||
)
|
||
if cr.status_code not in (200, 201):
|
||
print("from-template:", cr.status_code, cr.text[:800], file=sys.stderr)
|
||
return 4
|
||
aid = cr.json().get("id")
|
||
print("agent:", aid, cr.json().get("name"))
|
||
|
||
body = {
|
||
"agent_id": aid,
|
||
"input_data": {
|
||
"query": os.getenv("SMOKE_USER_QUERY") or "你好，请用一句话说明你能帮我做什么。",
|
||
"scenario_dsl": {
|
||
"version": "1",
|
||
"scene": "customer_service_smoke_tc",
|
||
"goal": "验证 TestClient 执行",
|
||
"constraints": ["简短"],
|
||
"deliverables": ["一句回复"],
|
||
"acceptance": ["有内容"],
|
||
"payload": {"channel": "smoke_tc"},
|
||
},
|
||
},
|
||
}
|
||
er = c.post("/api/v1/executions", headers=h, json=body)
|
||
if er.status_code not in (200, 201):
|
||
print("executions:", er.status_code, er.text[:800], file=sys.stderr)
|
||
return 5
|
||
eid = er.json()["id"]
|
||
print("execution:", eid, "(异步任务需 Celery；此处仅校验创建成功)")
|
||
print("OK — TestClient 路径：模板 + DSL 创建执行记录成功。请用 Celery 跑完后查 GET /api/v1/executions/{id}")
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main())
|