Patch: get_memory adds the page size parameter function and the filtering of user id

src/memos/api/handlers/memory_handler.py

@@ -180,7 +180,12 @@ def handle_get_memories(
     get_mem_req: GetMemoryRequest, naive_mem_cube: NaiveMemCube
 ) -> GetMemoryResponse:
     # TODO: Implement get memory with filter
-    memories = naive_mem_cube.text_mem.get_all(user_name=get_mem_req.mem_cube_id)["nodes"]
+    memories = naive_mem_cube.text_mem.get_all(
+        user_name=get_mem_req.mem_cube_id,
+        user_id=get_mem_req.user_id,
+        page=get_mem_req.page,
+        page_size=get_mem_req.page_size,
+    )["nodes"]
     preferences: list[TextualMemoryItem] = []
     if get_mem_req.include_preference and naive_mem_cube.pref_mem is not None:
         filter_params: dict[str, Any] = {}

src/memos/api/product_models.py

@@ -772,6 +772,13 @@ class GetMemoryRequest(BaseRequest):
     mem_cube_id: str = Field(..., description="Cube ID")
     user_id: str | None = Field(None, description="User ID")
     include_preference: bool = Field(True, description="Whether to handle preference memory")
+    page: int | None = Field(
+        None,
+        description="Page number (starts from 1). If None, exports all data without pagination.",
+    )
+    page_size: int | None = Field(
+        None, description="Number of items per page. If None, exports all data without pagination."
+    )
 
 
 class DeleteMemoryRequest(BaseRequest):

src/memos/graph_dbs/nebular.py

@@ -1207,7 +1207,7 @@ def clear(self, user_name: str | None = None) -> None:
 
     @timed
     def export_graph(
-        self, include_embedding: bool = False, user_name: str | None = None
+        self, include_embedding: bool = False, user_name: str | None = None, **kwargs
     ) -> dict[str, Any]:
         """
         Export all graph nodes and edges in a structured form.

src/memos/graph_dbs/polardb.py

@@ -2507,6 +2507,7 @@ def export_graph(
         user_name: str | None = None,
         page: int | None = None,
         page_size: int | None = None,
+        **kwargs,
     ) -> dict[str, Any]:
         """
         Export all graph nodes and edges in a structured form.

src/memos/memories/textual/tree.py

@@ -321,12 +321,20 @@ def get_by_ids(
     ) -> list[TextualMemoryItem]:
         raise NotImplementedError
 
-    def get_all(self, user_name: str | None = None) -> dict:
+    def get_all(
+        self,
+        user_name: str,
+        user_id: str | None = None,
+        page: int | None = None,
+        page_size: int | None = None,
+    ) -> dict:
         """Get all memories.
         Returns:
             list[TextualMemoryItem]: List of all memories.
         """
-        all_items = self.graph_store.export_graph(user_name=user_name)
+        all_items = self.graph_store.export_graph(
+            user_name=user_name, user_id=user_id, page=page, page_size=page_size
+        )
         return all_items
 
     def delete(self, memory_ids: list[str], user_name: str | None = None) -> None: