Browse Source

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

takatost 11 months ago
parent
commit
b82a2d97ef
1 changed files with 2 additions and 0 deletions
  1. 2 0
      api/core/app/apps/workflow/generate_task_pipeline.py

+ 2 - 0
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