feat: Add message listing for a letta batch (#1982)

2025-05-02 11:14:03 -07:00
parent 547ab3f613
commit d9d586e431
13 changed files with 289 additions and 48 deletions
--- a/letta/orm/message.py
+++ b/letta/orm/message.py
@@ -44,6 +44,10 @@ class Message(SqlalchemyBase, OrganizationMixin, AgentMixin):
    sender_id: Mapped[Optional[str]] = mapped_column(
        nullable=True, doc="The id of the sender of the message, can be an identity id or agent id"
    )
+    batch_item_id: Mapped[Optional[str]] = mapped_column(
+        nullable=True,
+        doc="The id of the LLMBatchItem that this message is associated with",
+    )

    # Monotonically increasing sequence for efficient/correct listing
    sequence_id: Mapped[int] = mapped_column(