Skip to content

Commit b999ff3

Browse files
whipser030黑布林CaralHsifridayL
authored
Patch: use manager _add_memories_parallel (#721)
* update reader and search strategy * set strategy reader and search config * fix install problem * fix * fix test * turn off graph recall * turn off graph recall * turn off graph recall * fix Searcher input bug * fix Searcher * fix Search * fix bug * adjust strategy reader * adjust strategy reader * adjust search config input * reformat code * re pr * format repair * fix time issue * develop feedback process * feedback handler configuration * upgrade feedback using * add threshold * update prompt * update prompt * fix handler * add feedback scheduler * add handler change node update * add handler change node update * add handler change node update * add handler change node update * fix interface input * add chunk and ratio filter * update stopwords * fix messages queue * add seach_by_keywords_LIKE * add doc filter * add retrieve query * add retrieve queies * patch info filter * add log and make embedding safety net * add log and make embedding safety net * deduplicate add objects * use _add_memories_parallel --------- Co-authored-by: 黑布林 <[email protected]> Co-authored-by: CaralHsi <[email protected]> Co-authored-by: chunyu li <[email protected]>
1 parent ed31a3d commit b999ff3

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/memos/mem_feedback/feedback.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def _pure_add(self, user_name: str, feedback_content: str, feedback_time: str, i
133133
memories = self.mem_reader.get_memory(scene_data, type="chat", info=info)
134134
to_add_memories = [item for scene in memories for item in scene]
135135
added_ids = self._retry_db_operation(
136-
lambda: self.memory_manager.add(to_add_memories, user_name=user_name)
136+
lambda: self.memory_manager.add(to_add_memories, user_name=user_name, use_batch=False)
137137
)
138138
logger.info(
139139
f"[Feedback Core: _pure_add] Pure added {len(added_ids)} memories for user {user_name}."
@@ -224,10 +224,10 @@ def _single_add_operation(
224224

225225
to_add_memory.id = ""
226226
added_ids = self._retry_db_operation(
227-
lambda: self.memory_manager.add([to_add_memory], user_name=user_name, mode=async_mode)
227+
lambda: self.memory_manager.add([to_add_memory], user_name=user_name, use_batch=False)
228228
)
229229

230-
logger.info(f"[Memory Feedback ADD] memory id: {added_ids[0]}")
230+
logger.info(f"[Memory Feedback ADD] memory id: {added_ids!s}")
231231
return {"id": added_ids[0], "text": to_add_memory.memory}
232232

233233
def _single_update_operation(

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,7 @@ def _add_memories_parallel(
131131
added_ids.extend(ids)
132132
except Exception as e:
133133
logger.exception("Memory processing error: ", exc_info=e)
134+
logger.info(f"[MemoryManager: _add_memories_parallel] Added {len(added_ids)} memories")
134135
return added_ids
135136

136137
def _add_memories_batch(

0 commit comments

Comments
 (0)