12 lines
482 B
Django/Jinja
12 lines
482 B
Django/Jinja
{%- for message in messages %}
|
|
{%- if message.role == "user" %}
|
|
{{- '<|im_start|>user\n' + message.content + '<|im_end|>\n' }}
|
|
{%- elif message.role == "system" %}
|
|
{{- '<|im_start|>system\n' + message.content + '<|im_end|>\n' }}
|
|
{%- elif message.role == "assistant" %}
|
|
{{- '<|im_start|>assistant\n' }}{% generation %}{{ message.content }}{% endgeneration %}{{ '<|im_end|>\n' }}
|
|
{%- endif %}
|
|
{%- endfor %}
|
|
{%- if add_generation_prompt %}
|
|
{{- '<|im_start|>assistant\n' }}
|
|
{%- endif %} |