Fix demo gradio being stuck bug

VRSEN · Apr 27, 2024 · 57524ba · 57524ba
1 parent 3afc5a5
commit 57524ba
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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
@@ -6,7 +6,7 @@
 
 setup(
     name='agency-swarm',
-    version='0.2.0',
+    version='0.2.1',
     author='VRSEN',
     author_email='[email protected]',
     description='An opensource agent orchestration framework built on top of the latest OpenAI Assistants API.',