fix: api client get_message models

Author: harvey_xiang

src/memos/api/client.py

@@ -26,7 +26,10 @@ def __init__(self, api_key: str | None = None, base_url: str | None = None):
         if not api_key:
             raise ValueError("MemOS API key is required")
 
-        self.headers = {"Content-Type": "application/json", "Authorization": f"Token {api_key}"}
+        self.headers = {
+          "Content-Type": "application/json",
+          "Authorization": f"Token {api_key}"
+        }
 
     def _validate_required_params(self, **params):
         """Validate required parameters - if passed, they must not be empty"""
@@ -42,14 +45,15 @@ def get_message(
         self._validate_required_params(user_id=user_id)
 
         url = f"{self.base_url}/get/message"
-        payload = {"user_id": user_id, "conversation_id": conversation_id}
+        payload = { "user_id": user_id, "conversation_id": conversation_id }
         for retry in range(MAX_RETRY_COUNT):
             try:
                 response = requests.post(
                     url, data=json.dumps(payload), headers=self.headers, timeout=30
                 )
                 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}")
@@ -66,14 +70,15 @@ def add_message(
         )
 
         url = f"{self.base_url}/add/message"
-        payload = {"messages": messages, "user_id": user_id, "conversation_id": conversation_id}
+        payload = { "messages": messages, "user_id": user_id, "conversation_id": conversation_id }
         for retry in range(MAX_RETRY_COUNT):
             try:
                 response = requests.post(
                     url, data=json.dumps(payload), headers=self.headers, timeout=30
                 )
                 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 +107,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}")

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"
     )