"""
对「知你客服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())