{%- for message in messages -%}
    {%- if message.role == "assistant" -%}
        {% generation %}
        {{ "<|im_start|>" + message.role + "\n" + message.content.strip() + "<|im_end|>" }}
        {% endgeneration %}
    {%- else -%}
        {{ "<|im_start|>" + message.role + "\n" + message.content.strip() + "<|im_end|>" }}
    {%- endif -%}
{%- endfor -%}
{%- if add_generation_prompt -%}
{{ "<|im_start|>assistant\n" }}
{%- endif -%}