From 165c189f0e9a809af97dbae716bc497f8ed6338c Mon Sep 17 00:00:00 2001 From: Eric Zhu Date: Wed, 16 Apr 2025 20:07:43 -0700 Subject: [PATCH] Print message types in Console (#6318) Resolves #6270 --- .../autogen-agentchat/src/autogen_agentchat/ui/_console.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/packages/autogen-agentchat/src/autogen_agentchat/ui/_console.py b/python/packages/autogen-agentchat/src/autogen_agentchat/ui/_console.py index 51d98ffe2..112ddafe4 100644 --- a/python/packages/autogen-agentchat/src/autogen_agentchat/ui/_console.py +++ b/python/packages/autogen-agentchat/src/autogen_agentchat/ui/_console.py @@ -173,7 +173,9 @@ async def Console( message = cast(BaseAgentEvent | BaseChatMessage, message) # type: ignore if not streaming_chunks: # Print message sender. - await aprint(f"{'-' * 10} {message.source} {'-' * 10}", end="\n", flush=True) + await aprint( + f"{'-' * 10} {message.__class__.__name__} ({message.source}) {'-' * 10}", end="\n", flush=True + ) if isinstance(message, ModelClientStreamingChunkEvent): await aprint(message.to_text(), end="") streaming_chunks.append(message.content)