From 4e08ab26e7ef39468b076a41734f645b7074dbdd Mon Sep 17 00:00:00 2001 From: cthomas Date: Wed, 28 May 2025 23:40:35 -0700 Subject: [PATCH] chore: add tracing to summarizer (#2515) --- letta/services/summarizer/summarizer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/letta/services/summarizer/summarizer.py b/letta/services/summarizer/summarizer.py index 485176b2..7024d500 100644 --- a/letta/services/summarizer/summarizer.py +++ b/letta/services/summarizer/summarizer.py @@ -10,6 +10,7 @@ from letta.schemas.enums import MessageRole from letta.schemas.letta_message_content import TextContent from letta.schemas.message import Message, MessageCreate from letta.services.summarizer.enums import SummarizationMode +from letta.tracing import trace_method logger = get_logger(__name__) @@ -36,6 +37,7 @@ class Summarizer: self.summarizer_agent = summarizer_agent # TODO: Move this to config + @trace_method def summarize(self, in_context_messages: List[Message], new_letta_messages: List[Message]) -> Tuple[List[Message], bool]: """ Summarizes or trims in_context_messages according to the chosen mode,