File size: 1,314 Bytes
4e974e7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | {% if 'role' in messages[0] %}{{ bos_token }}{% if messages[0]['role'] == 'system' %}{{'<start_of_turn>user
' + messages[0]['content'] | trim + ' ' + messages[1]['content'] | trim + '<end_of_turn>
'}}{% set messages = messages[2:] %}{% endif %}{% for message in messages %}{% if message['role'] == 'user' %}{{'<start_of_turn>user
' + message['content'] | trim + '<end_of_turn>
'}}{% elif message['role'] == 'assistant' %}{{'<start_of_turn>model
' + message['content'] | trim + '<end_of_turn>
' }}{% else %}{{ raise_exception('Only user and assistant roles are supported!') }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<start_of_turn>model
' }}{% endif %}{% else %}{{ bos_token }}{% if messages[0]['from'] == 'system' %}{{'<start_of_turn>user
' + messages[0]['value'] | trim + ' ' + messages[1]['value'] | trim + '<end_of_turn>
'}}{% set messages = messages[2:] %}{% endif %}{% for message in messages %}{% if message['from'] == 'human' %}{{'<start_of_turn>user
' + message['value'] | trim + '<end_of_turn>
'}}{% elif message['from'] == 'gpt' %}{{'<start_of_turn>model
' + message['value'] | trim + '<end_of_turn>
' }}{% else %}{{ raise_exception('Only user and assistant roles are supported!') }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<start_of_turn>model
' }}{% endif %}{% endif %} |