| 1 | {%- for message in messages -%} |
| 2 | {{- '<|im_start|>' + message['role'] + '\n' + message['content'] -}} |
| 3 | {%- if (loop.last and add_generation_prompt) or not loop.last -%} |
| 4 | {{- '<|im_end|>' + '\n' -}} |
| 5 | {%- endif -%} |
| 6 | {%- endfor -%} |
| 7 | |
| 8 | {%- if add_generation_prompt and messages[-1]['role'] != 'assistant' -%} |
| 9 | {{- '<|im_start|>assistant\n' -}} |
| 10 | {%- endif -%} |