Skip to content

Commit d213de3

Browse files
committed
fix: transaction error after run assistant chat
1 parent a4f9dfc commit d213de3

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

backend/apps/terminology/curd/terminology.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -455,12 +455,12 @@ def select_terminology_by_word(session: SessionDep, word: str, oid: int, datasou
455455
_list.append(Terminology(id=row.id, word=row.word, pid=row.pid))
456456

457457
if settings.EMBEDDING_ENABLED:
458-
try:
459-
model = EmbeddingModelCache.get_model()
458+
with session.begin_nested():
459+
try:
460+
model = EmbeddingModelCache.get_model()
460461

461-
embedding = model.embed_query(word)
462+
embedding = model.embed_query(word)
462463

463-
with session.begin():
464464
if datasource is not None:
465465
results = session.execute(text(embedding_sql_with_datasource),
466466
{'embedding_array': str(embedding), 'oid': oid,
@@ -472,9 +472,9 @@ def select_terminology_by_word(session: SessionDep, word: str, oid: int, datasou
472472
for row in results:
473473
_list.append(Terminology(id=row.id, word=row.word, pid=row.pid))
474474

475-
except Exception:
476-
traceback.print_exc()
477-
session.rollback()
475+
except Exception:
476+
traceback.print_exc()
477+
session.rollback()
478478

479479
_map: dict = {}
480480
_ids: list[int] = []

0 commit comments

Comments
 (0)