From 57524bae6d3f8d9593287177fd50c813d922099c Mon Sep 17 00:00:00 2001 From: Arsenii Shatokhin Date: Sat, 27 Apr 2024 05:53:36 +0400 Subject: [PATCH] Fix demo gradio being stuck bug --- README.md | 2 +- agency_swarm/threads/thread.py | 3 +++ setup.py | 2 +- 3 files changed, 5 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index b44a2373..2aa80f9d 100644 --- a/README.md +++ b/README.md @@ -27,7 +27,7 @@ Agency Swarm started as a desire and effort of Arsenii Shatokhin (aka VRSEN) to ## Installation ```bash -pip install agency-swarm +pip install -U agency-swarm ``` ## Getting Started diff --git a/agency_swarm/threads/thread.py b/agency_swarm/threads/thread.py index 94af6ea3..2f9b0c5e 100644 --- a/agency_swarm/threads/thread.py +++ b/agency_swarm/threads/thread.py @@ -208,6 +208,9 @@ def get_completion(self, continue + if event_handler: + event_handler.on_all_streams_end() + return full_message def _create_run(self, recipient_agent, additional_instructions, event_handler, tool_choice): diff --git a/setup.py b/setup.py index cee66ed6..459f2cb9 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ setup( name='agency-swarm', - version='0.2.0', + version='0.2.1', author='VRSEN', author_email='arseny9795@gmail.com', description='An opensource agent orchestration framework built on top of the latest OpenAI Assistants API.',