phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

memory retrival to print only the bytes injected

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-10 09:32:18 -08:00
parent 2e3d3a62a5
commit 885bb0900b
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/apis/agents/event_logger.py
View file

@ -171,12 +171,14 @@ class EventLogger:
and event_type == EventType.step_complete.value and event_type == EventType.step_complete.value
): ):
details = event.payload.step_details details = event.payload.step_details
content = interleaved_text_media_as_str(details.inserted_context) inserted_context = interleaved_text_media_as_str(
content = content[:200] + "..." if len(content) > 200 else content details.inserted_context
)
content = f"fetched {len(inserted_context)} bytes from {details.memory_bank_ids}"
yield event, LogEvent( yield event, LogEvent(
role=step_type, role=step_type,
content=f"Retrieved context from banks: {details.memory_bank_ids}.\n====\n{content}\n>", content=content,
color="cyan", color="cyan",
) )