klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 01:06:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: transaction error after run assistant chat

Browse Source
This commit is contained in:
ulleo
2025-09-29 19:14:27 +08:00
committed by ulleo
parent a4f9dfc8d4
commit d213de3eda
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/terminology/curd/terminology.py
+7 -7
View File
@@ -455,12 +455,12 @@ def select_terminology_by_word(session: SessionDep, word: str, oid: int, datasou
_list.append(Terminology(id=row.id, word=row.word, pid=row.pid))
if settings.EMBEDDING_ENABLED:
try:
model = EmbeddingModelCache.get_model()
with session.begin_nested():
try:
model = EmbeddingModelCache.get_model()
embedding = model.embed_query(word)
embedding = model.embed_query(word)
with session.begin():
if datasource is not None:
results = session.execute(text(embedding_sql_with_datasource),
{'embedding_array': str(embedding), 'oid': oid,
@@ -472,9 +472,9 @@ def select_terminology_by_word(session: SessionDep, word: str, oid: int, datasou
for row in results:
_list.append(Terminology(id=row.id, word=row.word, pid=row.pid))
except Exception:
traceback.print_exc()
session.rollback()
except Exception:
traceback.print_exc()
session.rollback()
_map: dict = {}
_ids: list[int] = []