Jelajahi Sumber

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

takatost 11 bulan lalu
induk
melakukan
b82a2d97ef
1 mengubah file dengan 2 tambahan dan 0 penghapusan
  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