optimize db connections

This commit is contained in:
takatost
2024-03-10 13:19:17 +08:00
parent 51f6ab49cf
commit de3978fdbb
7 changed files with 40 additions and 3 deletions

View File

@ -99,6 +99,8 @@ 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.