Skip to content

Commit 31557de

Browse files
committed
fix: robuster memory parser
1 parent 6a4234b commit 31557de

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/memos/mem_reader/simple_struct.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,9 @@ def _process_chat_data(self, scene_data_info, info):
5858
metadata=TreeNodeTextualMemoryMetadata(
5959
user_id=info.get("user_id"),
6060
session_id=info.get("session_id"),
61-
memory_type=memory_i_raw.get("memory_type", ""),
61+
memory_type=memory_i_raw.get("memory_type", "")
62+
.replace("长期记忆", "LongTermMemory")
63+
.replace("用户记忆", "UserMemory"),
6264
status="activated",
6365
tags=memory_i_raw.get("tags", [])
6466
if type(memory_i_raw.get("tags", [])) is list

0 commit comments

Comments
 (0)