aferrante/MNLP_M3_mcqa_modelv4

1 2{%- if tools %}3    {{- '<|im_start|>system4' }}5    {%- if messages[0].role == 'system' %}6        {{- messages[0].content + '7 8' }}9    {%- endif %}10    {{- "# Tools11 12You may call one or more functions to assist with the user query.13 14You are provided with function signatures within <tools></tools> XML tags:15<tools>" }}16    {%- for tool in tools %}17        {{- "18" }}19        {{- tool | tojson }}20    {%- endfor %}21    {{- "22</tools>23 24For each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:25<tool_call>26{\"name\": <function-name>, \"arguments\": <args-json-object>}27</tool_call><|im_end|>28" }}29{%- else %}30    {%- if messages[0].role == 'system' %}31        {{- '<|im_start|>system32' + messages[0].content + '<|im_end|>33' }}34    {%- endif %}35{%- endif %}36{%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}37{%- for forward_message in messages %}38    {%- set index = (messages|length - 1) - loop.index0 %}39    {%- set message = messages[index] %}40    {%- set current_content = message.content if message.content is not none else '' %}41    {%- set tool_start = '<tool_response>' %}42    {%- set tool_start_length = tool_start|length %}43    {%- set start_of_message = current_content[:tool_start_length] %}44    {%- set tool_end = '</tool_response>' %}45    {%- set tool_end_length = tool_end|length %}46    {%- set start_pos = (current_content|length) - tool_end_length %}47    {%- if start_pos < 0 %}48        {%- set start_pos = 0 %}49    {%- endif %}50    {%- set end_of_message = current_content[start_pos:] %}51    {%- if ns.multi_step_tool and message.role == "user" and not(start_of_message == tool_start and end_of_message == tool_end) %}52        {%- set ns.multi_step_tool = false %}53        {%- set ns.last_query_index = index %}54    {%- endif %}55{%- endfor %}56{%- for message in messages %}57    {%- if (message.role == "user") or (message.role == "system" and not loop.first) %}58        {{- '<|im_start|>' + message.role + '59' + message.content + '<|im_end|>' + '60' }}61    {%- elif message.role == "assistant" %}62        {%- set content = message.content %}63        {%- set reasoning_content = '' %}64        {%- if message.reasoning_content is defined and message.reasoning_content is not none %}65            {%- set reasoning_content = message.reasoning_content %}66        {%- else %}67            {%- if '</think>' in message.content %}68                {%- set content = (message.content.split('</think>')|last).lstrip('69') %}70                {%- set reasoning_content = (message.content.split('</think>')|first).rstrip('71') %}72                {%- set reasoning_content = (reasoning_content.split('<think>')|last).lstrip('73') %}74            {%- endif %}75        {%- endif %}76        {%- if loop.index0 > ns.last_query_index %}77            {%- if loop.last or (not loop.last and reasoning_content) %}78                {{- '<|im_start|>' + message.role + '79<think>80' + reasoning_content.strip('81') + '82</think>83 84' + content.lstrip('85') }}86            {%- else %}87                {{- '<|im_start|>' + message.role + '88' + content }}89            {%- endif %}90        {%- else %}91            {{- '<|im_start|>' + message.role + '92' + content }}93        {%- endif %}94        {%- if message.tool_calls %}95            {%- for tool_call in message.tool_calls %}96                {%- if (loop.first and content) or (not loop.first) %}97                    {{- '98' }}99                {%- endif %}100                {%- if tool_call.function %}101                    {%- set tool_call = tool_call.function %}102                {%- endif %}103                {{- '<tool_call>104{"name": "' }}105                {{- tool_call.name }}106                {{- '", "arguments": ' }}107                {%- if tool_call.arguments is string %}108                    {{- tool_call.arguments }}109                {%- else %}110                    {{- tool_call.arguments | tojson }}111                {%- endif %}112                {{- '}113</tool_call>' }}114            {%- endfor %}115        {%- endif %}116        {{- '<|im_end|>117' }}118    {%- elif message.role == "tool" %}119        {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}120            {{- '<|im_start|>user' }}121        {%- endif %}122        {{- '123<tool_response>124' }}125        {{- message.content }}126        {{- '127</tool_response>' }}128        {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}129            {{- '<|im_end|>130' }}131        {%- endif %}132    {%- endif %}133{%- endfor %}134{%- if add_generation_prompt %}135    {{- '<|im_start|>assistant136' }}137    {%- if enable_thinking is defined and enable_thinking is false %}138        {{- '<think>139 140</think>141 142' }}143    {%- endif %}144{%- endif %}145