File size: 764 Bytes
5db868d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
System: A chat between a human and an artificial intelligence assistant.The assistant gives helpful and polite answers to the human's questions.{{- '\n'}}
{%- for message in messages %}
{%- if (message['role']|lower == 'user') != (loop.index0 % 2 == 0) %}
{{- raise_exception('Roles must alternate User/Assistant/User/Assistant/...') }}
{%- endif %}
{%-if message['role']|lower == 'user' %}
{{- message['role']|capitalize + ': ' + message['content'] + '\n' }}
{%- elif message['role']|lower == 'assistant' %}
{{- message['role']|capitalize + ': ' + message['content'] + eos_token + '\n' }}
{%- else %}
{{- raise_exception('Only user and assistant roles are supported!') }}
 {%- endif %}
{%- endfor %}{%-if add_generation_prompt %}
{{- 'Assistant: '}}
{%- endif %}