Skip to content

Commit 5b29f8c

Browse files
committed
refactor: not showing userinput to working memory
1 parent 000e383 commit 5b29f8c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/memos/mem_scheduler/general_scheduler.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
ANSWER_LABEL,
1010
DEFAULT_MAX_QUERY_KEY_WORDS,
1111
QUERY_LABEL,
12+
WORKING_MEMORY_TYPE,
1213
MemCubeID,
1314
UserID,
1415
)
@@ -248,6 +249,9 @@ def _add_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
248249
mem_type = mem_item.metadata.memory_type
249250
mem_content = mem_item.memory
250251

252+
if mem_type == WORKING_MEMORY_TYPE:
253+
continue
254+
251255
self.log_adding_memory(
252256
memory=mem_content,
253257
memory_type=mem_type,

0 commit comments

Comments
 (0)