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

feat: Advanced Application support to use SQL Examples

Browse Source
This commit is contained in:
ulleo
2025-11-11 14:56:13 +08:00
committed by ulleo
parent 89a93a2e52
commit 3744131627
1 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/data_training/curd/data_training.py
+16 -14
View File
@@ -308,24 +308,26 @@ def select_training_by_question(session: SessionDep, question: str, oid: int, da
_list.append(DataTraining(id=row.id, question=row.question))
if settings.EMBEDDING_ENABLED:
try:
model = EmbeddingModelCache.get_model()
with session.begin_nested():
try:
model = EmbeddingModelCache.get_model()
embedding = model.embed_query(question)
embedding = model.embed_query(question)
if advanced_application_id is not None:
results = session.execute(text(embedding_sql_in_advanced_application),
{'embedding_array': str(embedding), 'oid': oid,
'advanced_application': advanced_application_id})
else:
results = session.execute(text(embedding_sql),
{'embedding_array': str(embedding), 'oid': oid, 'datasource': datasource})
if advanced_application_id is not None:
results = session.execute(text(embedding_sql_in_advanced_application),
{'embedding_array': str(embedding), 'oid': oid,
'advanced_application': advanced_application_id})
else:
results = session.execute(text(embedding_sql),
{'embedding_array': str(embedding), 'oid': oid, 'datasource': datasource})
for row in results:
_list.append(DataTraining(id=row.id, question=row.question))
for row in results:
_list.append(DataTraining(id=row.id, question=row.question))
except Exception:
traceback.print_exc()
except Exception:
traceback.print_exc()
session.rollback()
_map: dict = {}
_ids: list[int] = []