From 56d8b0fa49119a9ce42d159ca42d84923fa86fb8 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 26 Sep 2024 10:28:17 -0300 Subject: [PATCH] fix: handle coroutine check for on_disconnect callback in chat API (#3930) Handle coroutine check for on_disconnect callback in chat API --- src/backend/base/langflow/api/v1/chat.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/api/v1/chat.py b/src/backend/base/langflow/api/v1/chat.py index 00afd0d7d..1af57a061 100644 --- a/src/backend/base/langflow/api/v1/chat.py +++ b/src/backend/base/langflow/api/v1/chat.py @@ -448,7 +448,9 @@ class DisconnectHandlerStreamingResponse(StreamingResponse): message = await receive() if message["type"] == "http.disconnect": if self.on_disconnect: - await self.on_disconnect() + coro = self.on_disconnect() + if asyncio.iscoroutine(coro): + await coro break