1234567891011121314151617181920212223 |
- {%- if messages[0]['role'] == 'system' -%}
- {%- set system_message = messages[0]['content'] -%}
- {%- set messages = messages[1:] -%}
- {%- else -%}
- {% set system_message = '' -%}
- {%- endif -%}
- {{ bos_token + system_message }}
- {%- for message in messages -%}
- {%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
- {{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}
- {%- endif -%}
- {%- if message['role'] == 'user' -%}
- {{ 'USER: ' + message['content'] + '\n' }}
- {%- elif message['role'] == 'assistant' -%}
- {{ 'ASSISTANT: ' + message['content'] + eos_token + '\n' }}
- {%- endif -%}
- {%- endfor -%}
- {%- if add_generation_prompt -%}
- {{ 'ASSISTANT:' }}
- {% endif %}
|