feat: improve generate sql template

This commit is contained in:
ulleo
2025-10-23 13:56:28 +08:00
committed by ulleo
parent 5ec2a682c3
commit 07b4d9ef3b
2 changed files with 5 additions and 1 deletions
+1 -1
View File
@@ -186,7 +186,7 @@ class AiModelQuestion(BaseModel):
def sql_sys_question(self, db_type: Union[str, DB], enable_query_limit: bool = True):
_sql_template = get_sql_example_template(db_type)
_base_sql_rules = _sql_template['quot_rule'] + _sql_template['limit_rule'] + _sql_template['other_rule']
_query_limit = get_sql_template()['query_limit'] if enable_query_limit else ''
_query_limit = get_sql_template()['query_limit'] if enable_query_limit else get_sql_template()['no_query_limit']
_sql_examples = _sql_template['basic_example']
_example_engine = _sql_template['example_engine']
_example_answer_1 = _sql_template['example_answer_1_with_limit'] if enable_query_limit else _sql_template[
+4
View File
@@ -11,6 +11,10 @@ template:
如果用户没有指定数据条数的限制,输出的查询SQL必须加上1000条的数据条数限制。
如果用户指定的限制大于1000,则按1000处理。
</rule>
no_query_limit: |
<rule>
如果没有指定数据条数的限制,则查询的SQL默认返回全部数据
</rule>
system: |
<Instruction>
你是"SQLBOT",智能问数小助手,可以根据用户提问,专业生成SQL与可视化图表。