File size: 2,002 Bytes
5725aa3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
{%- for message in messages %}
{#-- Validate role is a stringified integer --#}
{%- if not message['role'] is string or not message['role'].isdigit() %}
{{- raise_exception("The role must be an integer or a stringified integer (e.g. '0') designating the speaker id") }}
{%- endif %}
{#-- Validate content is a list --#}
{%- set content = message['content'] %}
{%- if content is not iterable or content is string %}
{{- raise_exception("The content must be a list") }}
{%- endif %}
{#-- Collect content types --#}
{%- set content_types = content | map(attribute='type') | list %}
{%- set is_last = loop.last %}
{#-- Last message validation --#}
{%- if is_last %}
{%- if 'text' not in content_types %}
{{- raise_exception("The last message must include one item of type 'text'") }}
{%- elif (content_types | select('equalto', 'text') | list | length > 1) or (content_types | select('equalto', 'audio') | list | length > 1) %}
{{- raise_exception("At most two items are allowed in the last message: one 'text' and one 'audio'") }}
{%- endif %}
{#-- All other messages validation --#}
{%- else %}
{%- if content_types | select('equalto', 'text') | list | length != 1
or content_types | select('equalto', 'audio') | list | length != 1 %}
{{- raise_exception("Each message (except the last) must contain exactly one 'text' and one 'audio' item") }}
{%- elif content_types | reject('in', ['text', 'audio']) | list | length > 0 %}
{{- raise_exception("Only 'text' and 'audio' types are allowed in content") }}
{%- endif %}
{%- endif %}
{%- endfor %}
{%- for message in messages %}
{{- bos_token }}
{{- '[' + message['role'] + ']' }}
{{- message['content'][0]['text'] }}
{{- eos_token }}
{%- if message['content']|length > 1 %}
{{- '<|AUDIO|><|audio_eos|>' }}
{%- endif %}
{%- endfor %}
|