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

2024-04-29 12:42:09 +08:00
parent 3e9dbe3e0a
commit b82a2d97ef
1 changed files with 2 additions and 0 deletions
--- a/api/core/app/apps/workflow/generate_task_pipeline.py
+++ b/api/core/app/apps/workflow/generate_task_pipeline.py
@@ -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