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