yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-12 15:31:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Flush console output after every message. (#4475)

Browse Source
This commit is contained in:
afourney 2024-12-02 16:51:50 -08:00 committed by GitHub
parent 95dfe7bf22
commit c062c510b7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/packages/autogen-agentchat/src/autogen_agentchat/task/_console.py
View File

@ -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.")