15 lines
526 B
Django/Jinja
15 lines
526 B
Django/Jinja
{% for message in messages %}
|
|
{% if message['role'] == 'system' %}
|
|
{{ '<|im_start|>system\n' + message['content'] + '<|im_end|>' }}
|
|
{% elif message['role'] == 'user' %}
|
|
{{ '\n<|im_start|>user\n' + message['content'] + '<|im_end|>' }}
|
|
{% elif message['role'] == 'assistant' %}
|
|
{{ '\n<|im_start|>assistant\n' + message['content'] + '<|im_end|>' }}
|
|
{% endif %}
|
|
{% endfor %}
|
|
{% if add_generation_prompt %}
|
|
{{ '\n<|im_start|>assistant\n' }}
|
|
{% else %}
|
|
{{ '<|im_end|>' }}
|
|
{% endif %}
|