{% for message in messages %}
{% if message['role'] == 'system' %}
{{ message['content'] }}
{% elif message['role'] == 'user' %}
{{ '<|user|>
' + message['content'] + eos_token + '
' }}
{% elif message['role'] == 'assistant' %}
{% if '<think>' in message['content'] or '</think>' in message['content'] %}
{{ '<|assistant|>
<think>' + message['content'].split('<think>')[-1].split('</think>')[0] + '</think>
' + message['content'].split('</think>')[-1] + eos_token + '
' }}
{% else %}
{{ '<|assistant|>
' + message['content'] + eos_token + '
' }}
{% endif %}
{% endif %}
{% endfor %}
{% if add_generation_prompt %}
{{ '<|assistant|>' }}
{% endif %}