22 lines
903 B
Django/Jinja
22 lines
903 B
Django/Jinja
{%- set default_system_message = "You are a scientific reasoning AI assistant." %}
|
|
{{- bos_token }}
|
|
{%- if messages[0]['role'] == 'system' %}
|
|
{%- set system_message = messages[0]['content'] %}
|
|
{%- set loop_messages = messages[1:] %}
|
|
{%- else %}
|
|
{%- set system_message = default_system_message %}
|
|
{%- set loop_messages = messages %}
|
|
{%- endif %}
|
|
{{- '[SYSTEM_PROMPT]' + system_message + '[/SYSTEM_PROMPT]' }}
|
|
|
|
{%- for message in loop_messages %}
|
|
{%- if message['role'] == 'user' %}
|
|
{{- '[INST]' + message['content'] + '[/INST]' }}
|
|
{%- elif message['role'] == 'system' %}
|
|
{{- '[SYSTEM_PROMPT]' + message['content'] + '[/SYSTEM_PROMPT]' }}
|
|
{%- elif message['role'] == 'assistant' %}
|
|
{{- message['content'] + eos_token }}
|
|
{%- else %}
|
|
{{- raise_exception("Only user, system and assistant roles are supported!") }}
|
|
{%- endif %}
|
|
{%- endfor %} |