From d1a22d0b690c46abc2ac02a8cd88595c6f88b29f Mon Sep 17 00:00:00 2001
From: Charles Packer <packercharles@gmail.com>
Date: Sun, 10 Mar 2024 18:06:00 -0700
Subject: [PATCH] feat: add a `last_run` field to the agent state model (#1124)

---
 memgpt/models/pydantic_models.py        |  1 +
 memgpt/server/rest_api/agents/config.py |  2 +-
 memgpt/server/server.py                 | 20 ++++++++++++++++++--
 3 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/memgpt/models/pydantic_models.py b/memgpt/models/pydantic_models.py
index 11cf8fb9..85bda3d7 100644
--- a/memgpt/models/pydantic_models.py
+++ b/memgpt/models/pydantic_models.py
@@ -58,6 +58,7 @@ class AgentStateModel(BaseModel):
     user_id: uuid.UUID = Field(..., description="The unique identifier of the user associated with the agent.")
 
     # timestamps
+    # created_at: datetime = Field(default_factory=datetime.now, description="The unix timestamp of when the agent was created.")
     created_at: int = Field(..., description="The unix timestamp of when the agent was created.")
 
     # preset information
diff --git a/memgpt/server/rest_api/agents/config.py b/memgpt/server/rest_api/agents/config.py
index 069c2e97..fa1bc3c0 100644
--- a/memgpt/server/rest_api/agents/config.py
+++ b/memgpt/server/rest_api/agents/config.py
@@ -51,7 +51,7 @@ def validate_agent_name(name: str) -> str:
 def setup_agents_config_router(server: SyncServer, interface: QueuingInterface, password: str):
     get_current_user_with_server = partial(partial(get_current_user, server), password)
 
-    @router.get("/agents", tags=["agents"], response_model=GetAgentResponse)
+    @router.get("/agents/config", tags=["agents"], response_model=GetAgentResponse)
     def get_agent_config(
         agent_id: str = Query(..., description="Unique identifier of the agent whose config is requested."),
         user_id: uuid.UUID = Depends(get_current_user_with_server),
diff --git a/memgpt/server/server.py b/memgpt/server/server.py
index 13c0c965..74c2d8da 100644
--- a/memgpt/server/server.py
+++ b/memgpt/server/server.py
@@ -684,16 +684,32 @@ class SyncServer(LockingServer):
         }
         return agent_config
 
-    def list_agents(self, user_id: uuid.UUID) -> dict:
+    def list_agents(self, user_id: uuid.UUID, add_last_run: bool = True) -> dict:
         """List all available agents to a user"""
         if self.ms.get_user(user_id=user_id) is None:
             raise ValueError(f"User user_id={user_id} does not exist")
 
         agents_states = self.ms.list_agents(user_id=user_id)
+        agents_states_dicts = [self._agent_state_to_config(state) for state in agents_states]
+
+        if add_last_run:
+            # TODO add a get_message_obj_from_message_id(...) function
+            #      this would allow grabbing Message.created_by without having to load the agent object
+            for agent_state, return_dict in zip(agents_states, agents_states_dicts):
+                # Get the agent object (loaded in memory)
+                memgpt_agent = self._get_or_load_agent(user_id=user_id, agent_id=agent_state.id)
+
+                # Retrive the Message object via the recall storage or by directly access _messages
+                last_msg_obj = memgpt_agent._messages[-1]
+
+                # Update the return to include the last_run info
+                # NOTE: 'last_run' is just the timestamp on the latest message in the buffer
+                return_dict["last_run"] = last_msg_obj.created_at
+
         logger.info(f"Retrieved {len(agents_states)} agents for user {user_id}:\n{[vars(s) for s in agents_states]}")
         return {
             "num_agents": len(agents_states),
-            "agents": [self._agent_state_to_config(state) for state in agents_states],
+            "agents": agents_states_dicts,
         }
 
     def get_agent(self, user_id: uuid.UUID, agent_id: uuid.UUID):