16 lines
575 B
Django/Jinja
16 lines
575 B
Django/Jinja
{%- if messages[0].role == 'system' -%}
|
|
{{- '<|im_start|>system\n' + messages[0].content + '\n<|im_end|>\n' -}}
|
|
{%- set loop_messages = messages[1:] -%}
|
|
{%- else -%}
|
|
{%- set loop_messages = messages -%}
|
|
{%- endif -%}
|
|
{%- for message in loop_messages -%}
|
|
{%- if message.role == 'user' -%}
|
|
{{- '<|im_start|>user\n' + message.content + '\n<|im_end|>\n' -}}
|
|
{%- elif message.role == 'assistant' -%}
|
|
{{- '<|im_start|>assistant\n' + message.content + '\n<|im_end|>\n' -}}
|
|
{%- endif -%}
|
|
{%- endfor -%}
|
|
{%- if add_generation_prompt -%}
|
|
{{- '<|im_start|>assistant\n' -}}
|
|
{%- endif -%} |