klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix request mem0 without org_id raise error (#2121)

Browse Source
This commit is contained in:
非法操作
2025-01-06 18:46:13 +08:00
committed by GitHub
parent e9188a51fe
commit d4dbed9dbd
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
mem0/client/main.py
View File

@@ -145,7 +145,8 @@ class MemoryClient:
Raises: Raises:
APIError: If the API request fails. APIError: If the API request fails.
""" """
response = self.client.get(f"/v1/memories/{memory_id}/") params = self._prepare_params()
response = self.client.get(f"/v1/memories/{memory_id}/", params=params)
response.raise_for_status() response.raise_for_status()
capture_client_event("client.get", self, {"memory_id": memory_id}) capture_client_event("client.get", self, {"memory_id": memory_id})
return response.json() return response.json()
@@ -219,7 +220,8 @@ class MemoryClient:
Dict[str, Any]: The response from the server. Dict[str, Any]: The response from the server.
""" """
capture_client_event("client.update", self, {"memory_id": memory_id}) capture_client_event("client.update", self, {"memory_id": memory_id})
response = self.client.put(f"/v1/memories/{memory_id}/", json={"text": data}) params = self._prepare_params()
response = self.client.put(f"/v1/memories/{memory_id}/", json={"text": data}, params=params)
response.raise_for_status() response.raise_for_status()
return response.json() return response.json()
@@ -236,7 +238,8 @@ class MemoryClient:
Raises: Raises:
APIError: If the API request fails. APIError: If the API request fails.
""" """
response = self.client.delete(f"/v1/memories/{memory_id}/") params = self._prepare_params()
response = self.client.delete(f"/v1/memories/{memory_id}/", params=params)
response.raise_for_status() response.raise_for_status()
capture_client_event("client.delete", self, {"memory_id": memory_id}) capture_client_event("client.delete", self, {"memory_id": memory_id})
return response.json() return response.json()
@@ -273,7 +276,8 @@ class MemoryClient:
Raises: Raises:
APIError: If the API request fails. APIError: If the API request fails.
""" """
response = self.client.get(f"/v1/memories/{memory_id}/history/") params = self._prepare_params()
response = self.client.get(f"/v1/memories/{memory_id}/history/", params=params)
response.raise_for_status() response.raise_for_status()
capture_client_event("client.history", self, {"memory_id": memory_id}) capture_client_event("client.history", self, {"memory_id": memory_id})
return response.json() return response.json()
@@ -454,7 +458,8 @@ class AsyncMemoryClient:
@api_error_handler @api_error_handler
async def get(self, memory_id: str) -> Dict[str, Any]: async def get(self, memory_id: str) -> Dict[str, Any]:
response = await self.async_client.get(f"/v1/memories/{memory_id}/") params = self.sync_client._prepare_params()
response = await self.async_client.get(f"/v1/memories/{memory_id}/", params=params)
response.raise_for_status() response.raise_for_status()
capture_client_event("async_client.get", self.sync_client, {"memory_id": memory_id}) capture_client_event("async_client.get", self.sync_client, {"memory_id": memory_id})
return response.json() return response.json()
@@ -489,14 +494,16 @@ class AsyncMemoryClient:
@api_error_handler @api_error_handler
async def update(self, memory_id: str, data: str) -> Dict[str, Any]: async def update(self, memory_id: str, data: str) -> Dict[str, Any]:
response = await self.async_client.put(f"/v1/memories/{memory_id}/", json={"text": data}) params = self.sync_client._prepare_params()
response = await self.async_client.put(f"/v1/memories/{memory_id}/", json={"text": data}, params=params)
response.raise_for_status() response.raise_for_status()
capture_client_event("async_client.update", self.sync_client, {"memory_id": memory_id}) capture_client_event("async_client.update", self.sync_client, {"memory_id": memory_id})
return response.json() return response.json()
@api_error_handler @api_error_handler
async def delete(self, memory_id: str) -> Dict[str, Any]: async def delete(self, memory_id: str) -> Dict[str, Any]:
response = await self.async_client.delete(f"/v1/memories/{memory_id}/") params = self.sync_client._prepare_params()
response = await self.async_client.delete(f"/v1/memories/{memory_id}/", params=params)
response.raise_for_status() response.raise_for_status()
capture_client_event("async_client.delete", self.sync_client, {"memory_id": memory_id}) capture_client_event("async_client.delete", self.sync_client, {"memory_id": memory_id})
return response.json() return response.json()
@@ -511,7 +518,8 @@ class AsyncMemoryClient:
@api_error_handler @api_error_handler
async def history(self, memory_id: str) -> List[Dict[str, Any]]: async def history(self, memory_id: str) -> List[Dict[str, Any]]:
response = await self.async_client.get(f"/v1/memories/{memory_id}/history/") params = self.sync_client._prepare_params()
response = await self.async_client.get(f"/v1/memories/{memory_id}/history/", params=params)
response.raise_for_status() response.raise_for_status()
capture_client_event("async_client.history", self.sync_client, {"memory_id": memory_id}) capture_client_event("async_client.history", self.sync_client, {"memory_id": memory_id})
return response.json() return response.json()