agent init
This commit is contained in:
penghanyuan
295
api/test_conversation.py
Normal file
295
api/test_conversation.py
Normal file
@@ -0,0 +1,295 @@
|
||||
#!/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())
|
||||
Reference in New Issue
Block a user