소스 검색

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

takatost 11 달 전
부모
커밋
b82a2d97ef
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  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