#!/usr/bin/env python3 """ 多轮对话测试脚本 测试对话引导 Agent 和回忆录整理功能 """ import asyncio import json import uuid import httpx import websockets from datetime import datetime # 配置 BASE_URL = "http://localhost:8000" WS_URL = "ws://localhost:8000" # 测试用户信息 TEST_PHONE = f"138{uuid.uuid4().hex[:8]}" # 随机手机号避免冲突 TEST_PASSWORD = "test123456" TEST_NICKNAME = "测试用户" # 模拟用户的多轮对话内容（关于童年和教育阶段） CONVERSATION_MESSAGES = [ # 童年阶段 "我出生在南方一个小镇，小时候跟奶奶住在一起。", "奶奶家有个小院子，夏天的时候我们经常坐在院子里乘凉，她给我讲故事。", "那段时光真的很美好，我记得奶奶总是给我做红烧肉，那是我最爱吃的菜。", "小时候最开心的事就是过年，可以放鞭炮，还能收到压岁钱。", # 教育阶段 "后来我去城里上学了，那是我第一次离开家，心里特别害怕。", "初中的时候遇到了一个很好的语文老师，她鼓励我多读书，对我影响很大。", "高考那年压力特别大，但最后还是考上了理想的大学。", ] class ConversationTester: """对话测试器""" def __init__(self): self.token = None self.user_id = None self.conversation_id = str(uuid.uuid4()) async def register_or_login(self): """注册或登录用户""" async with httpx.AsyncClient(timeout=30.0) as client: # 先尝试注册 print(f"\n📝 注册用户: {TEST_PHONE}") resp = await client.post( f"{BASE_URL}/api/auth/register", json={ "phone": TEST_PHONE, "password": TEST_PASSWORD, "nickname": TEST_NICKNAME } ) if resp.status_code == 201: data = resp.json() self.token = data["access_token"] print(f"✅ 注册成功!") elif resp.status_code == 400 and "已被注册" in resp.text: # 已注册，尝试登录 print(f"ℹ️ 用户已存在，尝试登录...") resp = await client.post( f"{BASE_URL}/api/auth/login", json={ "phone": TEST_PHONE, "password": TEST_PASSWORD } ) if resp.status_code == 200: data = resp.json() self.token = data["access_token"] print(f"✅ 登录成功!") else: raise Exception(f"登录失败: {resp.text}") else: raise Exception(f"注册失败: {resp.text}") print(f"🔑 Token: {self.token[:30]}...") async def get_memoir_state(self): """获取回忆录状态""" async with httpx.AsyncClient(timeout=60.0) as client: resp = await client.get( f"{BASE_URL}/api/memoir-state", headers={"Authorization": f"Bearer {self.token}"} ) if resp.status_code != 200: print(f" ⚠️ 状态API返回 {resp.status_code}: {resp.text[:200]}") return {"current_stage": "unknown", "covered_stages": [], "slots": {}} return resp.json() async def get_chapters(self): """获取章节列表""" async with httpx.AsyncClient(timeout=60.0) as client: resp = await client.get( f"{BASE_URL}/api/chapters", headers={"Authorization": f"Bearer {self.token}"} ) if resp.status_code != 200: print(f" ⚠️ 章节API返回 {resp.status_code}: {resp.text[:200]}") return [] return resp.json() async def get_book(self): """获取回忆录信息""" async with httpx.AsyncClient(timeout=60.0) as client: resp = await client.get( f"{BASE_URL}/api/books/current", headers={"Authorization": f"Bearer {self.token}"} ) if resp.status_code != 200: print(f" ⚠️ 回忆录API返回 {resp.status_code}: {resp.text[:200]}") return {"message": "获取失败"} return resp.json() async def run_conversation(self): """运行多轮对话""" print(f"\n🔗 连接 WebSocket: {self.conversation_id}") ws_url = f"{WS_URL}/ws/conversation/{self.conversation_id}?token={self.token}" async with websockets.connect(ws_url) as ws: # 接收连接确认 msg = await ws.recv() data = json.loads(msg) print(f"✅ 连接成功: {data['type']}") # 多轮对话 for i, user_message in enumerate(CONVERSATION_MESSAGES, 1): print(f"\n{'='*60}") print(f"📤 第 {i} 轮对话") print(f"{'='*60}") print(f"👤 用户: {user_message}") # 发送消息 await ws.send(json.dumps({ "type": "text", "data": {"text": user_message} })) # 接收 Agent 回复（可能是多条消息） try: while True: msg = await asyncio.wait_for(ws.recv(), timeout=30) data = json.loads(msg) if data["type"] == "agent_response": msg_data = data['data'] total = msg_data.get('total', 1) index = msg_data.get('index', 0) print(f"🤖 Agent: {msg_data['text']}") # 如果是最后一条消息，退出循环 if index >= total - 1: break elif data["type"] == "error": print(f"❌ 错误: {data['data']['message']}") break else: break except asyncio.TimeoutError: print("⏰ 等待响应超时") # 短暂等待，模拟真实对话节奏 await asyncio.sleep(1) # 结束对话 print(f"\n{'='*60}") print("📭 结束对话") print(f"{'='*60}") await ws.send(json.dumps({ "type": "end_conversation", "conversation_id": self.conversation_id })) try: # 结束时会触发 process_conversation_segments，可能需要更长时间 msg = await asyncio.wait_for(ws.recv(), timeout=60) data = json.loads(msg) if data['type'] == 'error': print(f"❌ 结束对话错误: {data['data'].get('message', 'unknown')}") else: print(f"✅ 对话结束: {data['type']}") except asyncio.TimeoutError: print("⏰ 等待结束确认超时（但后台处理可能仍在进行）") async def check_results(self): """检查回忆录生成结果""" print(f"\n{'='*60}") print("📊 检查结果") print(f"{'='*60}") # 等待后台处理完成（后台任务有5秒debounce + 多次LLM调用） # 每条消息可能触发2-3次LLM调用，7条消息 = 约20次调用，每次约3秒 print("\n⏳ 等待后台处理完成...") for i in range(6): # 60秒，每10秒检查一次 print(f" 等待中... {(i+1)*10}秒") await asyncio.sleep(10) # 获取回忆录状态 print("\n📋 回忆录状态:") state = await self.get_memoir_state() print(f" 当前阶段: {state.get('current_stage', 'N/A')}") print(f" 已完成阶段: {state.get('covered_stages', [])}") # 显示已填充的 slots slots = state.get('slots', {}) for stage, stage_slots in slots.items(): filled = [k for k, v in stage_slots.items() if v.get('snippet')] if filled: print(f" {stage} 已填充: {filled}") for slot_name in filled: snippet = stage_slots[slot_name].get('snippet', '') if snippet: print(f" - {slot_name}: {snippet[:50]}...") # 获取章节 print("\n📚 生成的章节:") chapters = await self.get_chapters() if chapters: for ch in chapters: is_new = "🆕" if ch.get("is_new") else "" print(f" {is_new} [{ch.get('category', 'N/A')}] {ch.get('title', 'N/A')}") else: print(" (暂无章节)") # 获取回忆录 print("\n📖 回忆录信息:") book = await self.get_book() if "message" not in book: print(f" 标题: {book.get('title', 'N/A')}") print(f" 总字数: {book.get('total_words', 0)}") print(f" 有更新: {'是' if book.get('has_update') else '否'}") else: print(f" {book.get('message', 'N/A')}") # 显示回忆录完整内容 if chapters: print(f"\n{'='*60}") print("📜 回忆录完整内容") print(f"{'='*60}") for ch in chapters: category = ch.get('category', 'N/A') title = ch.get('title', '未命名章节') content = ch.get('content', '') print(f"\n{'─'*60}") print(f"【{title}】") print(f"{'─'*60}") if content: print(content) else: print("(暂无内容)") print(f"\n{'='*60}") async def main(): """主函数""" print("=" * 60) print("🎭 Life Echo 多轮对话测试") print(f"⏰ 开始时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") print("=" * 60) tester = ConversationTester() try: # 1. 注册/登录 await tester.register_or_login() # 2. 查看初始状态 print("\n📋 初始回忆录状态:") state = await tester.get_memoir_state() print(f" 当前阶段: {state.get('current_stage', 'N/A')}") # 3. 运行多轮对话 await tester.run_conversation() # 4. 检查结果 await tester.check_results() except Exception as e: print(f"\n❌ 测试失败: {e}") import traceback traceback.print_exc() print("\n" + "=" * 60) print(f"⏰ 结束时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") print("=" * 60) if __name__ == "__main__": asyncio.run(main())