From 76bfb8402d1f75366e283197eb417f7e4161a5a7 Mon Sep 17 00:00:00 2001 From: David Strand Date: Mon, 20 Jan 2025 23:04:23 +0800 Subject: [PATCH] Update autogen_groupchat.py --- autogen_groupchat.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/autogen_groupchat.py b/autogen_groupchat.py index 39ecdb1..f249208 100644 --- a/autogen_groupchat.py +++ b/autogen_groupchat.py @@ -60,12 +60,12 @@ class AutogenGroupChat: async for task_result in team.run_stream(task=task_input): if hasattr(task_result, "messages"): for message in task_result.messages: - self.chat_log.append(f"{message.source}: {message.content}") - self.last_message = f"{message.source}: {message.content}" - print(f"[Interactive Chat] {self.last_message}") - chat_output.append(f"{message.source}: {message.content}") - - asyncio.run(capture_chat()) + message_content = f"{message.source}: {message.content}" + self.chat_log.append(message_content) + if terminate_string not in message.content: + self.last_message = message.content + print(f"[Interactive Chat] {message_content}") + chat_output.append(message_content) except RuntimeError as e: print(f"[AutogenGroupChat] Runtime error: {e}")