125 lines
4.6 KiB
Python
125 lines
4.6 KiB
Python
|
"""E2E：知你客服16号 单轮「你好」（需 API + Celery + LLM）。
|
|||
|
|
|
|||
|
|
环境变量:
|
|||
|
|
USE_TESTCLIENT=1 走 FastAPI TestClient（不经 HTTP），仅校验创建执行 201。
|
|||
|
|
API_BASE 默认 http://127.0.0.1:8037；若该端口上 /health 无 checks，可改用本仓库单独起的实例，例如 http://127.0.0.1:8040。
|
|||
|
|
E2E_AGENT_NAME / E2E_QUERY
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import time
|
|||
|
|
import uuid
|
|||
|
|
|
|||
|
|
BACKEND_DIR = __file__.rsplit("scripts", 1)[0]
|
|||
|
|
API_BASE = os.environ.get("API_BASE", "http://127.0.0.1:8037")
|
|||
|
|
AGENT_NAME = os.environ.get("E2E_AGENT_NAME", "知你客服16号")
|
|||
|
|
QUERY = os.environ.get("E2E_QUERY", "你好")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> int:
|
|||
|
|
os.chdir(BACKEND_DIR)
|
|||
|
|
sys.path.insert(0, BACKEND_DIR)
|
|||
|
|
|
|||
|
|
if os.environ.get("USE_TESTCLIENT") == "1":
|
|||
|
|
from fastapi.testclient import TestClient
|
|||
|
|
from app.main import app
|
|||
|
|
from app.core.database import SessionLocal
|
|||
|
|
from app.core.security import create_access_token
|
|||
|
|
from app.models.agent import Agent
|
|||
|
|
from app.models.user import User
|
|||
|
|
|
|||
|
|
db = SessionLocal()
|
|||
|
|
try:
|
|||
|
|
agent = db.query(Agent).filter(Agent.name == AGENT_NAME).first()
|
|||
|
|
if not agent:
|
|||
|
|
print(f"数据库中未找到「{AGENT_NAME}」", file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
owner = db.query(User).filter(User.id == agent.user_id).first()
|
|||
|
|
user = owner or db.query(User).first()
|
|||
|
|
if not user:
|
|||
|
|
print("无可用用户", file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
token = create_access_token(data={"sub": user.id, "username": user.username})
|
|||
|
|
h = {"Authorization": f"Bearer {token}"}
|
|||
|
|
finally:
|
|||
|
|
db.close()
|
|||
|
|
|
|||
|
|
c = TestClient(app)
|
|||
|
|
er = c.post(
|
|||
|
|
"/api/v1/executions",
|
|||
|
|
headers=h,
|
|||
|
|
json={
|
|||
|
|
"agent_id": str(agent.id),
|
|||
|
|
"input_data": {"query": QUERY, "user_id": f"e2e_z16_tc_{uuid.uuid4().hex[:8]}"},
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
print("TestClient POST /executions:", er.status_code, er.text[:800])
|
|||
|
|
if er.status_code not in (200, 201):
|
|||
|
|
return 2
|
|||
|
|
print("OK — 本仓库 API 创建执行成功（异步需 Celery Worker 才会跑完）")
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
import httpx
|
|||
|
|
from app.core.database import SessionLocal
|
|||
|
|
from app.core.security import create_access_token
|
|||
|
|
from app.models.agent import Agent
|
|||
|
|
from app.models.user import User
|
|||
|
|
|
|||
|
|
db = SessionLocal()
|
|||
|
|
try:
|
|||
|
|
agent = db.query(Agent).filter(Agent.name == AGENT_NAME).first()
|
|||
|
|
if not agent:
|
|||
|
|
print(f"数据库中未找到「{AGENT_NAME}」", file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
owner = db.query(User).filter(User.id == agent.user_id).first()
|
|||
|
|
user = owner or db.query(User).first()
|
|||
|
|
if not user:
|
|||
|
|
print("无可用用户", file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
token = create_access_token(data={"sub": user.id, "username": user.username})
|
|||
|
|
headers = {"Authorization": f"Bearer {token}"}
|
|||
|
|
uid = f"e2e_z16_{uuid.uuid4().hex[:10]}"
|
|||
|
|
print(f"agent_id={agent.id} name={agent.name} Q={QUERY!r} user_id={uid}\n")
|
|||
|
|
|
|||
|
|
def poll(client: httpx.Client, execution_id: str, timeout: float = 300.0) -> dict:
|
|||
|
|
t0 = time.time()
|
|||
|
|
while time.time() - t0 < timeout:
|
|||
|
|
r = client.get(f"/api/v1/executions/{execution_id}", headers=headers)
|
|||
|
|
r.raise_for_status()
|
|||
|
|
data = r.json()
|
|||
|
|
st = data.get("status")
|
|||
|
|
if st == "completed":
|
|||
|
|
return data
|
|||
|
|
if st == "failed":
|
|||
|
|
print("error:", data.get("error_message"), file=sys.stderr)
|
|||
|
|
raise RuntimeError("执行失败")
|
|||
|
|
time.sleep(1)
|
|||
|
|
raise TimeoutError("超时")
|
|||
|
|
|
|||
|
|
with httpx.Client(base_url=API_BASE, timeout=300.0) as client:
|
|||
|
|
r = client.post(
|
|||
|
|
"/api/v1/executions",
|
|||
|
|
json={"agent_id": str(agent.id), "input_data": {"query": QUERY, "user_id": uid}},
|
|||
|
|
headers=headers,
|
|||
|
|
)
|
|||
|
|
if r.status_code >= 400:
|
|||
|
|
print(r.text, file=sys.stderr)
|
|||
|
|
r.raise_for_status()
|
|||
|
|
eid = r.json()["id"]
|
|||
|
|
out = poll(client, eid)
|
|||
|
|
od = out.get("output_data") or {}
|
|||
|
|
rtxt = od.get("result") if isinstance(od, dict) else None
|
|||
|
|
print("--- 回复 ---")
|
|||
|
|
print((rtxt or json.dumps(od, ensure_ascii=False))[:4000])
|
|||
|
|
finally:
|
|||
|
|
db.close()
|
|||
|
|
print("\n完成")
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
raise SystemExit(main())
|