Skip to content

Commit d632dde

Browse files
wustzdyzZhangSir
andauthored
add getMemory sdk (#827)
* fix: update get_memory sdk * fix: update get_memory sdk * fix: update get_memory sdk * fix: update get_memory sdk --------- Co-authored-by: Elvis <1693372324@qq.com>
1 parent 85860ce commit d632dde

2 files changed

Lines changed: 9 additions & 10 deletions

File tree

src/memos/api/client.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,9 @@ def search_memory(
177177
if retry == MAX_RETRY_COUNT - 1:
178178
raise
179179

180-
def get_memory(self, user_id: str, include_preference: str) -> MemOSGetMemoryResponse | None:
180+
def get_memory(
181+
self, user_id: str, include_preference: bool = True, page: int = 1, size: int = 10
182+
) -> MemOSGetMemoryResponse | None:
181183
"""get memories"""
182184
# Validate required parameters
183185
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
186188
payload = {
187189
"include_preference": include_preference,
188190
"user_id": user_id,
191+
"page": page,
192+
"size": size,
189193
}
190194

191195
for retry in range(MAX_RETRY_COUNT):

src/memos/api/product_models.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -872,8 +872,8 @@ class GetMemoryData(BaseModel):
872872
memory_detail_list: list[MemoryDetail] = Field(
873873
default_factory=list, alias="memory_detail_list", description="List of memory details"
874874
)
875-
message_detail_list: list[MessageDetail] | None = Field(
876-
None, alias="message_detail_list", description="List of message details (usually None)"
875+
preference_detail_list: list[MessageDetail] | None = Field(
876+
None, alias="preference_detail_list", description="List of preference detail"
877877
)
878878

879879

@@ -1025,23 +1025,18 @@ class MemOSGetMemoryResponse(BaseModel):
10251025

10261026
code: int = Field(..., description="Response status code")
10271027
message: str = Field(..., description="Response message")
1028-
data: SearchMemoryData = Field(..., description="Get results data")
1028+
data: GetMemoryData = Field(..., description="Get results data")
10291029

10301030
@property
10311031
def memories(self) -> list[MemoryDetail]:
10321032
"""Convenient access to memory list."""
10331033
return self.data.memory_detail_list
10341034

10351035
@property
1036-
def preferences(self) -> list[MemoryDetail]:
1036+
def preferences(self) -> list[MessageDetail] | None:
10371037
"""Convenient access to preference list."""
10381038
return self.data.preference_detail_list
10391039

1040-
@property
1041-
def tool_memories(self) -> list[MemoryDetail]:
1042-
"""Convenient access to tool_memory list."""
1043-
return self.data.tool_memory_detail_list
1044-
10451040

10461041
class MemOSGetKnowledgebaseFileResponse(BaseModel):
10471042
"""Response model for get KnowledgebaseFile operation based on actual API."""

0 commit comments

Comments
 (0)