Skip to content

Commit 8dcc065

Browse files
committed
feat: format sql during chat
1 parent b017498 commit 8dcc065

File tree

2 files changed

+4
-88
lines changed

2 files changed

+4
-88
lines changed

backend/apps/chat/task/llm.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import json
2+
import sqlparse
23
import logging
34
import traceback
45
import warnings
@@ -775,10 +776,11 @@ def run_task(llm_service: LLMService, in_chat: bool = True):
775776
# sql = llm_service.check_save_sql(res=full_sql_text)
776777

777778
print(sql)
779+
format_sql = sqlparse.format(sql, reindent=True)
778780
if in_chat:
779-
yield orjson.dumps({'content': sql, 'type': 'sql'}).decode() + '\n\n'
781+
yield orjson.dumps({'content': format_sql, 'type': 'sql'}).decode() + '\n\n'
780782
else:
781-
yield f'```sql\n{sql}\n```\n\n'
783+
yield f'```sql\n{format_sql}\n```\n\n'
782784

783785
# execute sql
784786
result = llm_service.execute_sql(sql=sql)

frontend/src/views/chat/component/PredictChartBlock.vue

Lines changed: 0 additions & 86 deletions
This file was deleted.

0 commit comments

Comments
 (0)