fix: db connections not being released during workflow execution (#3971)

This commit is contained in:
takatost 2024-04-29 12:42:09 +08:00 committed by GitHub
parent 3e9dbe3e0a
commit b82a2d97ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -380,6 +380,8 @@ class WorkflowAppGenerateTaskPipeline(BasedGenerateTaskPipeline, WorkflowCycleMa
self._task_state.answer += text
yield self._text_chunk_to_stream_response(text)
db.session.close()
def _is_stream_out_support(self, event: QueueTextChunkEvent) -> bool:
"""
Is stream out support