Skip to content

Commit b97c52a

Browse files
author
yuan.wang
committed
fix bug
1 parent 4fe7529 commit b97c52a

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

src/memos/api/handlers/memory_handler.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -167,9 +167,8 @@ def handle_delete_memories(delete_mem_req: DeleteMemoryRequest, naive_mem_cube:
167167

168168
try:
169169
if delete_mem_req.memory_ids is not None:
170-
naive_mem_cube.text_mem.delete(
171-
delete_mem_req.memory_ids, user_name=delete_mem_req.writable_cube_id
172-
)
170+
for cube_id in delete_mem_req.writable_cube_ids:
171+
naive_mem_cube.text_mem.delete(delete_mem_req.memory_ids, user_name=cube_id)
173172
if naive_mem_cube.pref_mem is not None:
174173
naive_mem_cube.pref_mem.delete(delete_mem_req.memory_ids)
175174
elif delete_mem_req.file_ids is not None:

src/memos/api/product_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ class APIChatCompleteRequest(BaseRequest):
227227
class DeleteMemoryRequest(BaseRequest):
228228
"""Request model for deleting memories."""
229229

230-
writable_cube_id: str = Field(..., description="Writable cube IDs")
230+
writable_cube_ids: list[str] = Field(..., description="Writable cube IDs")
231231
memory_ids: list[str] | None = Field(None, description="Memory IDs")
232232
file_ids: list[str] | None = Field(None, description="File IDs")
233233
filter: dict[str, Any] | None = Field(None, description="Filter for the memory")

0 commit comments

Comments
 (0)