Skip to content

Commit 6074e63

Browse files
author
yuan.wang
committed
feat: modify some code
1 parent 91abeab commit 6074e63

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

src/memos/multi_mem_cube/single_cube.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,8 @@ class SingleCubeView(MemCubeView):
5858
@timed
5959
def add_memories(self, add_req: APIADDRequest) -> list[dict[str, Any]]:
6060
"""
61-
Unified memory addition handling (text + preference memories).
62-
Now text_mem handles both types through integrated extraction.
61+
This is basically your current handle_add_memories logic,
62+
but scoped to a single cube_id.
6363
"""
6464
sync_mode = add_req.async_mode or self._get_sync_mode()
6565
self.logger.info(
@@ -73,7 +73,12 @@ def add_memories(self, add_req: APIADDRequest) -> list[dict[str, Any]]:
7373
project_id=add_req.project_id,
7474
)
7575

76-
# Call _process_text_mem which now handles both text and preference memories
76+
target_session_id = add_req.session_id or "default_session"
77+
self.logger.info(
78+
f"[SingleCubeView] cube={self.cube_id} "
79+
f"Processing add with mode={sync_mode}, session={target_session_id}"
80+
)
81+
7782
all_memories = self._process_text_mem(add_req, user_context, sync_mode)
7883

7984
self.logger.info(f"[SingleCubeView] cube={self.cube_id} total_results={len(all_memories)}")
@@ -110,7 +115,7 @@ def search_memories(self, search_req: APISearchRequest) -> dict[str, Any]:
110115
# Unified search through _search_text (includes all memory types)
111116
all_formatted_memories = self._search_text(search_req, user_context, search_mode)
112117

113-
# Build result with unified processing (handles text, tool, skill, and preference)
118+
# Build result with unified processing
114119
memories_result = post_process_textual_mem(
115120
memories_result,
116121
all_formatted_memories,

0 commit comments

Comments
 (0)