Skip to content

Commit 7119091

Browse files
authored
Fix/default add (#333)
* feat: add default processing in mem-reader * feat: not include embedding * feat: add logger to detect remove-old-memory error
1 parent 4fa115b commit 7119091

File tree

1 file changed

+21
-9
lines changed
  • src/memos/memories/textual/tree_text_memory/organize

1 file changed

+21
-9
lines changed

src/memos/memories/textual/tree_text_memory/organize/manager.py

Lines changed: 21 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import traceback
12
import uuid
23

34
from concurrent.futures import as_completed
@@ -65,15 +66,26 @@ def add(self, memories: list[TextualMemoryItem]) -> list[str]:
6566
except Exception as e:
6667
logger.exception("Memory processing error: ", exc_info=e)
6768

68-
self.graph_store.remove_oldest_memory(
69-
memory_type="WorkingMemory", keep_latest=self.memory_size["WorkingMemory"]
70-
)
71-
self.graph_store.remove_oldest_memory(
72-
memory_type="LongTermMemory", keep_latest=self.memory_size["LongTermMemory"]
73-
)
74-
self.graph_store.remove_oldest_memory(
75-
memory_type="UserMemory", keep_latest=self.memory_size["UserMemory"]
76-
)
69+
try:
70+
self.graph_store.remove_oldest_memory(
71+
memory_type="WorkingMemory", keep_latest=self.memory_size["WorkingMemory"]
72+
)
73+
except Exception:
74+
logger.warning(f"Remove WorkingMemory error: {traceback.format_exc()}")
75+
76+
try:
77+
self.graph_store.remove_oldest_memory(
78+
memory_type="LongTermMemory", keep_latest=self.memory_size["LongTermMemory"]
79+
)
80+
except Exception:
81+
logger.warning(f"Remove LongTermMemory error: {traceback.format_exc()}")
82+
83+
try:
84+
self.graph_store.remove_oldest_memory(
85+
memory_type="UserMemory", keep_latest=self.memory_size["UserMemory"]
86+
)
87+
except Exception:
88+
logger.warning(f"Remove UserMemory error: {traceback.format_exc()}")
7789

7890
self._refresh_memory_size()
7991
return added_ids

0 commit comments

Comments
 (0)