klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-10 02:07:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: format sql during chat

Browse Source
This commit is contained in:
ulleo
2025-07-21 15:57:59 +08:00
parent b0174989b2
commit 8dcc065b17
2 changed files with 4 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/chat/task/llm.py
+4 -2
View File
@@ -1,4 +1,5 @@
import json
import sqlparse
import logging
import traceback
import warnings
@@ -775,10 +776,11 @@ def run_task(llm_service: LLMService, in_chat: bool = True):
# sql = llm_service.check_save_sql(res=full_sql_text)
print(sql)
format_sql = sqlparse.format(sql, reindent=True)
if in_chat:
yield orjson.dumps({'content': sql, 'type': 'sql'}).decode() + '\n\n'
yield orjson.dumps({'content': format_sql, 'type': 'sql'}).decode() + '\n\n'
else:
yield f'```sql\n{sql}\n```\n\n'
yield f'```sql\n{format_sql}\n```\n\n'
# execute sql
result = llm_service.execute_sql(sql=sql)