Merge pull request #197 from bonk1t/fix/demo-gradio

Update the demo_gradio method to align with the new package version
VRSEN · Dec 8, 2024 · 77add5d · 77add5d
2 parents dbd0299 + 914b440
commit 77add5d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/agency_swarm/agency/agency.py b/agency_swarm/agency/agency.py
@@ -636,7 +636,7 @@ def on_all_streams_end(cls):
                     cls.message_output = None
                     chatbot_queue.put("[end]")
 
-            def bot(original_message, history):
+            def bot(original_message, history, dropdown):
                 nonlocal attachments
                 nonlocal message_file_names
                 nonlocal recipient_agent
@@ -757,7 +757,7 @@ def bot(original_message, history):
             dropdown.change(handle_dropdown_change, dropdown)
             file_upload.change(handle_file_upload, file_upload)
             msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-                bot, [msg, chatbot], [msg, chatbot, dropdown]
+                bot, [msg, chatbot, dropdown], [msg, chatbot, dropdown]
             )
 
             # Enable queuing for streaming intermediate outputs

diff --git a/notebooks/os_models_with_astra_assistants_api.ipynb b/notebooks/os_models_with_astra_assistants_api.ipynb
@@ -376,7 +376,7 @@
     "\n",
     "            return original_user_message, history + [[user_message, None]]\n",
     "\n",
-    "        def bot(original_message, history):\n",
+    "        def bot(original_message, history, dropdown):\n",
     "            nonlocal message_file_ids\n",
     "            nonlocal message_file_names\n",
     "            nonlocal recipient_agent\n",