Merge pull request #14 from j-jiseophan/fix-coroutine

fix coroutine error
2024-01-23 22:58:35 +08:00 · 2024-01-23 22:58:35 +08:00 · b24bdb070e
commit b24bdb070e
parent 33b57aea34 f64c242ee5
1 changed files with 4 additions and 1 deletions
--- a/custom_routes.py
+++ b/custom_routes.py
@ -203,7 +203,10 @@ async def send_json_override(self, event, data, sid=None):
    prompt_id = data.get('prompt_id')

    # now we send everything
-    await asyncio.wait([send(event, data), self.send_json_original(event, data, sid)])
+    await asyncio.wait([
+        asyncio.create_task(send(event, data)),
+        asyncio.create_task(self.send_json_original(event, data, sid))
+    ])

    if event == 'execution_start':
        update_run(prompt_id, Status.RUNNING)