diff --git a/backend/apps/chat/task/llm.py b/backend/apps/chat/task/llm.py index c2ea9658..b1111e70 100644 --- a/backend/apps/chat/task/llm.py +++ b/backend/apps/chat/task/llm.py @@ -29,7 +29,7 @@ save_select_datasource_answer, save_recommend_question_answer, \ get_old_questions, save_analysis_predict_record, rename_chat, get_chart_config, \ get_chat_chart_data, list_generate_sql_logs, list_generate_chart_logs, start_log, end_log, \ - get_last_execute_sql_error + get_last_execute_sql_error, format_json_data from apps.chat.models.chat_model import ChatQuestion, ChatRecord, Chat, RenameChat, ChatLog, OperationEnum, \ ChatFinishStep, AxisObj from apps.data_training.curd.data_training import get_training_template @@ -1133,7 +1133,7 @@ def run_task(self, in_chat: bool = True, stream: bool = True, # todo generate picture if chart['type'] != 'table': yield '### generated chart picture\n\n' - image_url = request_picture(self.record.chat_id, self.record.id, chart, result) + image_url = request_picture(self.record.chat_id, self.record.id, chart, format_json_data(result)) SQLBotLogUtil.info(image_url) if stream: yield f'![{chart["type"]}]({image_url})'