diff --git a/src/memos/api/client.py b/src/memos/api/client.py index 1129ddddf..91bc86829 100644 --- a/src/memos/api/client.py +++ b/src/memos/api/client.py @@ -177,7 +177,9 @@ def search_memory( if retry == MAX_RETRY_COUNT - 1: raise - def get_memory(self, user_id: str, include_preference: str) -> MemOSGetMemoryResponse | None: + def get_memory( + self, user_id: str, include_preference: bool = True, page: int = 1, size: int = 10 + ) -> MemOSGetMemoryResponse | None: """get memories""" # Validate required parameters self._validate_required_params(include_preference=include_preference, user_id=user_id) @@ -186,6 +188,8 @@ def get_memory(self, user_id: str, include_preference: str) -> MemOSGetMemoryRes payload = { "include_preference": include_preference, "user_id": user_id, + "page": page, + "size": size, } for retry in range(MAX_RETRY_COUNT): diff --git a/src/memos/api/product_models.py b/src/memos/api/product_models.py index ee7a45c2d..d5f301c9d 100644 --- a/src/memos/api/product_models.py +++ b/src/memos/api/product_models.py @@ -872,8 +872,8 @@ class GetMemoryData(BaseModel): memory_detail_list: list[MemoryDetail] = Field( default_factory=list, alias="memory_detail_list", description="List of memory details" ) - message_detail_list: list[MessageDetail] | None = Field( - None, alias="message_detail_list", description="List of message details (usually None)" + preference_detail_list: list[MessageDetail] | None = Field( + None, alias="preference_detail_list", description="List of preference detail" ) @@ -1025,7 +1025,7 @@ class MemOSGetMemoryResponse(BaseModel): code: int = Field(..., description="Response status code") message: str = Field(..., description="Response message") - data: SearchMemoryData = Field(..., description="Get results data") + data: GetMemoryData = Field(..., description="Get results data") @property def memories(self) -> list[MemoryDetail]: @@ -1033,15 +1033,10 @@ def memories(self) -> list[MemoryDetail]: return self.data.memory_detail_list @property - def preferences(self) -> list[MemoryDetail]: + def preferences(self) -> list[MessageDetail] | None: """Convenient access to preference list.""" return self.data.preference_detail_list - @property - def tool_memories(self) -> list[MemoryDetail]: - """Convenient access to tool_memory list.""" - return self.data.tool_memory_detail_list - class MemOSGetKnowledgebaseFileResponse(BaseModel): """Response model for get KnowledgebaseFile operation based on actual API."""