mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-29 18:23:59 +08:00
记忆增加时间,以便大模型判断先后关系
This commit is contained in:
@@ -7,6 +7,7 @@ class MemoryProvider(MemoryProviderBase):
|
|||||||
def __init__(self, config):
|
def __init__(self, config):
|
||||||
super().__init__(config)
|
super().__init__(config)
|
||||||
self.api_key = config.get("api_key", "")
|
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:
|
if len(self.api_key) == 0 or "你" in self.api_key:
|
||||||
logger.bind(tag=TAG).error("你还没配置Mem0ai的密钥,请在配置文件中配置密钥,否则无法提供记忆服务")
|
logger.bind(tag=TAG).error("你还没配置Mem0ai的密钥,请在配置文件中配置密钥,否则无法提供记忆服务")
|
||||||
self.use_mem0 = False
|
self.use_mem0 = False
|
||||||
@@ -27,7 +28,7 @@ class MemoryProvider(MemoryProviderBase):
|
|||||||
{"role": message.role, "content": message.content}
|
{"role": message.role, "content": message.content}
|
||||||
for message in msgs if message.role != "system"
|
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}")
|
logger.bind(tag=TAG).debug(f"Save memory result: {result}")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.bind(tag=TAG).error(f"保存记忆失败: {str(e)}")
|
logger.bind(tag=TAG).error(f"保存记忆失败: {str(e)}")
|
||||||
@@ -39,9 +40,29 @@ class MemoryProvider(MemoryProviderBase):
|
|||||||
try:
|
try:
|
||||||
results = self.client.search(
|
results = self.client.search(
|
||||||
query,
|
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}")
|
logger.bind(tag=TAG).debug(f"Query results: {memories_str}")
|
||||||
return memories_str
|
return memories_str
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|||||||
Reference in New Issue
Block a user