Fix subscriber close issue (#358)

airtai · Oct 10, 2024 · 0d8a0d1 · 0d8a0d1
1 parent 7a7e7a8
commit 0d8a0d1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fastagency/adapters/nats/base.py b/fastagency/adapters/nats/base.py
@@ -190,15 +190,15 @@ async def start_chat() -> None:  # type: ignore [return]
                 task = asyncio.create_task(start_chat())  # type: ignore
                 background_tasks.add(task)
 
-                def callback(t: asyncio.Task[Any]) -> None:
+                async def callback(t: asyncio.Task[Any]) -> None:
                     try:
                         background_tasks.discard(t)
-                        syncify(subscriber.close)()
+                        await subscriber.close()
                     except Exception as e:
                         logger.error(f"Error in callback: {e}")
                         logger.error(traceback.format_exc())
 
-                task.add_done_callback(callback)
+                task.add_done_callback(lambda t: asyncio.create_task(callback(t)))
 
             except Exception as e:
                 await self._send_error_msg(e, logger)