Skip to content

Commit df8000b

Browse files
committed
feat: create chart with new system message
1 parent ab2a1d1 commit df8000b

File tree

1 file changed

+7
-18
lines changed

1 file changed

+7
-18
lines changed

backend/apps/chat/task/llm.py

Lines changed: 7 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -145,16 +145,10 @@ def init_messages(self):
145145
count_limit = 0 - base_message_count_limit
146146

147147
self.sql_message = []
148-
if last_sql_messages is None or len(last_sql_messages) == 0:
149-
# add sys prompt
150-
self.sql_message.append(SystemMessage(content=self.chat_question.sql_sys_question()))
151-
else:
148+
# add sys prompt
149+
self.sql_message.append(SystemMessage(content=self.chat_question.sql_sys_question()))
150+
if last_sql_messages is not None and len(last_sql_messages) > 0:
152151
# limit count
153-
for last_sql_message in last_sql_messages:
154-
if last_sql_message['type'] == 'system':
155-
_msg = SystemMessage(content=last_sql_message['content'])
156-
self.sql_message.append(_msg)
157-
break
158152
for last_sql_message in last_sql_messages[count_limit:]:
159153
_msg: BaseMessage
160154
if last_sql_message['type'] == 'human':
@@ -167,16 +161,11 @@ def init_messages(self):
167161
last_chart_messages: List[dict[str, Any]] = orjson.loads(last_chart_message_str)
168162

169163
self.chart_message = []
170-
if last_chart_messages is None or len(last_chart_messages) == 0:
171-
# add sys prompt
172-
self.chart_message.append(SystemMessage(content=self.chat_question.chart_sys_question()))
173-
else:
164+
# add sys prompt
165+
self.chart_message.append(SystemMessage(content=self.chat_question.chart_sys_question()))
166+
167+
if last_chart_messages is not None and len(last_chart_messages) > 0:
174168
# limit count
175-
for last_chart_message in last_chart_messages:
176-
if last_chart_message['type'] == 'system':
177-
_msg = SystemMessage(content=last_chart_message['content'])
178-
self.chart_message.append(_msg)
179-
break
180169
for last_chart_message in last_chart_messages:
181170
_msg: BaseMessage
182171
if last_chart_message['type'] == 'human':

0 commit comments

Comments
 (0)