You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.9 KiB
79 lines
2.9 KiB
#!/usr/bin/env python3 |
|
""" |
|
mem0 Commands for OpenClaw |
|
""" |
|
|
|
from .mem0_client import Mem0Client |
|
|
|
mem0 = Mem0Client() |
|
|
|
async def handle_memory_command(args: str, context: dict) -> str: |
|
"""处理 /memory 命令""" |
|
parts = args.split(' ', 2) |
|
action = parts[0] if parts else 'help' |
|
|
|
user_id = context.get('user_id', 'default') |
|
agent_id = context.get('agent_id', 'main') |
|
|
|
if action == 'add': |
|
content = parts[1] if len(parts) > 1 else '' |
|
return await handle_add_memory(content, user_id, agent_id) |
|
elif action == 'search': |
|
query = parts[1] if len(parts) > 1 else '' |
|
return await handle_search_memory(query, user_id, agent_id) |
|
elif action == 'list': |
|
return await handle_list_memories(user_id, agent_id) |
|
elif action == 'delete': |
|
memory_id = parts[1] if len(parts) > 1 else '' |
|
return await handle_delete_memory(memory_id, user_id, agent_id) |
|
elif action == 'sync': |
|
return await handle_sync_memory(user_id, agent_id) |
|
elif action == 'status': |
|
return handle_status(user_id, agent_id) |
|
else: |
|
return get_help_text() |
|
|
|
async def handle_add_memory(content: str, user_id: str, agent_id: str) -> str: |
|
if not content: |
|
return "❌ 请提供记忆内容" |
|
result = mem0.add([{"role": "user", "content": content}], user_id=user_id, agent_id=agent_id) |
|
if 'error' in result: |
|
return f"❌ 添加失败：{result['error']}" |
|
return f"✅ 记忆已添加：{content[:100]}..." |
|
|
|
async def handle_search_memory(query: str, user_id: str, agent_id: str) -> str: |
|
if not query: |
|
return "❌ 请提供关键词" |
|
results = mem0.search(query, user_id=user_id, agent_id=agent_id, limit=5) |
|
if not results: |
|
return f"🔍 未找到相关记忆" |
|
response = f"🔍 找到 {len(results)} 条记忆:\n\n" |
|
for i, mem in enumerate(results, 1): |
|
response += f"{i}. {mem.get('memory', 'N/A')}\n" |
|
return response |
|
|
|
async def handle_list_memories(user_id: str, agent_id: str) -> str: |
|
results = mem0.get_all(user_id=user_id, agent_id=agent_id) |
|
if not results: |
|
return "📭 暂无记忆" |
|
return f"📋 共 {len(results)} 条记忆" |
|
|
|
async def handle_delete_memory(memory_id: str, user_id: str, agent_id: str) -> str: |
|
if not memory_id: |
|
return "❌ 请提供 ID" |
|
success = mem0.delete(memory_id, user_id=user_id, agent_id=agent_id) |
|
return "✅ 已删除" if success else "❌ 删除失败" |
|
|
|
async def handle_sync_memory(user_id: str, agent_id: str) -> str: |
|
return "🔄 同步功能开发中" |
|
|
|
def handle_status(user_id: str, agent_id: str) -> str: |
|
status = mem0.get_status() |
|
response = "📊 mem0 状态:\n" |
|
response += f"本地：{'✅' if status['local_initialized'] else '❌'}\n" |
|
response += f"共享：{'✅' if status['master_initialized'] else '❌'}\n" |
|
return response |
|
|
|
def get_help_text() -> str: |
|
return """📖 用法：/memory <命令> |
|
命令：add, search, list, delete, sync, status"""
|
|
|