mirror of
https://github.com/open-webui/open-webui.git
synced 2025-12-14 13:25:20 +00:00
45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
|
|
from mem0 import MemoryClient
|
|
import os
|
|
from logging import getLogger
|
|
log = getLogger(__name__)
|
|
|
|
mem0_api_key = os.getenv("MEM0_API_KEY")
|
|
memory_client = MemoryClient(api_key=mem0_api_key)
|
|
|
|
async def mem0_search(user_id: str, chat_id: str, last_message: str) -> list[str]:
|
|
"""
|
|
预留的 Mem0 检索接口：当前为占位实现。
|
|
未来可替换为实际检索逻辑，返回若干相关记忆条目（字符串）。
|
|
增加 chat_id 便于按会话窗口区分/隔离记忆。
|
|
"""
|
|
try:
|
|
# TODO: 接入真实 Mem0 检索
|
|
log.info(f"mem0_search called with user_id: {user_id}, chat_id: {chat_id}, last_message: {last_message}")
|
|
serach_rst = memory_client.search(
|
|
query=last_message,
|
|
filters={"user_id": user_id}
|
|
)
|
|
memories=serach_rst["results"] if "results" in serach_rst else serach_rst
|
|
log.info(f"mem0_search found {len(memories)} memories")
|
|
return [mem["text"] for mem in memories]
|
|
except Exception as e:
|
|
log.debug(f"Mem0 search failed: {e}")
|
|
return []
|
|
|
|
|
|
async def mem0_delete(user_id: str, chat_id: str) -> bool:
|
|
"""
|
|
删除指定用户在指定 chat 窗口下的所有 Mem0 相关记忆（占位实现）。
|
|
未来可替换为实际删除逻辑。
|
|
"""
|
|
try:
|
|
# TODO: 接入真实删除逻辑（如按 chat_id 过滤）
|
|
log.info(f"mem0_delete called with user_id: {user_id}, chat_id: {chat_id}")
|
|
memory_client.delete(
|
|
filters={"user_id": user_id}
|
|
)
|
|
return True
|
|
except Exception as e:
|
|
log.debug(f"Mem0 delete failed: {e}")
|
|
return False
|