diff --git a/letta/agent.py b/letta/agent.py index 1096668e..2b7441e3 100644 --- a/letta/agent.py +++ b/letta/agent.py @@ -1042,6 +1042,7 @@ class Agent(BaseAgent): num_archival_memory=agent_manager_passage_size, num_recall_memory=message_manager_size, num_tokens_external_memory_summary=num_tokens_external_memory_summary, + external_memory_summary=external_memory_summary, # top-level information context_window_size_max=self.agent_state.llm_config.context_window, context_window_size_current=num_tokens_used_total, diff --git a/letta/schemas/memory.py b/letta/schemas/memory.py index 797eac57..ab877949 100644 --- a/letta/schemas/memory.py +++ b/letta/schemas/memory.py @@ -30,6 +30,9 @@ class ContextWindowOverview(BaseModel): num_tokens_external_memory_summary: int = Field( ..., description="The number of tokens in the external memory summary (archival + recall metadata)." ) + external_memory_summary: str = Field( + ..., description="The metadata summary of the external memory sources (archival + recall metadata)." + ) # context window breakdown (in tokens) # this should all add up to context_window_size_current diff --git a/tests/test_server.py b/tests/test_server.py index 4775ed91..c2b77ea8 100644 --- a/tests/test_server.py +++ b/tests/test_server.py @@ -515,6 +515,7 @@ def test_get_context_window_overview(server: SyncServer, user, agent_id): assert overview.num_archival_memory is not None assert overview.num_recall_memory is not None assert overview.num_tokens_external_memory_summary is not None + assert overview.external_memory_summary is not None assert overview.num_tokens_system is not None assert overview.system_prompt is not None assert overview.num_tokens_core_memory is not None