File size: 4,919 Bytes
a3cb486 |
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 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 |
{%- 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='', is_first_sp=true, is_first_user=true, is_last_user=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 + '
' + message['content'] %}
{%- endif %}
{%- endif %}
{%- endfor %}
{{- bos_token }}
{{- ns.system_prompt }}
{%- if tools %}
{%- if ns.system_prompt != '' %}
{{- '
# Tools
You may call one or more functions to assist with the user query.' }}
{%- else %}
{{- '# Tools
You may call one or more functions to assist with the user query.' }}
{%- endif %}
{{- '
You are provided with function signatures within <tools></tools> XML tags:' }}
{{- '
<tools>
' }}
{%- for tool in tools %}
{%- if loop.index0 > 0 %}
{{- '
' }}
{%- endif %}
{{- tool | tojson }}
{%- endfor %}
{{- '
</tools>
' }}
{{- 'For function call returns, you should first print <tool_calls>' }}
{{- 'For each function call, you should return object like:
' }}
{{- '<tool_call>function_name
```json
function_arguments_in_json_format
```</tool_call>' }}
{{- 'At the end of function call returns, you should print </tool_calls>' }}
{%- endif %}
{%- if ns.system_prompt != '' or tools %}
{{- '<|extra_4|>' }}
{%- endif %}
{%- for message in messages %}
{%- if message['role'] == 'user' %}
{%- set ns.is_tool = false %}
{%- set ns.is_first = false %}
{%- set ns.is_last_user = true %}
{%- if ns.is_first_user %}
{{- message['content'] + '<|extra_0|>' }}
{%- set ns.is_first_user = false %}
{%- else %}
{{- bos_token + message['content'] + '<|extra_0|>' }}
{%- endif %}
{%- endif %}
{%- if message['role'] == 'assistant' and message['tool_calls'] is defined and message['tool_calls'] is not none %}
{%- set ns.is_last_user = false %}
{%- if ns.is_tool %}
{{- '</tool_responses>' + '<|extra_0|>' }}
{%- endif %}
{%- set ns.is_first = false %}
{%- set ns.is_tool = false %}
{%- set ns.is_output_first = true %}
{%- for tool in message['tool_calls'] %}
{%- set arguments = tool['function']['arguments'] %}
{%- if arguments is not string %}
{%- set arguments = arguments | tojson %}
{%- endif %}
{%- if not ns.is_first %}
{%- if message['content'] is none %}
{{- '<tool_calls><tool_call>' + tool['function']['name'] + '
' + '```json' + '
' + arguments + '
' + '```' + '</tool_call>' }}
{%- else %}
{{- message['content'] + '<tool_calls><tool_call>' + tool['function']['name'] + '
' + '```json' + '
' + arguments + '
' + '```' + '</tool_call>' }}
{%- endif %}
{%- set ns.is_first = true %}
{%- else %}
{{- '
' + '<tool_call>' + tool['function']['name'] + '
' + '```json' + '
' + arguments + '
' + '```' + '</tool_call>' }}
{%- endif %}
{%- endfor %}
{{- '</tool_calls>' + eos_token }}
{%- endif %}
{%- if message['role'] == 'assistant' and (message['tool_calls'] is not defined or message['tool_calls'] is none) %}
{%- set content = message['content'] %}
{%- if '<answer>' in content and not loop.last %}
{%- set content = content.split('<answer>')[-1].strip('</answer>').strip() %}
{%- endif %}
{%- set ns.is_last_user = false %}
{%- if ns.is_tool %}
{{- '</tool_responses>' + '<|extra_0|>' + content + eos_token }}
{%- set ns.is_tool = false %}
{%- else %}
{{- content + eos_token }}
{%- endif %}
{%- endif %}
{%- if message['role'] == 'tool' %}
{%- set ns.is_last_user = false %}
{%- set ns.is_tool = true %}
{%- if ns.is_output_first %}
{{- bos_token + '<tool_responses><tool_response>' + message['content'] + '</tool_response>' }}
{%- set ns.is_output_first = false %}
{%- else %}
{{- '
<tool_response>' + message['content'] + '</tool_response>' }}
{%- endif %}
{%- endif %}
{%- endfor %}
{%- if ns.is_tool %}
{{- '</tool_responses>' + '<|extra_0|>' }}
{%- endif %}
{%- if add_generation_prompt and not ns.is_last_user and not ns.is_tool %}
{{- '<|extra_0|>' }}
{%- endif %}
{%- if enable_thinking is defined and not enable_thinking %}
{{- '<think>
</think>
' }}
{%- endif %} |