Skip to content

Commit cd7b7e8

Browse files
CarltonXiangfridayLharvey_xiangCaralHsi
authored
Fix/api client (#357)
* fix: api client get_message models * fix: format error --------- Co-authored-by: chunyu li <[email protected]> Co-authored-by: harvey_xiang <[email protected]> Co-authored-by: CaralHsi <[email protected]>
1 parent ea8e631 commit cd7b7e8

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

src/memos/api/client.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ def get_message(
5050
)
5151
response.raise_for_status()
5252
response_data = response.json()
53+
5354
return MemOSGetMessagesResponse(**response_data)
5455
except Exception as e:
5556
logger.error(f"Failed to get messages (retry {retry + 1}/3): {e}")
@@ -74,6 +75,7 @@ def add_message(
7475
)
7576
response.raise_for_status()
7677
response_data = response.json()
78+
7779
return MemOSAddResponse(**response_data)
7880
except Exception as e:
7981
logger.error(f"Failed to add memory (retry {retry + 1}/3): {e}")
@@ -102,6 +104,7 @@ def search_memory(
102104
)
103105
response.raise_for_status()
104106
response_data = response.json()
107+
105108
return MemOSSearchResponse(**response_data)
106109
except Exception as e:
107110
logger.error(f"Failed to search memory (retry {retry + 1}/3): {e}")

src/memos/api/product_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ class GetMessagesData(BaseModel):
191191
"""Data model for get messages response based on actual API."""
192192

193193
message_detail_list: list[MessageDetail] = Field(
194-
default_factory=list, alias="memory_detail_list", description="List of message details"
194+
default_factory=list, alias="message_detail_list", description="List of message details"
195195
)
196196

197197

0 commit comments

Comments
 (0)