chat_template.jinja
16.4 KB · 351 lines · plaintext Raw
1 {%- macro format_parameters(properties, required, filter_keys=false) -%}
2 {%- set standard_keys = ['description', 'type', 'properties', 'required', 'nullable'] -%}
3 {%- set ns = namespace(found_first=false) -%}
4 {%- for key, value in properties | dictsort -%}
5 {%- set add_comma = false -%}
6 {%- if not filter_keys or key not in standard_keys -%}
7 {%- if ns.found_first %},{% endif -%}
8 {%- set ns.found_first = true -%}
9 {{ key }}:{
10 {%- if value['description'] -%}
11 description:<|"|>{{ value['description'] }}<|"|>
12 {%- set add_comma = true -%}
13 {%- endif -%}
14 {%- if value['type'] | upper == 'STRING' -%}
15 {%- if value['enum'] -%}
16 {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
17 enum:{{ format_argument(value['enum']) }}
18 {%- endif -%}
19 {%- elif value['type'] | upper == 'ARRAY' -%}
20 {%- if value['items'] is mapping and value['items'] -%}
21 {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
22 items:{
23 {%- set ns_items = namespace(found_first=false) -%}
24 {%- for item_key, item_value in value['items'] | dictsort -%}
25 {%- if item_value is not none -%}
26 {%- if ns_items.found_first %},{% endif -%}
27 {%- set ns_items.found_first = true -%}
28 {%- if item_key == 'properties' -%}
29 properties:{
30 {%- if item_value is mapping -%}
31 {{- format_parameters(item_value, value['items']['required'] | default([])) -}}
32 {%- endif -%}
33 }
34 {%- elif item_key == 'required' -%}
35 required:[
36 {%- for req_item in item_value -%}
37 <|"|>{{- req_item -}}<|"|>
38 {%- if not loop.last %},{% endif -%}
39 {%- endfor -%}
40 ]
41 {%- elif item_key == 'type' -%}
42 {%- if item_value is string -%}
43 type:{{ format_argument(item_value | upper) }}
44 {%- else -%}
45 type:{{ format_argument(item_value | map('upper') | list) }}
46 {%- endif -%}
47 {%- else -%}
48 {{ item_key }}:{{ format_argument(item_value) }}
49 {%- endif -%}
50 {%- endif -%}
51 {%- endfor -%}
52 }
53 {%- endif -%}
54 {%- endif -%}
55 {%- if value['nullable'] %}
56 {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
57 nullable:true
58 {%- endif -%}
59 {%- if value['type'] | upper == 'OBJECT' -%}
60 {%- if value['properties'] is defined and value['properties'] is mapping -%}
61 {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
62 properties:{
63 {{- format_parameters(value['properties'], value['required'] | default([])) -}}
64 }
65 {%- elif value is mapping -%}
66 {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
67 properties:{
68 {{- format_parameters(value, value['required'] | default([]), filter_keys=true) -}}
69 }
70 {%- endif -%}
71 {%- if value['required'] -%}
72 {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
73 required:[
74 {%- for item in value['required'] | default([]) -%}
75 <|"|>{{- item -}}<|"|>
76 {%- if not loop.last %},{% endif -%}
77 {%- endfor -%}
78 ]
79 {%- endif -%}
80 {%- endif -%}
81 {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
82 type:<|"|>{{ value['type'] | upper }}<|"|>}
83 {%- endif -%}
84 {%- endfor -%}
85 {%- endmacro -%}
86 {%- macro format_function_declaration(tool_data) -%}
87 declaration:{{- tool_data['function']['name'] -}}{description:<|"|>{{- tool_data['function']['description'] -}}<|"|>
88 {%- set params = tool_data['function']['parameters'] -%}
89 {%- if params -%}
90 ,parameters:{
91 {%- if params['properties'] -%}
92 properties:{ {{- format_parameters(params['properties'], params['required']) -}} },
93 {%- endif -%}
94 {%- if params['required'] -%}
95 required:[
96 {%- for item in params['required'] -%}
97 <|"|>{{- item -}}<|"|>
98 {{- ',' if not loop.last -}}
99 {%- endfor -%}
100 ],
101 {%- endif -%}
102 {%- if params['type'] -%}
103 type:<|"|>{{- params['type'] | upper -}}<|"|>}
104 {%- endif -%}
105 {%- endif -%}
106 {%- if 'response' in tool_data['function'] -%}
107 {%- set response_declaration = tool_data['function']['response'] -%}
108 ,response:{
109 {%- if response_declaration['description'] -%}
110 description:<|"|>{{- response_declaration['description'] -}}<|"|>,
111 {%- endif -%}
112 {%- if response_declaration['type'] | upper == 'OBJECT' -%}
113 type:<|"|>{{- response_declaration['type'] | upper -}}<|"|>}
114 {%- endif -%}
115 {%- endif -%}
116 }
117 {%- endmacro -%}
118 {%- macro format_argument(argument, escape_keys=True) -%}
119 {%- if argument is string -%}
120 {{- '<|"|>' + argument + '<|"|>' -}}
121 {%- elif argument is boolean -%}
122 {{- 'true' if argument else 'false' -}}
123 {%- elif argument is mapping -%}
124 {{- '{' -}}
125 {%- set ns = namespace(found_first=false) -%}
126 {%- for key, value in argument | dictsort -%}
127 {%- if ns.found_first %},{% endif -%}
128 {%- set ns.found_first = true -%}
129 {%- if escape_keys -%}
130 {{- '<|"|>' + key + '<|"|>' -}}
131 {%- else -%}
132 {{- key -}}
133 {%- endif -%}
134 :{{- format_argument(value, escape_keys=escape_keys) -}}
135 {%- endfor -%}
136 {{- '}' -}}
137 {%- elif argument is sequence -%}
138 {{- '[' -}}
139 {%- for item in argument -%}
140 {{- format_argument(item, escape_keys=escape_keys) -}}
141 {%- if not loop.last %},{% endif -%}
142 {%- endfor -%}
143 {{- ']' -}}
144 {%- else -%}
145 {{- argument -}}
146 {%- endif -%}
147 {%- endmacro -%}
148 {%- macro strip_thinking(text) -%}
149 {%- set ns = namespace(result='') -%}
150 {%- for part in text.split('<channel|>') -%}
151 {%- if '<|channel>' in part -%}
152 {%- set ns.result = ns.result + part.split('<|channel>')[0] -%}
153 {%- else -%}
154 {%- set ns.result = ns.result + part -%}
155 {%- endif -%}
156 {%- endfor -%}
157 {{- ns.result | trim -}}
158 {%- endmacro -%}
159
160 {%- macro format_tool_response_block(tool_name, response) -%}
161 {{- '<|tool_response>' -}}
162 {%- if response is mapping -%}
163 {{- 'response:' + tool_name + '{' -}}
164 {%- for key, value in response | dictsort -%}
165 {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
166 {%- if not loop.last %},{% endif -%}
167 {%- endfor -%}
168 {{- '}' -}}
169 {%- else -%}
170 {{- 'response:' + tool_name + '{value:' + format_argument(response, escape_keys=False) + '}' -}}
171 {%- endif -%}
172 {{- '<tool_response|>' -}}
173 {%- endmacro -%}
174
175 {%- set ns = namespace(prev_message_type=None) -%}
176 {%- set loop_messages = messages -%}
177 {{- bos_token -}}
178 {#- Handle System/Tool Definitions Block -#}
179 {%- if (enable_thinking is defined and enable_thinking) or tools or messages[0]['role'] in ['system', 'developer'] -%}
180 {{- '<|turn>system\n' -}}
181 {#- Inject Thinking token at the very top of the FIRST system turn -#}
182 {%- if enable_thinking is defined and enable_thinking -%}
183 {{- '<|think|>\n' -}}
184 {%- set ns.prev_message_type = 'think' -%}
185 {%- endif -%}
186 {%- if messages[0]['role'] in ['system', 'developer'] -%}
187 {%- if messages[0]['content'] is string -%}
188 {{- messages[0]['content'] | trim -}}
189 {%- elif messages[0]['content'] is sequence -%}
190 {%- for item in messages[0]['content'] -%}
191 {{- item['text'] | trim + ' '-}}
192 {%- endfor -%}
193 {%- endif -%}
194 {%- set loop_messages = messages[1:] -%}
195 {%- endif -%}
196 {%- if tools -%}
197 {%- for tool in tools %}
198 {{- '<|tool>' -}}
199 {{- format_function_declaration(tool) | trim -}}
200 {{- '<tool|>' -}}
201 {%- endfor %}
202 {%- set ns.prev_message_type = 'tool' -%}
203 {%- endif -%}
204 {{- '<turn|>\n' -}}
205 {%- endif %}
206
207 {#- Pre-scan: find last user message index for reasoning guard -#}
208 {%- set ns_turn = namespace(last_user_idx=-1) -%}
209 {%- for i in range(loop_messages | length) -%}
210 {%- if loop_messages[i]['role'] == 'user' -%}
211 {%- set ns_turn.last_user_idx = i -%}
212 {%- endif -%}
213 {%- endfor -%}
214
215 {#- Loop through messages -#}
216 {%- for message in loop_messages -%}
217 {%- if message['role'] != 'tool' -%}
218 {%- set ns.prev_message_type = None -%}
219 {%- set role = 'model' if message['role'] == 'assistant' else message['role'] -%}
220 {#- Detect continuation: suppress duplicate <|turn>model when previous non-tool message was also assistant -#}
221 {%- set prev_nt = namespace(role=None, found=false) -%}
222 {%- if loop.index0 > 0 -%}
223 {%- for j in range(loop.index0 - 1, -1, -1) -%}
224 {%- if not prev_nt.found -%}
225 {%- if loop_messages[j]['role'] != 'tool' -%}
226 {%- set prev_nt.role = loop_messages[j]['role'] -%}
227 {%- set prev_nt.found = true -%}
228 {%- endif -%}
229 {%- endif -%}
230 {%- endfor -%}
231 {%- endif -%}
232 {%- set continue_same_model_turn = (role == 'model' and prev_nt.role == 'assistant') -%}
233 {%- if not continue_same_model_turn -%}
234 {{- '<|turn>' + role + '\n' }}
235 {%- endif -%}
236
237 {#- Render reasoning/reasoning_content as thinking channel -#}
238 {%- set thinking_text = message.get('reasoning') or message.get('reasoning_content') -%}
239 {%- if thinking_text and loop.index0 > ns_turn.last_user_idx and message.get('tool_calls') -%}
240 {{- '<|channel>thought\n' + thinking_text + '\n<channel|>' -}}
241 {%- endif -%}
242
243 {%- if message['tool_calls'] -%}
244 {%- for tool_call in message['tool_calls'] -%}
245 {%- set function = tool_call['function'] -%}
246 {{- '<|tool_call>call:' + function['name'] + '{' -}}
247 {%- if function['arguments'] is mapping -%}
248 {%- set ns_args = namespace(found_first=false) -%}
249 {%- for key, value in function['arguments'] | dictsort -%}
250 {%- if ns_args.found_first %},{% endif -%}
251 {%- set ns_args.found_first = true -%}
252 {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
253 {%- endfor -%}
254 {%- elif function['arguments'] is string -%}
255 {{- function['arguments'] -}}
256 {%- endif -%}
257 {{- '}<tool_call|>' -}}
258 {%- endfor -%}
259 {%- set ns.prev_message_type = 'tool_call' -%}
260 {%- endif -%}
261
262 {%- set ns_tr_out = namespace(flag=false) -%}
263 {%- if message.get('tool_responses') -%}
264 {#- Legacy: tool_responses embedded on the assistant message (Google/Gemma native) -#}
265 {%- for tool_response in message['tool_responses'] -%}
266 {{- format_tool_response_block(tool_response['name'] | default('unknown'), tool_response['response']) -}}
267 {%- set ns_tr_out.flag = true -%}
268 {%- set ns.prev_message_type = 'tool_response' -%}
269 {%- endfor -%}
270 {%- elif message.get('tool_calls') -%}
271 {#- OpenAI Chat Completions: forward-scan consecutive role:tool messages -#}
272 {%- set ns_tool_scan = namespace(stopped=false) -%}
273 {%- for k in range(loop.index0 + 1, loop_messages | length) -%}
274 {%- if ns_tool_scan.stopped -%}
275 {%- elif loop_messages[k]['role'] != 'tool' -%}
276 {%- set ns_tool_scan.stopped = true -%}
277 {%- else -%}
278 {%- set follow = loop_messages[k] -%}
279 {#- Resolve tool_call_id to function name -#}
280 {%- set ns_tname = namespace(name=follow.get('name') | default('unknown')) -%}
281 {%- for tc in message['tool_calls'] -%}
282 {%- if tc.get('id') == follow.get('tool_call_id') -%}
283 {%- set ns_tname.name = tc['function']['name'] -%}
284 {%- endif -%}
285 {%- endfor -%}
286 {#- Handle content as string or content-parts array -#}
287 {%- set tool_body = follow.get('content') -%}
288 {%- if tool_body is string -%}
289 {{- format_tool_response_block(ns_tname.name, tool_body) -}}
290 {%- elif tool_body is sequence and tool_body is not string -%}
291 {%- set ns_txt = namespace(s='') -%}
292 {%- for part in tool_body -%}
293 {%- if part.get('type') == 'text' -%}
294 {%- set ns_txt.s = ns_txt.s + (part.get('text') | default('')) -%}
295 {%- endif -%}
296 {%- endfor -%}
297 {{- format_tool_response_block(ns_tname.name, ns_txt.s) -}}
298 {%- else -%}
299 {{- format_tool_response_block(ns_tname.name, tool_body) -}}
300 {%- endif -%}
301 {%- set ns_tr_out.flag = true -%}
302 {%- set ns.prev_message_type = 'tool_response' -%}
303 {%- endif -%}
304 {%- endfor -%}
305 {%- endif -%}
306
307 {%- set captured_content -%}
308 {%- if message['content'] is string -%}
309 {%- if role == 'model' -%}
310 {{- strip_thinking(message['content']) -}}
311 {%- else -%}
312 {{- message['content'] | trim -}}
313 {%- endif -%}
314 {%- elif message['content'] is sequence -%}
315 {%- for item in message['content'] -%}
316 {%- if item['type'] == 'text' -%}
317 {%- if role == 'model' -%}
318 {{- strip_thinking(item['text']) -}}
319 {%- else -%}
320 {{- item['text'] | trim -}}
321 {%- endif -%}
322 {%- elif item['type'] == 'image' -%}
323 {{- '<|image|>' -}}
324 {%- set ns.prev_message_type = 'image' -%}
325 {%- elif item['type'] == 'audio' -%}
326 {{- '<|audio|>' -}}
327 {%- set ns.prev_message_type = 'audio' -%}
328 {%- elif item['type'] == 'video' -%}
329 {{- '<|video|>' -}}
330 {%- set ns.prev_message_type = 'video' -%}
331 {%- endif -%}
332 {%- endfor -%}
333 {%- endif -%}
334 {%- endset -%}
335
336 {{- captured_content -}}
337 {%- set has_content = captured_content | trim | length > 0 -%}
338
339 {%- if ns.prev_message_type == 'tool_call' and not ns_tr_out.flag -%}
340 {{- '<|tool_response>' -}}
341 {%- elif not (ns_tr_out.flag and not has_content) -%}
342 {{- '<turn|>\n' -}}
343 {%- endif -%}
344 {%- endif -%}
345 {%- endfor -%}
346
347 {%- if add_generation_prompt -%}
348 {%- if ns.prev_message_type != 'tool_response' and ns.prev_message_type != 'tool_call' -%}
349 {{- '<|turn>model\n' -}}
350 {%- endif -%}
351 {%- endif -%}