#!/usr/bin/env python3
"""
mem0 功能测试脚本
"""

import sys
import os
sys.path.insert(0, '/root/.openclaw/workspace/skills/mem0-integration')

from mem0_client import Mem0Client

print("=" * 60)
print("🧪 mem0 功能测试")
print("=" * 60)

# 测试 1：初始化
print("\n🔍 测试 1: 初始化 mem0 Client...")
try:
    client = Mem0Client()
    print("✅ Client 创建成功")
except Exception as e:
    print(f"❌ Client 创建失败：{e}")
    sys.exit(1)

# 测试 2：状态检查
print("\n📊 测试 2: 状态检查...")
status = client.get_status()
print(f"  本地记忆：{'✅' if status['local_initialized'] else '❌'}")
print(f"  共享记忆：{'✅' if status['master_initialized'] else '❌'}")

if not status['local_initialized']:
    print("\n⚠️  本地记忆未初始化，可能原因:")
    print("  1. Qdrant 未启动")
    print("  2. 配置错误")
    print("  3. mem0ai 版本问题")
    sys.exit(1)

# 测试 3：添加记忆
print("\n📝 测试 3: 添加记忆...")
test_content = "测试记忆：OpenClaw mem0 集成测试"
result = client.add(
    [{"role": "user", "content": test_content}],
    user_id="test_user",
    agent_id="main"
)

if 'error' in result:
    print(f"❌ 添加失败：{result['error']}")
else:
    print(f"✅ 记忆已添加")
    print(f"  内容：{test_content}")

# 测试 4：搜索记忆
print("\n🔍 测试 4: 搜索记忆...")
results = client.search("OpenClaw", user_id="test_user", agent_id="main", limit=5)
print(f"  找到 {len(results)} 条记忆")

if results:
    for i, mem in enumerate(results, 1):
        memory_text = mem.get('memory', 'N/A')
        print(f"    {i}. {memory_text[:100]}...")
else:
    print("  ⚠️  未找到记忆（可能 Qdrant 集合为空）")

# 测试 5：获取所有记忆
print("\n📋 测试 5: 获取所有记忆...")
all_memories = client.get_all(user_id="test_user", agent_id="main")
print(f"  共 {len(all_memories)} 条记忆")

# 测试 6：状态报告
print("\n📊 测试 6: 最终状态...")
final_status = client.get_status()
print(f"  本地 Qdrant: {final_status['config']['local']['vector_store']['config']['host']}:{final_status['config']['local']['vector_store']['config']['port']}")
print(f"  中心 Qdrant: {final_status['config']['master']['vector_store']['config']['host']}:{final_status['config']['master']['vector_store']['config']['port']}")

print("\n" + "=" * 60)
print("✅ 测试完成")
print("=" * 60)