From 1c3275b3c9ff497a76a1900fc54ca7984abec42d Mon Sep 17 00:00:00 2001
From: cthomas <caren@letta.com>
Date: Sat, 19 Jul 2025 21:59:40 -0700
Subject: [PATCH] feat: remove batch items selectin from agents model (#3422)

---
 letta/orm/agent.py                           | 2 +-
 letta/serialize_schemas/marshmallow_agent.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/letta/orm/agent.py b/letta/orm/agent.py
index ce51d729..8713506d 100644
--- a/letta/orm/agent.py
+++ b/letta/orm/agent.py
@@ -138,7 +138,7 @@ class Agent(SqlalchemyBase, OrganizationMixin, AsyncAttrs):
         viewonly=True,
         back_populates="manager_agent",
     )
-    batch_items: Mapped[List["LLMBatchItem"]] = relationship("LLMBatchItem", back_populates="agent", lazy="selectin")
+    batch_items: Mapped[List["LLMBatchItem"]] = relationship("LLMBatchItem", back_populates="agent", lazy="raise")
     file_agents: Mapped[List["FileAgent"]] = relationship(
         "FileAgent",
         back_populates="agent",
diff --git a/letta/serialize_schemas/marshmallow_agent.py b/letta/serialize_schemas/marshmallow_agent.py
index 55003db1..b1340071 100644
--- a/letta/serialize_schemas/marshmallow_agent.py
+++ b/letta/serialize_schemas/marshmallow_agent.py
@@ -135,4 +135,5 @@ class MarshmallowAgentSchema(BaseSchema):
             "identities",
             "is_deleted",
             "groups",
+            "batch_items",
         )