Fix: Use /add endpoint for adding memories (not /memories)
This commit is contained in:
@@ -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]:
|
||||
|
||||
Reference in New Issue
Block a user