From 64a1a8b14e85c68877dfbb4ae5085f56c20fdee8 Mon Sep 17 00:00:00 2001
From: Charles Packer <packercharles@gmail.com>
Date: Sun, 4 Jan 2026 12:00:24 -0800
Subject: [PATCH] feat: expose agent_id to the messages search api endpoint
 (#8252)

---
 fern/openapi.json                            | 12 ++++++++++++
 letta/schemas/message.py                     |  1 +
 letta/server/rest_api/routers/v1/messages.py |  1 +
 3 files changed, 14 insertions(+)

diff --git a/fern/openapi.json b/fern/openapi.json
index 633914df..f4daae74 100644
--- a/fern/openapi.json
+++ b/fern/openapi.json
@@ -36982,6 +36982,18 @@
             "description": "Search mode to use",
             "default": "hybrid"
           },
+          "agent_id": {
+            "anyOf": [
+              {
+                "type": "string"
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "title": "Agent Id",
+            "description": "Filter messages by agent ID"
+          },
           "limit": {
             "type": "integer",
             "maximum": 100,
diff --git a/letta/schemas/message.py b/letta/schemas/message.py
index d09360dc..2ebde224 100644
--- a/letta/schemas/message.py
+++ b/letta/schemas/message.py
@@ -2310,6 +2310,7 @@ class MessageSearchRequest(BaseModel):
 class SearchAllMessagesRequest(BaseModel):
     query: str = Field(..., description="Text query for full-text search")
     search_mode: Literal["vector", "fts", "hybrid"] = Field("hybrid", description="Search mode to use")
+    agent_id: Optional[str] = Field(None, description="Filter messages by agent ID")
     limit: int = Field(50, description="Maximum number of results to return", ge=1, le=100)
     start_date: Optional[datetime] = Field(None, description="Filter messages created after this date")
     end_date: Optional[datetime] = Field(None, description="Filter messages created on or before this date")
diff --git a/letta/server/rest_api/routers/v1/messages.py b/letta/server/rest_api/routers/v1/messages.py
index 3a72ed8f..e7e1a6ec 100644
--- a/letta/server/rest_api/routers/v1/messages.py
+++ b/letta/server/rest_api/routers/v1/messages.py
@@ -73,6 +73,7 @@ async def search_all_messages(
         actor=actor,
         query_text=request.query,
         search_mode=request.search_mode,
+        agent_id=request.agent_id,
         limit=request.limit,
         start_date=request.start_date,
         end_date=request.end_date,