117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
|
"""
|
|||
|
|
对「知你客服9号」做多轮 API 测试（不默认重启 Celery，避免打断本机 Worker）。
|
|||
|
|
轮次：①自我介绍姓名 ②陈述偏好（供摘要/关键词召回）③闲聊 ④问姓名 + 问偏好
|
|||
|
|
|
|||
|
|
用法:
|
|||
|
|
cd backend && .\\venv\\Scripts\\python.exe scripts/e2e_zhini9_test.py
|
|||
|
|
环境变量: API_BASE, E2E_AGENT_NAME（默认 知你客服9号）
|
|||
|
|
"""
|
|||
|
|
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", "知你客服9号")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> int:
|
|||
|
|
os.chdir(BACKEND_DIR)
|
|||
|
|
sys.path.insert(0, BACKEND_DIR)
|
|||
|
|
|
|||
|
|
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_z9_{uuid.uuid4().hex[:10]}"
|
|||
|
|
print(f"agent_id={agent.id} name={agent.name} 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("超时")
|
|||
|
|
|
|||
|
|
def extract_reply(out: dict) -> str:
|
|||
|
|
od = out.get("output_data") or {}
|
|||
|
|
if isinstance(od, dict):
|
|||
|
|
r = od.get("result")
|
|||
|
|
if isinstance(r, str):
|
|||
|
|
return r[:500]
|
|||
|
|
return json.dumps(od, ensure_ascii=False)[:500]
|
|||
|
|
|
|||
|
|
rounds = [
|
|||
|
|
"我的名字叫阿九",
|
|||
|
|
"记住：我最爱吃火锅，不喜欢甜食。",
|
|||
|
|
"今天天气不错吧？",
|
|||
|
|
"我叫什么名字？你还记得我喜欢吃什么吗？",
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
with httpx.Client(base_url=API_BASE, timeout=300.0) as client:
|
|||
|
|
for i, q in enumerate(rounds, 1):
|
|||
|
|
r = client.post(
|
|||
|
|
"/api/v1/executions",
|
|||
|
|
json={"agent_id": str(agent.id), "input_data": {"query": q, "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)
|
|||
|
|
print(f"--- 第{i}轮 ---\nQ: {q}\nA: {extract_reply(out)}\n")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
from app.core.config import settings
|
|||
|
|
import redis as redis_lib
|
|||
|
|
|
|||
|
|
url = getattr(settings, "REDIS_URL", None) or "redis://localhost:6379/0"
|
|||
|
|
rc = redis_lib.from_url(url, decode_responses=True)
|
|||
|
|
key = f"user_memory_{uid}"
|
|||
|
|
raw = rc.get(key)
|
|||
|
|
print(f"Redis {key}:", "有" if raw else "无")
|
|||
|
|
if raw:
|
|||
|
|
mem = json.loads(raw)
|
|||
|
|
print("conversation_summary 前120字:", str(mem.get("conversation_summary", ""))[:120])
|
|||
|
|
print("user_profile:", mem.get("user_profile"))
|
|||
|
|
print("history 条数:", len(mem.get("conversation_history") or []))
|
|||
|
|
except Exception as ex:
|
|||
|
|
print("Redis 检查:", ex)
|
|||
|
|
|
|||
|
|
finally:
|
|||
|
|
db.close()
|
|||
|
|
print("完成")
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
raise SystemExit(main())
|