From 75d1b50f5d45a0257ca4d0dcf14bdcbb00e1d2a9 Mon Sep 17 00:00:00 2001 From: cthomas Date: Thu, 29 May 2025 16:45:29 -0700 Subject: [PATCH] feat: add size async caching (#2532) --- letta/orm/passage.py | 6 +++++- letta/orm/sqlalchemy_base.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/letta/orm/passage.py b/letta/orm/passage.py index 76790fc2..eb38b691 100644 --- a/letta/orm/passage.py +++ b/letta/orm/passage.py @@ -90,4 +90,8 @@ class AgentPassage(BasePassage, AgentMixin): Index("agent_passages_created_at_id_idx", "created_at", "id"), {"extend_existing": True}, ) - return (Index("agent_passages_created_at_id_idx", "created_at", "id"), {"extend_existing": True}) + return ( + Index("ix_agent_passages_org_agent", "organization_id", "agent_id"), + Index("agent_passages_created_at_id_idx", "created_at", "id"), + {"extend_existing": True}, + ) diff --git a/letta/orm/sqlalchemy_base.py b/letta/orm/sqlalchemy_base.py index 8154a0ea..04e27e11 100644 --- a/letta/orm/sqlalchemy_base.py +++ b/letta/orm/sqlalchemy_base.py @@ -840,7 +840,7 @@ class SqlalchemyBase(CommonSqlalchemyMetaMixins, Base): **kwargs, ): logger.debug(f"Calculating size for {cls.__name__} with filters {kwargs}") - query = select(func.count()).select_from(cls) + query = select(func.count(1)).select_from(cls) if actor: query = cls.apply_access_predicate(query, actor, access, access_type)