|
| 1 | +{%- if not add_generation_prompt is defined -%} |
| 2 | + {%- set add_generation_prompt = false -%} |
| 3 | +{%- endif -%} |
| 4 | +{%- if not thinking is defined -%} |
| 5 | + {%- if enable_thinking is defined -%} |
| 6 | + {%- set thinking = enable_thinking -%} |
| 7 | + {%- else -%} |
| 8 | + {%- set thinking = false -%} |
| 9 | + {%- endif -%} |
| 10 | +{%- endif -%} |
| 11 | +{%- if not drop_thinking is defined -%} |
| 12 | + {%- set drop_thinking = true -%} |
| 13 | +{%- endif -%} |
| 14 | +{%- set dsml_token = '｜DSML｜' -%} |
| 15 | +{%- set thinking_start_token = '<think>' -%} |
| 16 | +{%- set thinking_end_token = '</think>' -%} |
| 17 | +{%- set reasoning_effort_high = 'Reasoning Effort: Absolute maximum with no shortcuts permitted.\nYou MUST be very thorough in your thinking and comprehensively decompose the problem to resolve the root cause, rigorously stress-testing your logic against all potential paths, edge cases, and adversarial scenarios.\nExplicitly write out your entire deliberation process, documenting every intermediate step, considered alternative, and rejected hypothesis to ensure absolutely no assumption is left unchecked.\n\n' -%} |
| 18 | +{%- set reasoning_effort_max = 'Reasoning Effort: Beyond maximum — exhaustive, relentless, and uncompromising.\nYou MUST reason with the utmost depth and rigor, leaving absolutely nothing to chance: exhaustively decompose the problem into its most fundamental components, trace every causal chain to its root, and resolve the underlying cause rather than any surface symptom.\nDo not stop reasoning until you have independently verified the solution from multiple angles and are certain that no assumption remains unchecked and no error remains undiscovered.\n\n' -%} |
| 19 | +{%- set response_format_template = '## Response Format:\n\nYou MUST strictly adhere to the following schema to reply:\n' -%} |
| 20 | +{%- set has_tools = false -%} |
| 21 | +{%- set tools_header = '## Tools\n\nYou have access to a set of tools to help answer the user\'s question. You can invoke tools by writing a "<' + dsml_token + 'tool_calls>" block like the following:\n\n<' + dsml_token + 'tool_calls>\n<' + dsml_token + 'invoke name="$TOOL_NAME">\n<' + dsml_token + 'parameter name="$PARAMETER_NAME" string="true|false">$PARAMETER_VALUE</' + dsml_token + 'parameter>\n...\n</' + dsml_token + 'invoke>\n<' + dsml_token + 'invoke name="$TOOL_NAME2">\n...\n</' + dsml_token + 'invoke>\n</' + dsml_token + 'tool_calls>\n\nString parameters should be specified as is and set `string="true"`. For all other types (numbers, booleans, arrays, objects), pass the value in JSON format and set `string="false"`.\n\nIf thinking_mode is enabled (triggered by ' + thinking_start_token + '), you MUST output your complete reasoning inside ' + thinking_start_token + '...' + thinking_end_token + ' BEFORE any tool calls or final response.\n\nOtherwise, output directly after ' + thinking_end_token + ' with tool calls or final response.\n\n### Available Tool Schemas\n\n' -%} |
| 22 | +{%- set tools_footer = '\nYou MUST strictly follow the above defined tool name and parameter schemas to invoke tool calls.\n' -%} |
| 23 | +{%- set ns = namespace(system_prompt='', is_first_sp=true, has_tool_calls=false) -%} |
| 24 | +{%- for message in messages -%} |
| 25 | + {%- if message['role'] == 'system' -%} |
| 26 | + {%- if ns.is_first_sp -%} |
| 27 | + {%- set ns.system_prompt = ns.system_prompt + (message['content'] or '') -%} |
| 28 | + {%- set ns.is_first_sp = false -%} |
| 29 | + {%- else -%} |
| 30 | + {%- set ns.system_prompt = ns.system_prompt + '\n\n' + (message['content'] or '') -%} |
| 31 | + {%- endif -%} |
| 32 | + {%- endif -%} |
| 33 | +{%- endfor -%} |
| 34 | +{%- if tools is defined and tools -%} |
| 35 | + {%- set has_tools = true -%} |
| 36 | + {%- set ts = namespace(schemas='') -%} |
| 37 | + {%- for tool in tools -%} |
| 38 | + {%- if tool['type'] == 'function' -%} |
| 39 | + {%- set ts.schemas = ts.schemas + (tool['function'] | tojson) + '\n' -%} |
| 40 | + {%- endif -%} |
| 41 | + {%- endfor -%} |
| 42 | + {%- if ns.system_prompt -%} |
| 43 | + {%- set ns.system_prompt = ns.system_prompt + '\n\n' + tools_header + ts.schemas + tools_footer -%} |
| 44 | + {%- else -%} |
| 45 | + {%- set ns.system_prompt = tools_header + ts.schemas + tools_footer -%} |
| 46 | + {%- endif -%} |
| 47 | +{%- endif -%} |
| 48 | +{%- if response_format is defined -%} |
| 49 | + {%- if ns.system_prompt -%} |
| 50 | + {%- set ns.system_prompt = ns.system_prompt + '\n\n' -%} |
| 51 | + {%- endif -%} |
| 52 | + {%- set ns.system_prompt = ns.system_prompt + response_format_template + (response_format | tojson) -%} |
| 53 | +{%- endif -%} |
| 54 | +{{- bos_token -}} |
| 55 | +{%- if messages and thinking and reasoning_effort is defined and reasoning_effort == 'high' -%} |
| 56 | + {{- reasoning_effort_high -}} |
| 57 | +{%- elif messages and thinking and reasoning_effort is defined and reasoning_effort == 'max' -%} |
| 58 | + {{- reasoning_effort_max -}} |
| 59 | +{%- endif -%} |
| 60 | +{{- ns.system_prompt -}} |
| 61 | +{%- set last_user_idx = namespace(value=-1) -%} |
| 62 | +{%- for message in messages -%} |
| 63 | + {%- if message['role'] == 'user' or message['role'] == 'developer' or message['role'] == 'tool' -%} |
| 64 | + {%- set last_user_idx.value = loop.index0 -%} |
| 65 | + {%- endif -%} |
| 66 | +{%- endfor -%} |
| 67 | +{%- set state = namespace(in_user=false) -%} |
| 68 | +{%- for message in messages -%} |
| 69 | + {%- if message['role'] == 'tool' -%} |
| 70 | + {%- set ns.has_tool_calls = true -%} |
| 71 | + {%- endif -%} |
| 72 | +{%- endfor -%} |
| 73 | +{%- for message in messages -%} |
| 74 | + {%- if message['role'] == 'user' or message['role'] == 'developer' -%} |
| 75 | + {%- if state.in_user -%} |
| 76 | + {{- '\n\n' -}} |
| 77 | + {%- else -%} |
| 78 | + {{- '<｜User｜>' -}} |
| 79 | + {%- set state.in_user = true -%} |
| 80 | + {%- endif -%} |
| 81 | + {{- message['content'] or '' -}} |
| 82 | + {%- elif message['role'] == 'tool' -%} |
| 83 | + {%- if state.in_user -%} |
| 84 | + {{- '\n\n' -}} |
| 85 | + {%- else -%} |
| 86 | + {{- '<｜User｜>' -}} |
| 87 | + {%- set state.in_user = true -%} |
| 88 | + {%- endif -%} |
| 89 | + {{- '<tool_result>' + (message['content'] or '') + '</tool_result>' -}} |
| 90 | + {%- elif message['role'] == 'assistant' -%} |
| 91 | + {%- set state.in_user = false -%} |
| 92 | + {{- '<｜Assistant｜>' -}} |
| 93 | + {%- set is_after_last_user = loop.index0 > last_user_idx.value -%} |
| 94 | + {%- set keep_reasoning = thinking and ((not drop_thinking) or has_tools or is_after_last_user or ns.has_tool_calls) -%} |
| 95 | + {%- if keep_reasoning -%} |
| 96 | + {{- thinking_start_token -}} |
| 97 | + {%- if message['reasoning_content'] is defined and message['reasoning_content'] -%} |
| 98 | + {{- message['reasoning_content'] -}} |
| 99 | + {%- endif -%} |
| 100 | + {{- thinking_end_token -}} |
| 101 | + {%- else -%} |
| 102 | + {{- thinking_end_token -}} |
| 103 | + {%- endif -%} |
| 104 | + {%- if message['content'] is defined and message['content'] -%} |
| 105 | + {{- message['content'] -}} |
| 106 | + {%- endif -%} |
| 107 | + {%- if message['tool_calls'] -%} |
| 108 | + {{- '\n\n<' + dsml_token + 'tool_calls>\n' -}} |
| 109 | + {%- for tool in message['tool_calls'] -%} |
| 110 | + {%- set func = tool['function'] -%} |
| 111 | + {{- '<' + dsml_token + 'invoke name="' + func['name'] + '">\n' -}} |
| 112 | + {%- set args = func['arguments'] -%} |
| 113 | + {%- if args is string -%} |
| 114 | + {%- set args = args | from_json -%} |
| 115 | + {%- endif -%} |
| 116 | + {%- for key, val in args.items() -%} |
| 117 | + {%- if val is string -%} |
| 118 | + {{- '<' + dsml_token + 'parameter name="' + key + '" string="true">' + val + '</' + dsml_token + 'parameter>\n' -}} |
| 119 | + {%- else -%} |
| 120 | + {{- '<' + dsml_token + 'parameter name="' + key + '" string="false">' + (val | tojson) + '</' + dsml_token + 'parameter>\n' -}} |
| 121 | + {%- endif -%} |
| 122 | + {%- endfor -%} |
| 123 | + {%- if not args -%} |
| 124 | + {{- '\n' -}} |
| 125 | + {%- endif -%} |
| 126 | + {{- '</' + dsml_token + 'invoke>\n' -}} |
| 127 | + {%- endfor -%} |
| 128 | + {{- '</' + dsml_token + 'tool_calls>' -}} |
| 129 | + {%- endif -%} |
| 130 | + {{- '<｜end▁of▁sentence｜>' -}} |
| 131 | + {%- endif -%} |
| 132 | +{%- endfor -%} |
| 133 | +{%- if add_generation_prompt -%} |
| 134 | + {{- '<｜Assistant｜>' -}} |
| 135 | + {%- if thinking -%} |
| 136 | + {{- thinking_start_token -}} |
| 137 | + {%- else -%} |
| 138 | + {{- thinking_end_token -}} |
| 139 | + {%- endif -%} |
| 140 | +{%- endif -%} |
0 commit comments