youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 08:58:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

optimize db connections

Browse Source
This commit is contained in:
takatost
2024-03-10 14:49:52 +08:00
parent 3d6b06696e
commit 7693ba8797
5 changed files with 12 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/app/apps/workflow/generate_task_pipeline.py
View File

@ -99,13 +99,15 @@ class WorkflowAppGenerateTaskPipeline(WorkflowBasedGenerateTaskPipeline):
self._output_moderation_handler = self._init_output_moderation()
self._stream = stream
db.session.close()
def process(self) -> Union[dict, Generator]:
"""
Process generate task pipeline.
:return:
"""
db.session.refresh(self._workflow)
db.session.refresh(self._user)
db.session.close()
if self._stream:
return self._process_stream_response()
else: