MemGPT/tests/helpers/utils.py
2024-10-03 14:46:36 -07:00

12 lines
394 B
Python

from typing import Union
from letta import LocalClient, RESTClient
def cleanup(client: Union[LocalClient, RESTClient], agent_uuid: str):
# Clear all agents
for agent_state in client.list_agents():
if agent_state.name == agent_uuid:
client.delete_agent(agent_id=agent_state.id)
print(f"Deleted agent: {agent_state.name} with ID {str(agent_state.id)}")