{% for message in messages -%}
{% if message['role'] == 'assistant' -%}
<|im_start|>assistant
{% generation %}{{ message['content'] }}{% endgeneration %}<|im_end|>
{% else -%}
<|im_start|>{{ message['role'] }}
{{ message['content'] }}<|im_end|>
{% endif -%}
{% endfor -%}
{% if add_generation_prompt -%}
<|im_start|>assistant
{% generation %}{% endgeneration %}
{% endif -%}