@@ -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