tokenizer/chat_template.jinja
| 1 | {%- if tools %} |
| 2 | {{- '<|im_start|>system\n' }} |
| 3 | {%- if messages[0]['role'] == 'system' %} |
| 4 | {{- messages[0]['content'] }} |
| 5 | {%- else %} |
| 6 | {{- 'You are a helpful assistant.' }} |
| 7 | {%- endif %} |
| 8 | {{- "\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }} |
| 9 | {%- for tool in tools %} |
| 10 | {{- "\n" }} |
| 11 | {{- tool | tojson }} |
| 12 | {%- endfor %} |
| 13 | {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }} |
| 14 | {%- else %} |
| 15 | {%- if messages[0]['role'] == 'system' %} |
| 16 | {{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }} |
| 17 | {%- else %} |
| 18 | {{- '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n' }} |
| 19 | {%- endif %} |
| 20 | {%- endif %} |
| 21 | {%- for message in messages %} |
| 22 | {%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %} |
| 23 | {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }} |
| 24 | {%- elif message.role == "assistant" %} |
| 25 | {{- '<|im_start|>' + message.role }} |
| 26 | {%- if message.content %} |
| 27 | {{- '\n' + message.content }} |
| 28 | {%- endif %} |
| 29 | {%- for tool_call in message.tool_calls %} |
| 30 | {%- if tool_call.function is defined %} |
| 31 | {%- set tool_call = tool_call.function %} |
| 32 | {%- endif %} |
| 33 | {{- '\n<tool_call>\n{"name": "' }} |
| 34 | {{- tool_call.name }} |
| 35 | {{- '", "arguments": ' }} |
| 36 | {{- tool_call.arguments | tojson }} |
| 37 | {{- '}\n</tool_call>' }} |
| 38 | {%- endfor %} |
| 39 | {{- '<|im_end|>\n' }} |
| 40 | {%- elif message.role == "tool" %} |
| 41 | {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %} |
| 42 | {{- '<|im_start|>user' }} |
| 43 | {%- endif %} |
| 44 | {{- '\n<tool_response>\n' }} |
| 45 | {{- message.content }} |
| 46 | {{- '\n</tool_response>' }} |
| 47 | {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %} |
| 48 | {{- '<|im_end|>\n' }} |
| 49 | {%- endif %} |
| 50 | {%- endif %} |
| 51 | {%- endfor %} |
| 52 | {%- if add_generation_prompt %} |
| 53 | {{- '<|im_start|>assistant\n' }} |
| 54 | {%- endif %} |
| 55 | |