mirror of
https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git
synced 2026-11-08 01:06:59 +08:00
feat: Advanced Application support to use SQL Examples
This commit is contained in:
@@ -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:
|
||||
|
||||
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user