|
|
#!/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)
|
|
|
|
