Преглед изворни кода

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

tags/1.8.1
耐小心 пре 2 месеци
родитељ
комит
d5a521eef2
No account linked to committer's email address
1 измењених фајлова са 4 додато и 3 уклоњено
  1. 4
    3
      api/core/app/task_pipeline/easy_ui_based_generate_task_pipeline.py

+ 4
- 3
api/core/app/task_pipeline/easy_ui_based_generate_task_pipeline.py Прегледај датотеку

@@ -472,9 +472,10 @@ class EasyUIBasedGenerateTaskPipeline(BasedGenerateTaskPipeline):
:param event: agent thought event
:return:
"""
agent_thought: Optional[MessageAgentThought] = (
db.session.query(MessageAgentThought).where(MessageAgentThought.id == event.agent_thought_id).first()
)
with Session(db.engine, expire_on_commit=False) as session:
agent_thought: Optional[MessageAgentThought] = (
session.query(MessageAgentThought).where(MessageAgentThought.id == event.agent_thought_id).first()
)

if agent_thought:
return AgentThoughtStreamResponse(

Loading…
Откажи
Сачувај