2f4bd358f1
Signed-off-by: Naitong Yu <ntyu@baai.ac.cn> Signed-off-by: jiangxin <horizon94@outlook.com> Co-authored-by: Jason Fang <jasonfang3900@gmail.com> Co-authored-by: jiangxin <horizon94@outlook.com>
13 lines
407 B
Django/Jinja
13 lines
407 B
Django/Jinja
{%- for message in messages %}
|
|
{%- if message['role'] == 'user' %}
|
|
{{- '<_user>' + message['content']|trim }}
|
|
{%- elif message['role'] == 'system' %}
|
|
{{- '<_system>' + message['content']|trim }}
|
|
{%- elif message['role'] == 'assistant' %}
|
|
{{- '<_bot>' + message['content'] }}
|
|
{%- endif %}
|
|
{%- endfor %}
|
|
{%- if add_generation_prompt %}
|
|
{{- '<_bot>' }}
|
|
{%- endif %}
|