| {%- if not add_generation_prompt is defined -%} |
| {%- set add_generation_prompt = true -%} |
| {%- endif -%} |
| {%- if not cls_token is defined -%} |
| {%- set cls_token = "<|begin_of_sentence|>" -%} |
| {%- endif -%} |
| {%- if not sep_token is defined -%} |
| {%- set sep_token = "<|end_of_sentence|>" -%} |
| {%- endif -%} |
| {{- cls_token -}} |
| {%- for message in messages -%} |
| {%- if message["role"] == "user" -%} |
| {{- "User: " + message["content"] + " |
| " -}} |
| {%- elif message["role"] == "assistant" -%} |
| {{- "Assistant: " + message["content"] + sep_token -}} |
| {%- elif message["role"] == "system" -%} |
| {{- message["content"] + " |
| " -}} |
| {%- endif -%} |
| {%- endfor -%} |
| {%- if add_generation_prompt -%} |
| {{- "Assistant: " -}} |
| {%- endif -%} |