Skip to content

Commit b4fe866

Browse files
author
yuan.wang
committed
fix bug in get and delete
1 parent 43a7903 commit b4fe866

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

src/memos/api/handlers/memory_handler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def handle_get_subgraph(
160160

161161
def handle_get_memories(get_mem_req: GetMemoryRequest, naive_mem_cube: Any) -> GetMemoryResponse:
162162
# TODO: Implement get memory with filter
163-
memories = naive_mem_cube.text_mem.get_all(user_name=get_mem_req.mem_cube_id)
163+
memories = naive_mem_cube.text_mem.get_all(user_name=get_mem_req.mem_cube_id)["nodes"]
164164
filter_params: dict[str, Any] = {}
165165
if get_mem_req.user_id is not None:
166166
filter_params["user_id"] = get_mem_req.user_id
@@ -170,7 +170,7 @@ def handle_get_memories(get_mem_req: GetMemoryRequest, naive_mem_cube: Any) -> G
170170
return GetMemoryResponse(
171171
message="Memories retrieved successfully",
172172
data={
173-
"text_mem": [format_memory_item(mem) for mem in memories],
173+
"text_mem": memories,
174174
"pref_mem": [format_memory_item(mem) for mem in preferences],
175175
},
176176
)
@@ -188,5 +188,5 @@ def handle_delete_memories(delete_mem_req: DeleteMemoryRequest, naive_mem_cube:
188188
)
189189
return DeleteMemoryResponse(
190190
message="Memories deleted successfully",
191-
data="success",
191+
data={"status": "success"},
192192
)

src/memos/api/routers/server_router.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ def get_all_memories(memory_req: GetMemoryPlaygroundRequest):
245245
)
246246

247247

248-
@router.get("/get_memory", summary="Get memories for user", response_model=GetMemoryResponse)
248+
@router.post("/get_memory", summary="Get memories for user", response_model=GetMemoryResponse)
249249
def get_memories(memory_req: GetMemoryRequest):
250250
return handlers.memory_handler.handle_get_memories(
251251
get_mem_req=memory_req,
@@ -257,4 +257,6 @@ def get_memories(memory_req: GetMemoryRequest):
257257
"/delete_memory", summary="Delete memories for user", response_model=DeleteMemoryResponse
258258
)
259259
def delete_memories(memory_req: DeleteMemoryRequest):
260-
return handlers.memory_handler.handle_delete_memories(memory_req)
260+
return handlers.memory_handler.handle_delete_memories(
261+
delete_mem_req=memory_req, naive_mem_cube=naive_mem_cube
262+
)

0 commit comments

Comments
 (0)