16 lines
586 B
Django/Jinja
16 lines
586 B
Django/Jinja
{%- for message in messages %}
|
|
{%- if message["role"] == "system" %}
|
|
{{- "<|im_start|>system\n" + message["content"] + "<|im_end|>\n" }}
|
|
{%- elif message["role"] == "user" %}
|
|
{{- "<|im_start|>user\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 %} |