From 6fd73d4dcf59573682785ea861ce242fe98fc4cd Mon Sep 17 00:00:00 2001
From: cthomas <caren@letta.com>
Date: Fri, 5 Dec 2025 17:42:02 -0800
Subject: [PATCH] chore: add more summarize tracing (#6532)

---
 letta/services/summarizer/summarizer.py                | 1 +
 letta/services/summarizer/summarizer_all.py            | 2 ++
 letta/services/summarizer/summarizer_sliding_window.py | 2 ++
 3 files changed, 5 insertions(+)

diff --git a/letta/services/summarizer/summarizer.py b/letta/services/summarizer/summarizer.py
index 659c0ec7..f82200e7 100644
--- a/letta/services/summarizer/summarizer.py
+++ b/letta/services/summarizer/summarizer.py
@@ -408,6 +408,7 @@ def simple_message_wrapper(openai_msg: dict) -> Message:
         raise ValueError(f"Unknown role: {openai_msg['role']}")
 
 
+@trace_method
 async def simple_summary(
     messages: List[Message],
     llm_config: LLMConfig,
diff --git a/letta/services/summarizer/summarizer_all.py b/letta/services/summarizer/summarizer_all.py
index 25a7ed5e..a27ae091 100644
--- a/letta/services/summarizer/summarizer_all.py
+++ b/letta/services/summarizer/summarizer_all.py
@@ -1,6 +1,7 @@
 from typing import List
 
 from letta.log import get_logger
+from letta.otel.tracing import trace_method
 from letta.schemas.llm_config import LLMConfig
 from letta.schemas.message import Message
 from letta.schemas.user import User
@@ -10,6 +11,7 @@ from letta.services.summarizer.summarizer_config import SummarizerConfig
 logger = get_logger(__name__)
 
 
+@trace_method
 async def summarize_all(
     # Required to tag LLM calls
     actor: User,
diff --git a/letta/services/summarizer/summarizer_sliding_window.py b/letta/services/summarizer/summarizer_sliding_window.py
index e1f9f350..5fd77785 100644
--- a/letta/services/summarizer/summarizer_sliding_window.py
+++ b/letta/services/summarizer/summarizer_sliding_window.py
@@ -2,6 +2,7 @@ from typing import List, Tuple
 
 from letta.helpers.message_helper import convert_message_creates_to_messages
 from letta.log import get_logger
+from letta.otel.tracing import trace_method
 from letta.schemas.agent import AgentState
 from letta.schemas.enums import MessageRole
 from letta.schemas.letta_message_content import TextContent
@@ -41,6 +42,7 @@ async def count_tokens(actor: User, llm_config: LLMConfig, messages: List[Messag
     return tokens
 
 
+@trace_method
 async def summarize_via_sliding_window(
     # Required to tag LLM calls
     actor: User,