27 lines
1.1 KiB
Plaintext
27 lines
1.1 KiB
Plaintext
|
{{- bos_token }}
|
||
|
|
{#- This block extracts the system message, so we can slot it into the right place. #}
|
||
|
|
{%- if messages[0]['role'] == 'system' %}
|
||
|
|
{%- set system_message = messages[0]['content']|trim %}
|
||
|
|
{%- set messages = messages[1:] %}
|
||
|
|
{%- else %}
|
||
|
|
{%- set system_message = "" %}
|
||
|
|
{%- endif %}
|
||
|
|
|
||
|
|
{#- System message + builtin tools #}
|
||
|
|
{{- "<|start_header_id|>system<|end_header_id|>\n\n" }}
|
||
|
|
{{- system_message }}
|
||
|
|
{{- "<|eot_id|>" }}
|
||
|
|
|
||
|
|
{%- for message in messages %}
|
||
|
|
{%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
|
||
|
|
{%- if message.role == 'user' %}
|
||
|
|
{{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n' }}{% generation %}{{- message['content'] | trim + '<|eot_id|>' }}{% endgeneration %}
|
||
|
|
{%- else %}
|
||
|
|
{{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' }}
|
||
|
|
{%- endif %}
|
||
|
|
{%- endif %}
|
||
|
|
{%- endfor %}
|
||
|
|
{%- if add_generation_prompt %}
|
||
|
|
{{- '<|start_header_id|>user<|end_header_id|>\n\n' }}
|
||
|
|
{%- endif %}
|