12 lines
811 B
Django/Jinja
12 lines
811 B
Django/Jinja
{% set has_system = messages|selectattr('role', 'equalto', 'system')|list|length > 0 %}{% if not has_system %}{{ '<|im_start|>system
|
|
You are Villanova, a helpful AI assistant built by Villanova.AI.<|im_end|>
|
|
' }}{% endif %}{% for message in messages %}{% if message['role'] == 'system' %}{{ '<|im_start|>system
|
|
' + message['content'] + '<|im_end|>
|
|
' }}{% elif message['role'] == 'user' %}{{ '<|im_start|>user
|
|
' + message['content'] + '<|im_end|>
|
|
' }}{% elif message['role'] == 'assistant' %}{{ '<|im_start|>assistant
|
|
' + message['content'] }}{% if not loop.last %}{{ '<|im_end|>
|
|
' }}{% else %}{{ eos_token }}{% endif %}{% elif message['role'] == 'tool' %}{{ '<|im_start|>tool
|
|
' + message['content'] + '<|im_end|>
|
|
' }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant
|
|
' }}{% endif %} |