fix: add session rollback after save datasource in chat error

This commit is contained in:
ulleo
2025-09-30 11:07:15 +08:00
committed by ulleo
parent fb332a79f5
commit 2b4a6066dc
+10 -4
View File
@@ -495,10 +495,16 @@ class LLMService:
_engine_type = self.chat_question.engine
_chat.engine_type = _ds.type_name
# save chat
self.session.add(_chat)
self.session.flush()
self.session.refresh(_chat)
self.session.commit()
with self.session.begin_nested():
# 为了能继续记日志,先单独处理下事务
try:
self.session.add(_chat)
self.session.flush()
self.session.refresh(_chat)
self.session.commit()
except Exception as e:
self.session.rollback()
raise e
elif data['fail']:
raise SingleMessageError(data['fail'])