Print message types in Console (#6318)

Resolves #6270
This commit is contained in:
Eric Zhu 2025-04-16 20:07:43 -07:00 committed by GitHub
parent bb792b0734
commit 165c189f0e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -173,7 +173,9 @@ async def Console(
message = cast(BaseAgentEvent | BaseChatMessage, message) # type: ignore message = cast(BaseAgentEvent | BaseChatMessage, message) # type: ignore
if not streaming_chunks: if not streaming_chunks:
# Print message sender. # 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): if isinstance(message, ModelClientStreamingChunkEvent):
await aprint(message.to_text(), end="") await aprint(message.to_text(), end="")
streaming_chunks.append(message.content) streaming_chunks.append(message.content)