feat: json parse

Author: XiaJunjie2020

backend/apps/chat/api/chat.py

@@ -1,4 +1,5 @@
 import json
+import orjson
 import traceback
 from typing import List
 
@@ -138,7 +139,7 @@ def run_task():
             # execute sql
             result = llm_service.execute_sql(sql=sql)
             llm_service.save_sql_data(session=session, data_obj=result)
-            yield json.dumps({'content': json.dumps(result, ensure_ascii=False), 'type': 'sql-data'}) + '\n\n'
+            yield json.dumps({'content': orjson.dumps(result).decode(), 'type': 'sql-data'}) + '\n\n'
 
             # generate chart
             chart_res = llm_service.generate_chart(session=session)

backend/apps/chat/task/llm.py

@@ -1,4 +1,5 @@
 import json
+import orjson
 import logging
 import warnings
 from typing import Any, List, Union, Dict
@@ -218,7 +219,7 @@ def save_error(self, session: SessionDep, message: str):
 
     def save_sql_data(self, session: SessionDep, data_obj: Dict[str, Any]):
         return save_sql_exec_data(session=session, record_id=self.record.id,
-                                  data=json.dumps(data_obj, ensure_ascii=False))
+                                  data=orjson.dumps(data_obj).decode())
 
     def finish(self, session: SessionDep):
         return finish_record(session=session, record_id=self.record.id)