File size: 2,583 Bytes
56d3e79 b2e9b88 56d3e79 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | {%- if not add_generation_prompt is defined -%}
{%- set add_generation_prompt = false -%}
{%- endif -%}
{%- set ns = namespace(is_first=false, is_tool=false, is_output_first=true, system_prompt="") -%}
{%- for message in messages -%}
{%- if message["role"] == "system" -%}
{%- set ns.system_prompt = message["content"] -%}
{%- endif -%}
{%- endfor -%}
{{- bos_token -}}
{{- ns.system_prompt -}}
{%- for message in messages -%}
{%- if message["role"] == "user" -%}
{%- set ns.is_tool = false -%}
{{- "<|User|>" + message["content"] -}}
{%- endif -%}
{%- if message["role"] == "assistant" and message["content"] is none -%}
{%- set ns.is_tool = false -%}
{%- for tool in message["tool_calls"] -%}
{%- if not ns.is_first -%}
{{- "<|Assistant|><|tool▁calls▁begin|><|tool▁call▁begin|>" + tool["type"] + "<|tool▁sep|>" + tool["function"]["name"] + "\n" + "```json" + "\n" + tool["function"]["arguments"] + "\n" + "```" + "<|tool▁call▁end|>" -}}
{%- set ns.is_first = true -%}
{%- else -%}
{{- "\n" + "<|tool▁call▁begin|>" + tool["type"] + "<|tool▁sep|>" + tool["function"]["name"] + "\n" + "```json" + "\n" + tool["function"]["arguments"] + "\n" + "```" + "<|tool▁call▁end|>" -}}
{{- "<|tool▁calls▁end|><|end▁of▁sentence|>" -}}
{%- endif -%}
{%- endfor -%}
{%- endif -%}
{%- if message["role"] == "assistant" and message["content"] is not none -%}
{%- if ns.is_tool -%}
{{- "<|tool▁outputs▁end|>" + 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] -%}
{%- endif -%}
{% generation %}
{{- "<|Assistant|>" + content + "<|end▁of▁sentence|>" -}}
{% endgeneration %}
{%- endif -%}
{%- endif -%}
{%- if message["role"] == "tool" -%}
{%- set ns.is_tool = true -%}
{%- if ns.is_output_first -%}
{{- "<|tool▁outputs▁begin|><|tool▁output▁begin|>" + message["content"] + "<|tool▁output▁end|>" -}}
{%- set ns.is_output_first = false -%}
{%- else -%}
{{- "\n<|tool▁output▁begin|>" + message["content"] + "<|tool▁output▁end|>" -}}
{%- endif -%}
{%- endif -%}
{%- endfor -%}
{%- if ns.is_tool -%}
{{- "<|tool▁outputs▁end|>" -}}
{%- endif -%}
{%- if add_generation_prompt and not ns.is_tool -%}
{% generation %}
{{- "<|Assistant|><think>\n" -}}
{% endgeneration %}
{%- endif -%} |