chat_template.jinja

{# Kirim OSS Safeguard R1 10B - Chat Template #}
{# This template formats conversations for the model #}

{# Begin text marker #}
<|begin_of_text|>

{# Process each message in the conversation #}
{% for message in messages %}
    {# System message #}
    {% if message['role'] == 'system' %}
<|start_header_id|>system<|end_header_id|>

{{ message['content'] }}<|eot_id|>
    
    {# User message #}
    {% elif message['role'] == 'user' %}
<|start_header_id|>user<|end_header_id|>

{{ message['content'] }}<|eot_id|>
    
    {# Assistant message #}
    {% elif message['role'] == 'assistant' %}
<|start_header_id|>assistant<|end_header_id|>

{{ message['content'] }}<|eot_id|>
    
    {# Function/tool call message (optional) #}
    {% elif message['role'] == 'function' %}
<|start_header_id|>function<|end_header_id|>

{{ message['content'] }}<|eot_id|>
    
    {# Tool message (optional) #}
    {% elif message['role'] == 'tool' %}
<|start_header_id|>tool<|end_header_id|>

{{ message['content'] }}<|eot_id|>
    {% endif %}
{% endfor %}

{# Add generation prompt if needed #}
{% if add_generation_prompt %}
<|start_header_id|>assistant<|end_header_id|>

{% endif %}