10 lines
685 B
Django/Jinja
10 lines
685 B
Django/Jinja
{% if messages[0]['role'] == 'system' %}{% set system_message = messages[0]['content'] %}{% set messages = messages[1:] %}{% else %}{% set system_message = 'You are a helpful assistant.' %}{% endif %}{{ '<|im_start|>system
|
|
' + system_message + '<|im_end|>
|
|
' }}{% for message in messages %}{% if message['role'] == 'user' %}{{ '<|im_start|>user
|
|
' + message['content'] + '<|im_end|>
|
|
' }}{% elif message['role'] == 'assistant' %}{{ '<|im_start|>assistant
|
|
' + message['content'] + '<|im_end|>
|
|
' }}{% elif message['role'] == 'tool' %}{{ '<|im_start|>tool
|
|
' + message['content'] + '<|im_end|>
|
|
' }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant
|
|
' }}{% endif %} |