diff --git a/src/memos/api/handlers/memory_handler.py b/src/memos/api/handlers/memory_handler.py index 83f51428c..dc72d0112 100644 --- a/src/memos/api/handlers/memory_handler.py +++ b/src/memos/api/handlers/memory_handler.py @@ -187,6 +187,9 @@ def handle_get_memories( def handle_delete_memories(delete_mem_req: DeleteMemoryRequest, naive_mem_cube: NaiveMemCube): + logger.info( + f"[Delete memory request] writable_cube_ids: {delete_mem_req.writable_cube_ids}, memory_ids: {delete_mem_req.memory_ids}" + ) # Validate that only one of memory_ids, file_ids, or filter is provided provided_params = [ delete_mem_req.memory_ids is not None, @@ -201,7 +204,8 @@ def handle_delete_memories(delete_mem_req: DeleteMemoryRequest, naive_mem_cube: try: if delete_mem_req.memory_ids is not None: - naive_mem_cube.text_mem.delete(delete_mem_req.memory_ids) + for cube_id in delete_mem_req.writable_cube_ids: + naive_mem_cube.text_mem.delete(delete_mem_req.memory_ids, user_name=cube_id) if naive_mem_cube.pref_mem is not None: naive_mem_cube.pref_mem.delete(delete_mem_req.memory_ids) elif delete_mem_req.file_ids is not None: diff --git a/src/memos/api/product_models.py b/src/memos/api/product_models.py index d2e7c5946..16ae86638 100644 --- a/src/memos/api/product_models.py +++ b/src/memos/api/product_models.py @@ -111,7 +111,7 @@ class ChatRequest(BaseRequest): ) # ==== Extended capabilities ==== - internet_search: bool = Field(True, description="Whether to use internet search") + internet_search: bool = Field(False, description="Whether to use internet search") threshold: float = Field(0.5, description="Threshold for filtering references") # ==== Backward compatibility ==== @@ -699,7 +699,7 @@ class APIChatCompleteRequest(BaseRequest): ) # ==== Extended capabilities ==== - internet_search: bool = Field(True, description="Whether to use internet search") + internet_search: bool = Field(False, description="Whether to use internet search") threshold: float = Field(0.5, description="Threshold for filtering references") # ==== Backward compatibility ==== @@ -728,6 +728,7 @@ class GetMemoryRequest(BaseRequest): class DeleteMemoryRequest(BaseRequest): """Request model for deleting memories.""" + writable_cube_ids: list[str] = Field(..., description="Writable cube IDs") memory_ids: list[str] | None = Field(None, description="Memory IDs") file_ids: list[str] | None = Field(None, description="File IDs") filter: dict[str, Any] | None = Field(None, description="Filter for the memory")