File tree Expand file tree Collapse file tree 2 files changed +4
-1
lines changed
Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Original file line number Diff line number Diff 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 } " )
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments