diff --git a/client.py b/client.py index 7b34cf8..51b856d 100644 --- a/client.py +++ b/client.py @@ -16,7 +16,7 @@ class LocalMem0Client: Expects Mem0 server at MEM0_BASE_URL with endpoints: - POST /search - GET /memories - - POST /memories + - POST /add """ def __init__(self, base_url: str, timeout: float = 10.0): @@ -30,11 +30,19 @@ class LocalMem0Client: } ) - def _request(self, method: str, endpoint: str, json: Optional[Dict] = None, params: Optional[Dict] = None) -> Dict: + def _request( + self, + method: str, + endpoint: str, + json: Optional[Dict] = None, + params: Optional[Dict] = None, + ) -> Dict: """Make HTTP request with error handling.""" url = f"{self.base_url}{endpoint}" try: - resp = self.session.request(method, url, json=json, params=params, timeout=self.timeout) + resp = self.session.request( + method, url, json=json, params=params, timeout=self.timeout + ) resp.raise_for_status() return resp.json() except requests.exceptions.Timeout: @@ -87,8 +95,15 @@ class LocalMem0Client: "agent_id": filters.get("agent_id"), } if not infer: - payload["messages"] = [{"role": "user", "content": messages[0].get("content", "") if isinstance(messages[0], dict) else messages[0]}] - return self._request("POST", "/memories", json=payload) + payload["messages"] = [ + { + "role": "user", + "content": messages[0].get("content", "") + if isinstance(messages[0], dict) + else messages[0], + } + ] + return self._request("POST", "/add", json=payload) @staticmethod def _unwrap_results(response: Any) -> List[Dict]: