13 lines
707 B
Django/Jinja
13 lines
707 B
Django/Jinja
{% set loop_messages = messages %}{% for message in loop_messages %}{% set header = '<|start_header_id|>' + message['role'] + '<|end_header_id|>
|
|
|
|
' %}{% if loop.index0 == 0 %}{{ bos_token }}{% endif %}{{ header }}{% if message['role'] == 'assistant' %}{% generation %}{% if message.get('reasoning_content') %}{{ '<think>
|
|
' + message['reasoning_content'] | trim + '
|
|
</think>
|
|
|
|
' + message['content'] | trim + '<|eot_id|>' }}{% else %}{{ '<think>
|
|
|
|
</think>
|
|
|
|
' + message['content'] | trim + '<|eot_id|>' }}{% endif %}{% endgeneration %}{% else %}{{ message['content'] | trim + '<|eot_id|>' }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|start_header_id|>assistant<|end_header_id|>
|
|
|
|
' }}{% endif %} |