| {%- if not add_generation_prompt is defined -%} | |
| {%- set add_generation_prompt = false -%} | |
| {%- endif -%} | |
| {%- if not thinking is defined -%} | |
| {%- set thinking = false -%} | |
| {%- endif -%} | |
| {%- set ns = namespace(is_first=false, is_tool=false, system_prompt="", is_first_sp=true, is_last_user=false, is_only_sys=false, is_prefix=false) -%} | |
| {%- for message in messages -%} | |
| {%- if message["role"] == "system" -%} | |
| {%- if ns.is_first_sp -%} | |
| {%- set ns.system_prompt = ns.system_prompt + message["content"] -%} | |
| {%- set ns.is_first_sp = false -%} | |
| {%- else -%} | |
| {%- set ns.system_prompt = ns.system_prompt + "\n\n" + message["content"] -%} | |
| {%- endif -%} | |
| {%- set ns.is_only_sys = true -%} | |
| {%- endif -%} | |
| {%- endfor -%} | |
| {{- bos_token -}} | |
| {{- ns.system_prompt -}} | |
| {%- for message in messages -%} | |
| {%- if message["role"] == "user" -%} | |
| {%- set ns.is_tool = false -%} | |
| {%- set ns.is_first = false -%} | |
| {%- set ns.is_last_user = true -%} | |
| {{- "<｜User｜>" + message["content"] -}} | |
| {%- endif -%} | |
| {%- if message["role"] == "assistant" and message["tool_calls"] is defined and message["tool_calls"] is not none -%} | |
| {%- if ns.is_last_user or ns.is_only_sys -%} | |
| {{- "<｜Assistant｜></think>" -}} | |
| {%- endif -%} | |
| {%- set ns.is_last_user = false -%} | |
| {%- set ns.is_first = false -%} | |
| {%- set ns.is_tool = false -%} | |
| {%- for tool in message["tool_calls"] -%} | |
| {%- if not ns.is_first -%} | |
| {%- if message["content"] is none -%} | |
| {{- "<｜tool▁calls▁begin｜><｜tool▁call▁begin｜>" + tool["function"]["name"] + "<｜tool▁sep｜>" + tool["function"]["arguments"] + "<｜tool▁call▁end｜>" -}} | |
| {%- else -%} | |
| {{- message["content"] + "<｜tool▁calls▁begin｜><｜tool▁call▁begin｜>" + tool["function"]["name"] + "<｜tool▁sep｜>" + tool["function"]["arguments"] + "<｜tool▁call▁end｜>" -}} | |
| {%- endif -%} | |
| {%- set ns.is_first = true -%} | |
| {%- else -%} | |
| {{- "<｜tool▁call▁begin｜>" + tool["function"]["name"] + "<｜tool▁sep｜>" + tool["function"]["arguments"] + "<｜tool▁call▁end｜>" -}} | |
| {%- endif -%} | |
| {%- endfor -%} | |
| {{- "<｜tool▁calls▁end｜><｜end▁of▁sentence｜>" -}} | |
| {%- endif -%} | |
| {%- if message["role"] == "assistant" and (message["tool_calls"] is not defined or message["tool_calls"] is none) -%} | |
| {%- if ns.is_last_user -%} | |
| {{- "<｜Assistant｜>" -}} | |
| {%- if message["prefix"] is defined and message["prefix"] and thinking -%} | |
| {{- "<think>" -}} | |
| {%- else -%} | |
| {{- "</think>" -}} | |
| {%- endif -%} | |
| {%- endif -%} | |
| {%- if message["prefix"] is defined and message["prefix"] -%} | |
| {%- set ns.is_prefix = true -%} | |
| {%- endif -%} | |
| {%- set ns.is_last_user = false -%} | |
| {%- if ns.is_tool -%} | |
| {{- message["content"] + "<｜end▁of▁sentence｜>" -}} | |
| {%- set ns.is_tool = false -%} | |
| {%- else -%} | |
| {%- set content = message["content"] -%} | |
| {%- if "</think>" in content -%} | |
| {%- set content = content.split("</think>", 1)[1] -%} | |
| {%- endif -%} | |
| {{- content + "<｜end▁of▁sentence｜>" -}} | |
| {%- endif -%} | |
| {%- endif -%} | |
| {%- if message["role"] == "tool" -%} | |
| {%- set ns.is_last_user = false -%} | |
| {%- set ns.is_tool = true -%} | |
| {{- "<｜tool▁output▁begin｜>" + message["content"] + "<｜tool▁output▁end｜>" -}} | |
| {%- endif -%} | |
| {%- if message["role"] != "system" -%} | |
| {%- set ns.is_only_sys = false -%} | |
| {%- endif -%} | |
| {%- endfor -%} | |
| {%- if add_generation_prompt and not ns.is_tool -%} | |
| {%- if ns.is_last_user or ns.is_only_sys or not ns.is_prefix -%} | |
| {{- "<｜Assistant｜>" -}} | |
| {%- if not thinking -%} | |
| {{- "</think>" -}} | |
| {%- else -%} | |
| {{- "<think>" -}} | |
| {%- endif -%} | |
| {%- endif -%} | |
| {%- endif -%} |