From b82a2d97ef9268253e8619d448bebcb15a893e91 Mon Sep 17 00:00:00 2001
From: takatost <takatost@users.noreply.github.com>
Date: Mon, 29 Apr 2024 12:42:09 +0800
Subject: [PATCH] fix: db connections not being released during workflow
 execution (#3971)

---
 api/core/app/apps/workflow/generate_task_pipeline.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/api/core/app/apps/workflow/generate_task_pipeline.py b/api/core/app/apps/workflow/generate_task_pipeline.py
index 9ccafa447d..a7061a77bb 100644
--- 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