|
|
|
|
|
{{ bos_token }} |
|
|
{%- if messages[0]['role'] == 'system' -%} |
|
|
{%- if messages[0]['content'] is string -%} |
|
|
{%- set first_user_prefix = messages[0]['content'] + ' |
|
|
|
|
|
' -%} |
|
|
{%- else -%} |
|
|
{%- set first_user_prefix = messages[0]['content'][0]['text'] + ' |
|
|
|
|
|
' -%} |
|
|
{%- endif -%} |
|
|
{%- set loop_messages = messages[1:] -%} |
|
|
{%- else -%} |
|
|
{%- set first_user_prefix = "" -%} |
|
|
{%- set loop_messages = messages -%} |
|
|
{%- endif -%} |
|
|
{%- for message in loop_messages -%} |
|
|
{%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%} |
|
|
{{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }} |
|
|
{%- endif -%} |
|
|
{%- if (message['role'] == 'assistant') -%} |
|
|
{%- set role = "model" -%} |
|
|
{%- else -%} |
|
|
{%- set role = message['role'] -%} |
|
|
{%- endif -%} |
|
|
{{ '<start_of_turn>' + role + ' |
|
|
' + (first_user_prefix if loop.first else "") }} |
|
|
{%- if message['content'] is string -%} |
|
|
{{ message['content'] | trim }} |
|
|
{%- elif message['content'] is iterable -%} |
|
|
{%- for item in message['content'] -%} |
|
|
{%- if item['type'] == 'image' -%} |
|
|
{{ '<start_of_image>' }} |
|
|
{%- elif item['type'] == 'text' -%} |
|
|
{{ item['text'] | trim }} |
|
|
{%- endif -%} |
|
|
{%- endfor -%} |
|
|
{%- elif message['content'] is defined -%} |
|
|
{{ raise_exception("Invalid content type") }} |
|
|
{%- endif -%} |
|
|
{{ '<end_of_turn> |
|
|
' }} |
|
|
{%- endfor -%} |
|
|
{%- if add_generation_prompt -%} |
|
|
{{'<start_of_turn>model |
|
|
'}} |
|
|
{%- endif -%} |
|
|
|
|
|
|
