From df8000bfc3fdceab9cb8044b8f2b9a195c080ab9 Mon Sep 17 00:00:00 2001 From: ulleo Date: Tue, 29 Jul 2025 17:08:03 +0800 Subject: [PATCH 1/2] feat: create chart with new system message --- backend/apps/chat/task/llm.py | 25 +++++++------------------ 1 file changed, 7 insertions(+), 18 deletions(-) diff --git a/backend/apps/chat/task/llm.py b/backend/apps/chat/task/llm.py index cbf48e2a..42f910dc 100644 --- a/backend/apps/chat/task/llm.py +++ b/backend/apps/chat/task/llm.py @@ -145,16 +145,10 @@ class LLMService: count_limit = 0 - base_message_count_limit self.sql_message = [] - if last_sql_messages is None or len(last_sql_messages) == 0: - # add sys prompt - self.sql_message.append(SystemMessage(content=self.chat_question.sql_sys_question())) - else: + # add sys prompt + self.sql_message.append(SystemMessage(content=self.chat_question.sql_sys_question())) + if last_sql_messages is not None and len(last_sql_messages) > 0: # limit count - for last_sql_message in last_sql_messages: - if last_sql_message['type'] == 'system': - _msg = SystemMessage(content=last_sql_message['content']) - self.sql_message.append(_msg) - break for last_sql_message in last_sql_messages[count_limit:]: _msg: BaseMessage if last_sql_message['type'] == 'human': @@ -167,16 +161,11 @@ class LLMService: last_chart_messages: List[dict[str, Any]] = orjson.loads(last_chart_message_str) self.chart_message = [] - if last_chart_messages is None or len(last_chart_messages) == 0: - # add sys prompt - self.chart_message.append(SystemMessage(content=self.chat_question.chart_sys_question())) - else: + # add sys prompt + self.chart_message.append(SystemMessage(content=self.chat_question.chart_sys_question())) + + if last_chart_messages is not None and len(last_chart_messages) > 0: # limit count - for last_chart_message in last_chart_messages: - if last_chart_message['type'] == 'system': - _msg = SystemMessage(content=last_chart_message['content']) - self.chart_message.append(_msg) - break for last_chart_message in last_chart_messages: _msg: BaseMessage if last_chart_message['type'] == 'human': From 73ee2646aa1170355d15ff8e899138f035444b38 Mon Sep 17 00:00:00 2001 From: fit2cloud-chenyw Date: Tue, 29 Jul 2025 17:14:32 +0800 Subject: [PATCH 2/2] perf: upgrade dependency --- backend/pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/pyproject.toml b/backend/pyproject.toml index ea6d490a..4ae65a4b 100644 --- a/backend/pyproject.toml +++ b/backend/pyproject.toml @@ -36,7 +36,7 @@ dependencies = [ "pyyaml (>=6.0.2,<7.0.0)", "fastapi-mcp (>=0.3.4,<0.4.0)", "tabulate>=0.9.0", - "sqlbot-xpack==0.0.3.9", + "sqlbot-xpack==0.0.3.10", "fastapi-cache2>=0.2.2", "sqlparse>=0.5.3", "redis>=6.2.0",