mirror of
https://github.com/vllm-project/vllm.git
synced 2026-08-04 12:58:07 +00:00
2f4bd358f1
Signed-off-by: Naitong Yu <[email protected]> Signed-off-by: jiangxin <[email protected]> Co-authored-by: Jason Fang <[email protected]> Co-authored-by: jiangxin <[email protected]>
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 %}
|