Skip to content

Commit 4fe7529

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

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

src/memos/api/handlers/memory_handler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,9 @@ 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(delete_mem_req.memory_ids)
170+
naive_mem_cube.text_mem.delete(
171+
delete_mem_req.memory_ids, user_name=delete_mem_req.writable_cube_id
172+
)
171173
if naive_mem_cube.pref_mem is not None:
172174
naive_mem_cube.pref_mem.delete(delete_mem_req.memory_ids)
173175
elif delete_mem_req.file_ids is not None:

src/memos/api/product_models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +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")
230231
memory_ids: list[str] | None = Field(None, description="Memory IDs")
231232
file_ids: list[str] | None = Field(None, description="File IDs")
232233
filter: dict[str, Any] | None = Field(None, description="Filter for the memory")

0 commit comments

Comments
 (0)