From 6f1964c575054f2cbd147a9bcb1e043cf028d034 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 e70f0e0b..183b33f2 100644 --- a/letta/server/server.py +++ b/letta/server/server.py @@ -328,6 +328,15 @@ class SyncServer(Server): ) ) + 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: