fix: Fix database connection leak in EasyUIBasedGenerateTaskPipeline (#24815)

2025-09-01 14:48:56 +08:00
parent 1b401063e8
commit d5a521eef2
1 changed files with 4 additions and 3 deletions
--- a/api/core/app/task_pipeline/easy_ui_based_generate_task_pipeline.py
+++ b/api/core/app/task_pipeline/easy_ui_based_generate_task_pipeline.py
@@ -472,8 +472,9 @@ class EasyUIBasedGenerateTaskPipeline(BasedGenerateTaskPipeline):
        :param event: agent thought event
        :return:
        """
+        with Session(db.engine, expire_on_commit=False) as session:
            agent_thought: Optional[MessageAgentThought] = (
-            db.session.query(MessageAgentThought).where(MessageAgentThought.id == event.agent_thought_id).first()
+                session.query(MessageAgentThought).where(MessageAgentThought.id == event.agent_thought_id).first()
            )

        if agent_thought: