From 885bb0900bb19238435b58f7e20584bec0729bb6 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Tue, 10 Dec 2024 09:32:18 -0800 Subject: [PATCH] memory retrival to print only the bytes injected --- llama_stack/apis/agents/event_logger.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/llama_stack/apis/agents/event_logger.py b/llama_stack/apis/agents/event_logger.py index 737ba385c..4c379999e 100644 --- a/llama_stack/apis/agents/event_logger.py +++ b/llama_stack/apis/agents/event_logger.py @@ -171,12 +171,14 @@ class EventLogger: and event_type == EventType.step_complete.value ): details = event.payload.step_details - content = interleaved_text_media_as_str(details.inserted_context) - content = content[:200] + "..." if len(content) > 200 else content + inserted_context = interleaved_text_media_as_str( + details.inserted_context + ) + content = f"fetched {len(inserted_context)} bytes from {details.memory_bank_ids}" yield event, LogEvent( role=step_type, - content=f"Retrieved context from banks: {details.memory_bank_ids}.\n====\n{content}\n>", + content=content, color="cyan", )