File size: 651 Bytes
4df9b6a 235a450 4df9b6a 235a450 4df9b6a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | {%- for message in messages %}
{%- set role = message['role'] %}
{%- set content = message['content'] %}
{%- if role == 'system' %}
{{ '<|im_start|>system
' + content.rstrip() + '<|im_end|>
' }}
{%- elif role == 'user' %}
{{ '<|im_start|>user
' + content.rstrip() + '<|im_end|>
' }}
{%- elif role == 'assistant' %}
{{ '<|im_start|>assistant
' + content.rstrip() }}
{%- if not content.rstrip().endswith('<|im_end|>') %}
{{ '<|im_end|>' }}
{%- endif %}
{{ '
' }}
{%- elif role == 'tool' %}
{{ '<|im_start|>tool
' + content.rstrip() + '<|im_end|>
' }}
{%- endif %}
{%- endfor %}
{%- if add_generation_prompt %}
{{ '<|im_start|>assistant
' }}
{%- endif %} |