16 lines
568 B
Django/Jinja
16 lines
568 B
Django/Jinja
{%- for message in messages -%}
|
|
{%- set role = message['role'] -%}
|
|
{%- set content = message['content'] -%}
|
|
{%- if role == 'system' -%}
|
|
{{- '<|im_start|>system\n' + content.strip() + '<|im_end|>\n' -}}
|
|
{%- elif role == 'user' -%}
|
|
{{- '<|im_start|>user\n' + content.strip() + '<|im_end|>\n' -}}
|
|
{%- elif role == 'assistant' -%}
|
|
{{- '<|im_start|>assistant\n' + content.strip() + '<|im_end|>\n' -}}
|
|
{%- endif -%}
|
|
{%- endfor -%}
|
|
|
|
{%- if add_generation_prompt -%}
|
|
{{- '<|im_start|>assistant\n<think>\n' -}}
|
|
{%- endif -%}
|