{%- for message in messages -%}
    {%- set role = message['role'] -%}
    {%- set content = message['content'] -%}
    {%- if role == 'system' -%}
        {{- '<|im_start|>system\n' + content.strip() + '<|im_end|>\n' -}}
    {%- elif role == 'user' -%}
        {{- '<|im_start|>user\n' + content.strip() + '<|im_end|>\n' -}}
    {%- elif role == 'assistant' -%}
        {{- '<|im_start|>assistant\n' + content.strip() + '<|im_end|>\n' -}}
    {%- endif -%}
{%- endfor -%}

{%- if add_generation_prompt -%}
    {{- '<|im_start|>assistant\n<think>\n' -}}
{%- endif -%}