[bugfix] fix rstrip (#5519)

Author: Jintao-Huang

swift/llm/template/base.py

@@ -1058,6 +1058,7 @@ def _swift_prepare_inputs(self, inputs):
                     i += 1
                 pre_message['content'], tool_content = self.agent_template._format_tool_responses(
                     pre_content, messages[i_start:i + 1])
+                # where tool_content is a List.
                 messages[i_start:i + 1] = [{'role': 'tool', 'content': tool_content}]
                 i = i_start + 1
             elif pre_role == 'assistant' and role == 'assistant' or pre_role == 'user' and role == 'user':

swift/llm/template/template/utils.py

@@ -45,8 +45,10 @@ def _swift_prepare_inputs(self, inputs):
                 # Delete the content before '</think>' in all assistant turns except the last round.
                 if message['role'] == 'assistant' and isinstance(message['content'], str) and i != len(messages) - 1:
                     if self.with_answer:
-                        message['content'] = message['content'].split('<answer>')[-1].rstrip().rstrip(
-                            '</answer>').strip()
+                        message['content'] = message['content'].split('<answer>')[-1].rstrip()
+                        if message['content'].endswith('</answer>'):
+                            message['content'] = message['content'][:-len('</answer>')]
+                        message['content'] = message['content'].strip()
                     else:
                         message['content'] = message['content'].split('</think>')[-1].strip()