{%- if messages[0]['role'] == 'system' %}
    {%- set system_message = messages[0]['content'] | trim + ' ' %}
    {%- set messages = messages[1:] %}
{%- else %}
    {%- set system_message = 'A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user\'s questions. ' %}
{%- endif %}

{{- system_message }}
{%- for message in messages %}
    {%- if message['role'] == 'user' %}
        {{- 'USER: ' + message['content'] | trim }}
    {%- elif message['role'] == 'assistant' %}
        {{- ' ASSISTANT: '  + message['content'] | trim + '</s>' }}
    {%- endif %}
{%- endfor %}

{%- if add_generation_prompt %}
    {{- ' ASSISTANT:' }}
{%- endif %}