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