mirror of
https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git
synced 2026-11-08 01:06:59 +08:00
fix: add session rollback after save datasource in chat error
This commit is contained in:
@@ -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'])
|
||||
|
||||
Reference in New Issue
Block a user