Add delete users method (#1683)

This commit is contained in:
Dev Khant
2024-08-13 22:35:04 +05:30
committed by GitHub
parent 31ef9135e7
commit f19dfe70d7
2 changed files with 33 additions and 0 deletions

View File

@@ -245,6 +245,17 @@ class MemoryClient:
capture_client_event("client.users", self)
return response.json()
@api_error_handler
def delete_users(self) -> Dict[str, Any]:
"""Delete all users, agents, or sessions."""
entities = self.users()
for entity in entities["results"]:
response = self.client.delete(f"/entities/{entity['type']}/{entity['id']}/")
response.raise_for_status()
capture_client_event("client.delete_users", self)
return {"message": "All users, agents, and sessions deleted."}
def reset(self):
"""Reset the client. (Not implemented)