Skip to content

Commit e20587d

Browse files
committed
feat: Advanced Application support to use SQL Examples
1 parent 3744131 commit e20587d

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

backend/apps/data_training/curd/data_training.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -266,9 +266,9 @@ def save_embeddings(session_maker, ids: List[int]):
266266
LIMIT {settings.EMBEDDING_DATA_TRAINING_TOP_COUNT}
267267
"""
268268
embedding_sql_in_advanced_application = f"""
269-
SELECT id, datasource, question, similarity
269+
SELECT id, advanced_application, question, similarity
270270
FROM
271-
(SELECT id, datasource, question, oid, enabled,
271+
(SELECT id, advanced_application, question, oid, enabled,
272272
( 1 - (embedding <=> :embedding_array) ) AS similarity
273273
FROM data_training AS child
274274
) TEMP
@@ -340,8 +340,7 @@ def select_training_by_question(session: SessionDep, question: str, oid: int, da
340340
if len(_ids) == 0:
341341
return []
342342

343-
t_list = session.query(DataTraining.id, DataTraining.datasource, DataTraining.question,
344-
DataTraining.description).filter(
343+
t_list = session.query(DataTraining.id, DataTraining.question, DataTraining.description).filter(
345344
and_(DataTraining.id.in_(_ids))).all()
346345

347346
for row in t_list:

backend/pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ dependencies = [
3939
"pyyaml (>=6.0.2,<7.0.0)",
4040
"fastapi-mcp (>=0.3.4,<0.4.0)",
4141
"tabulate>=0.9.0",
42-
"sqlbot-xpack>=0.0.3.43,<1.0.0",
42+
"sqlbot-xpack>=0.0.3.44,<1.0.0",
4343
"fastapi-cache2>=0.2.2",
4444
"sqlparse>=0.5.3",
4545
"redis>=6.2.0",

0 commit comments

Comments
 (0)