20 lines
554 B
Django/Jinja
20 lines
554 B
Django/Jinja
{%- if messages[0].role == 'system' %}
|
|
{{- '<|im_start|>system
|
|
' + messages[0].content + '<|im_end|>
|
|
' }}
|
|
{%- endif %}
|
|
{%- for message in messages %}
|
|
{%- if message.role == 'user' or (message.role == 'system' and not loop.first) %}
|
|
{{- '<|im_start|>' + message.role + '
|
|
' + message.content + '<|im_end|>
|
|
' }}
|
|
{%- elif message.role == 'assistant' %}
|
|
{{- '<|im_start|>assistant
|
|
' + message.content + '<|im_end|>
|
|
' }}
|
|
{%- endif %}
|
|
{%- endfor %}
|
|
{%- if add_generation_prompt %}
|
|
{{- '<|im_start|>assistant
|
|
' }}
|
|
{%- endif %} |