diff --git a/src/memos/api/client.py b/src/memos/api/client.py index d45276f2..912f883a 100644 --- a/src/memos/api/client.py +++ b/src/memos/api/client.py @@ -50,6 +50,7 @@ def get_message( ) response.raise_for_status() response_data = response.json() + return MemOSGetMessagesResponse(**response_data) except Exception as e: logger.error(f"Failed to get messages (retry {retry + 1}/3): {e}") @@ -74,6 +75,7 @@ def add_message( ) response.raise_for_status() response_data = response.json() + return MemOSAddResponse(**response_data) except Exception as e: logger.error(f"Failed to add memory (retry {retry + 1}/3): {e}") @@ -102,6 +104,7 @@ def search_memory( ) response.raise_for_status() response_data = response.json() + return MemOSSearchResponse(**response_data) except Exception as e: logger.error(f"Failed to search memory (retry {retry + 1}/3): {e}") diff --git a/src/memos/api/product_models.py b/src/memos/api/product_models.py index 7e425415..2d03d294 100644 --- a/src/memos/api/product_models.py +++ b/src/memos/api/product_models.py @@ -191,7 +191,7 @@ class GetMessagesData(BaseModel): """Data model for get messages response based on actual API.""" message_detail_list: list[MessageDetail] = Field( - default_factory=list, alias="memory_detail_list", description="List of message details" + default_factory=list, alias="message_detail_list", description="List of message details" )