diff --git a/core/providers/memory/mem0ai/mem0ai.py b/core/providers/memory/mem0ai/mem0ai.py index de4eb29e..e319b542 100644 --- a/core/providers/memory/mem0ai/mem0ai.py +++ b/core/providers/memory/mem0ai/mem0ai.py @@ -7,6 +7,7 @@ class MemoryProvider(MemoryProviderBase): def __init__(self, config): super().__init__(config) self.api_key = config.get("api_key", "") + self.api_version = config.get("api_version", "v1.1") if len(self.api_key) == 0 or "你" in self.api_key: logger.bind(tag=TAG).error("你还没配置Mem0ai的密钥,请在配置文件中配置密钥,否则无法提供记忆服务") self.use_mem0 = False @@ -27,7 +28,7 @@ class MemoryProvider(MemoryProviderBase): {"role": message.role, "content": message.content} for message in msgs if message.role != "system" ] - result = self.client.add(messages, user_id=self.role_id, output_format="v1.1") + result = self.client.add(messages, user_id=self.role_id, output_format=self.api_version) logger.bind(tag=TAG).debug(f"Save memory result: {result}") except Exception as e: logger.bind(tag=TAG).error(f"保存记忆失败: {str(e)}") @@ -39,9 +40,29 @@ class MemoryProvider(MemoryProviderBase): try: results = self.client.search( query, - user_id= self.role_id, + user_id=self.role_id, + output_format=self.api_version ) - memories_str = "\n".join(f"- {entry['memory']}" for entry in results) + if not results or 'results' not in results: + return "" + + # Format each memory entry with its update time up to minutes + memories = [] + for entry in results['results']: + # Split timestamp and get date + time up to minutes + timestamp = entry.get('updated_at', '').split('.')[0] # Remove milliseconds + if timestamp: + try: + # Parse and reformat the timestamp + dt = timestamp.replace('T', ' ').split(':') # Split time components + formatted_time = f"{dt[0]}:{dt[1]}:{dt[2]}" # Keep only HH:MM:SS + except: + formatted_time = timestamp + memory = entry.get('memory', '') + if timestamp and memory: + memories.append(f"[{formatted_time}] {memory}") + + memories_str = "\n".join(f"- {memory}" for memory in memories) logger.bind(tag=TAG).debug(f"Query results: {memories_str}") return memories_str except Exception as e: