From e20587da994967b44dd6537eab80686f31eb0892 Mon Sep 17 00:00:00 2001 From: ulleo Date: Tue, 11 Nov 2025 15:08:43 +0800 Subject: [PATCH] feat: Advanced Application support to use SQL Examples --- backend/apps/data_training/curd/data_training.py | 7 +++---- backend/pyproject.toml | 2 +- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/backend/apps/data_training/curd/data_training.py b/backend/apps/data_training/curd/data_training.py index 07aabe8f..011b7fdb 100644 --- a/backend/apps/data_training/curd/data_training.py +++ b/backend/apps/data_training/curd/data_training.py @@ -266,9 +266,9 @@ ORDER BY similarity DESC LIMIT {settings.EMBEDDING_DATA_TRAINING_TOP_COUNT} """ embedding_sql_in_advanced_application = f""" -SELECT id, datasource, question, similarity +SELECT id, advanced_application, question, similarity FROM -(SELECT id, datasource, question, oid, enabled, +(SELECT id, advanced_application, question, oid, enabled, ( 1 - (embedding <=> :embedding_array) ) AS similarity FROM data_training AS child ) TEMP @@ -340,8 +340,7 @@ def select_training_by_question(session: SessionDep, question: str, oid: int, da if len(_ids) == 0: return [] - t_list = session.query(DataTraining.id, DataTraining.datasource, DataTraining.question, - DataTraining.description).filter( + t_list = session.query(DataTraining.id, DataTraining.question, DataTraining.description).filter( and_(DataTraining.id.in_(_ids))).all() for row in t_list: diff --git a/backend/pyproject.toml b/backend/pyproject.toml index 7a5eedfe..30d01097 100644 --- a/backend/pyproject.toml +++ b/backend/pyproject.toml @@ -39,7 +39,7 @@ dependencies = [ "pyyaml (>=6.0.2,<7.0.0)", "fastapi-mcp (>=0.3.4,<0.4.0)", "tabulate>=0.9.0", - "sqlbot-xpack>=0.0.3.43,<1.0.0", + "sqlbot-xpack>=0.0.3.44,<1.0.0", "fastapi-cache2>=0.2.2", "sqlparse>=0.5.3", "redis>=6.2.0",