diff --git a/python/packages/autogen-agentchat/src/autogen_agentchat/task/_console.py b/python/packages/autogen-agentchat/src/autogen_agentchat/task/_console.py index 9899366cd..596f9de48 100644 --- a/python/packages/autogen-agentchat/src/autogen_agentchat/task/_console.py +++ b/python/packages/autogen-agentchat/src/autogen_agentchat/task/_console.py @@ -58,6 +58,7 @@ async def Console( f"Duration: {duration:.2f} seconds\n" ) sys.stdout.write(output) + sys.stdout.flush() # mypy ignore last_processed = message # type: ignore @@ -71,6 +72,7 @@ async def Console( total_usage.completion_tokens += message.chat_message.models_usage.completion_tokens total_usage.prompt_tokens += message.chat_message.models_usage.prompt_tokens sys.stdout.write(output) + sys.stdout.flush() # Print summary. if message.inner_messages is not None: @@ -85,6 +87,7 @@ async def Console( f"Duration: {duration:.2f} seconds\n" ) sys.stdout.write(output) + sys.stdout.flush() # mypy ignore last_processed = message # type: ignore @@ -97,6 +100,7 @@ async def Console( total_usage.completion_tokens += message.models_usage.completion_tokens total_usage.prompt_tokens += message.models_usage.prompt_tokens sys.stdout.write(output) + sys.stdout.flush() if last_processed is None: raise ValueError("No TaskResult or Response was processed.")