{%- for message in messages %}
    {%- if message["role"] == "system" %}
        {{- "<|im_start|>system\n" + message["content"] + "<|im_end|>\n" }}
    {%- elif message["role"] == "user" %}
        {{- "<|im_start|>user\n" + message["content"] + "<|im_end|>\n" }}
    {%- elif message["role"] == "assistant" %}
        {{- "<|im_start|>assistant\n" }}
        {% generation %}
            {{- message["content"] }}
        {% endgeneration %}
        {{- "<|im_end|>\n" }}
    {%- endif %}
{%- endfor %}
{%- if add_generation_prompt %}
    {{- "<|im_start|>assistant\n" }}
{%- endif %}