feat: Advanced Application support to use SQL Examples

This commit is contained in:
ulleo
2025-11-11 15:09:41 +08:00
committed by ulleo
parent 3744131627
commit e20587da99
2 changed files with 4 additions and 5 deletions
@@ -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:
+1 -1
View File
@@ -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",