From 1ce7aa5b8fbe4b4275ae82f92bd650be2c52603d Mon Sep 17 00:00:00 2001 From: Matthew Zhou Date: Thu, 21 Nov 2024 09:53:30 -0800 Subject: [PATCH] fix: Add save agents (#2080) --- letta/server/server.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/letta/server/server.py b/letta/server/server.py index e70f0e0be4..183b33f257 100644 --- a/letta/server/server.py +++ b/letta/server/server.py @@ -328,6 +328,15 @@ def __init__( ) ) + def save_agents(self): + """Saves all the agents that are in the in-memory object store""" + for agent_d in self.active_agents: + try: + save_agent(agent_d["agent"], self.ms) + logger.info(f"Saved agent {agent_d['agent_id']}") + except Exception as e: + logger.exception(f"Error occurred while trying to save agent {agent_d['agent_id']}:\n{e}") + def _get_agent(self, user_id: str, agent_id: str) -> Union[Agent, None]: """Get the agent object from the in-memory object store""" for d in self.active_agents: