jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: handle coroutine check for on_disconnect callback in chat API (#3930)

Browse source 
Handle coroutine check for on_disconnect callback in chat API
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-09-26 10:28:17 -03:00 • committed by GitHub
commit 56d8b0fa49
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/api/v1/chat.py
View file

@ -448,7 +448,9 @@ class DisconnectHandlerStreamingResponse(StreamingResponse):
message = await receive() message = await receive()
if message["type"] == "http.disconnect": if message["type"] == "http.disconnect":
if self.on_disconnect: if self.on_disconnect:
await self.on_disconnect() coro = self.on_disconnect()
if asyncio.iscoroutine(coro):
await coro
break break