File size: 752 Bytes
389c0ee | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | {# ───── main loop ───── #}
{%- for message in messages -%}
{%- set content = message.content if message.content is string else "" -%}
{%- if (message.role == "user") or (message.role == "system") -%}
{{ "<|im_start|>" + message.role + "\n" + content + "<|im_end|>\n" }}
{%- elif message.role == "assistant" -%}
{%- generation -%}
{{ "<|im_start|>assistant\n" + content + "<|im_end|>\n" }}
{%- endgeneration -%}
{%- elif message.role == "tool" -%}
{{ "<|im_start|>" + "user\n" + content + "<|im_end|>\n" }}
{%- endif -%}
{%- endfor -%}
{# ───── generation prompt ───── #}
{%- if add_generation_prompt -%}
{{ "<|im_start|>assistant\n" }}
{%- endif -%} |