20 lines
564 B
Django/Jinja
20 lines
564 B
Django/Jinja
{%- for message in messages -%}
|
|
{%- set content = message.content if message.content is string else '' -%}
|
|
{%- if message.role == 'system' -%}
|
|
{{ '<|im_start|>system
|
|
' + content + '<|im_end|>
|
|
' }}
|
|
{%- elif message.role == 'user' -%}
|
|
{{ '<|im_start|>user
|
|
' + content + '<|im_end|>
|
|
' }}
|
|
{%- elif message.role == 'assistant' -%}
|
|
{% generation %}{{ '<|im_start|>assistant
|
|
' + content + '<|im_end|>'}}{% endgeneration %}
|
|
{%- endif -%}
|
|
{%- endfor -%}
|
|
{%- if add_generation_prompt -%}
|
|
{{ '<|im_start|>assistant
|
|
' }}
|
|
{%- endif -%}
|