Update chat_template.jinja
Browse files- chat_template.jinja +14 -97
chat_template.jinja
CHANGED
|
@@ -1,99 +1,16 @@
|
|
| 1 |
-
{%-
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
{%- else %}
|
| 13 |
-
{%- if messages[0].role == 'system' %}
|
| 14 |
-
{{- '<|im_start|>system\n' + messages[0].content + '<|im_end|>\n' }}
|
| 15 |
-
{%- endif %}
|
| 16 |
-
{%- endif %}
|
| 17 |
-
{%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
|
| 18 |
-
{%- for forward_message in messages %}
|
| 19 |
-
{%- set index = (messages|length - 1) - loop.index0 %}
|
| 20 |
-
{%- set message = messages[index] %}
|
| 21 |
-
{%- set current_content = message.content if message.content is defined and message.content is not none else '' %}
|
| 22 |
-
{%- set tool_start = '<tool_response>' %}
|
| 23 |
-
{%- set tool_start_length = tool_start|length %}
|
| 24 |
-
{%- set start_of_message = current_content[:tool_start_length] %}
|
| 25 |
-
{%- set tool_end = '</tool_response>' %}
|
| 26 |
-
{%- set tool_end_length = tool_end|length %}
|
| 27 |
-
{%- set start_pos = (current_content|length) - tool_end_length %}
|
| 28 |
-
{%- if start_pos < 0 %}
|
| 29 |
-
{%- set start_pos = 0 %}
|
| 30 |
-
{%- endif %}
|
| 31 |
-
{%- set end_of_message = current_content[start_pos:] %}
|
| 32 |
-
{%- if ns.multi_step_tool and message.role == "user" and not(start_of_message == tool_start and end_of_message == tool_end) %}
|
| 33 |
-
{%- set ns.multi_step_tool = false %}
|
| 34 |
-
{%- set ns.last_query_index = index %}
|
| 35 |
-
{%- endif %}
|
| 36 |
-
{%- endfor %}
|
| 37 |
-
{%- for message in messages %}
|
| 38 |
-
{%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
|
| 39 |
-
{{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
|
| 40 |
-
{%- elif message.role == "assistant" %}
|
| 41 |
-
{%- set m_content = message.content if message.content is defined and message.content is not none else '' %}
|
| 42 |
-
{%- set content = m_content %}
|
| 43 |
-
{%- set reasoning_content = '' %}
|
| 44 |
-
{%- if message.reasoning_content is defined and message.reasoning_content is not none %}
|
| 45 |
-
{%- set reasoning_content = message.reasoning_content %}
|
| 46 |
-
{%- else %}
|
| 47 |
-
{%- if '</think>' in m_content %}
|
| 48 |
-
{%- set content = (m_content.split('</think>')|last).lstrip('\n') %}
|
| 49 |
-
{%- set reasoning_content = (m_content.split('</think>')|first).rstrip('\n') %}
|
| 50 |
-
{%- set reasoning_content = (reasoning_content.split('<think>')|last).lstrip('\n') %}
|
| 51 |
-
{%- endif %}
|
| 52 |
-
{%- endif %}
|
| 53 |
-
{%- if loop.index0 > ns.last_query_index %}
|
| 54 |
-
{%- if loop.last or (not loop.last and (not reasoning_content.strip() == '')) %}
|
| 55 |
-
{{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content.strip('\n') + '\n</think>\n\n' + content.lstrip('\n') }}
|
| 56 |
-
{%- else %}
|
| 57 |
-
{{- '<|im_start|>' + message.role + '\n' + content }}
|
| 58 |
-
{%- endif %}
|
| 59 |
-
{%- else %}
|
| 60 |
-
{{- '<|im_start|>' + message.role + '\n' + content }}
|
| 61 |
-
{%- endif %}
|
| 62 |
-
{%- if message.tool_calls %}
|
| 63 |
-
{%- for tool_call in message.tool_calls %}
|
| 64 |
-
{%- if (loop.first and content) or (not loop.first) %}
|
| 65 |
-
{{- '\n' }}
|
| 66 |
-
{%- endif %}
|
| 67 |
-
{%- if tool_call.function %}
|
| 68 |
-
{%- set tool_call = tool_call.function %}
|
| 69 |
-
{%- endif %}
|
| 70 |
-
{{- '<tool_call>\n{"name": "' }}
|
| 71 |
-
{{- tool_call.name }}
|
| 72 |
-
{{- '", "arguments": ' }}
|
| 73 |
-
{%- if tool_call.arguments is string %}
|
| 74 |
-
{{- tool_call.arguments }}
|
| 75 |
-
{%- else %}
|
| 76 |
-
{{- tool_call.arguments | tojson }}
|
| 77 |
-
{%- endif %}
|
| 78 |
-
{{- '}\n</tool_call>' }}
|
| 79 |
-
{%- endfor %}
|
| 80 |
-
{%- endif %}
|
| 81 |
-
{{- '<|im_end|>\n' }}
|
| 82 |
-
{%- elif message.role == "tool" %}
|
| 83 |
-
{%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
|
| 84 |
-
{{- '<|im_start|>user' }}
|
| 85 |
-
{%- endif %}
|
| 86 |
-
{{- '\n<tool_response>\n' }}
|
| 87 |
-
{{- message.content }}
|
| 88 |
-
{{- '\n</tool_response>' }}
|
| 89 |
-
{%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
|
| 90 |
-
{{- '<|im_end|>\n' }}
|
| 91 |
-
{%- endif %}
|
| 92 |
-
{%- endif %}
|
| 93 |
{%- endfor %}
|
| 94 |
{%- if add_generation_prompt %}
|
| 95 |
-
|
| 96 |
-
|
| 97 |
-
|
| 98 |
-
{%- endif %}
|
| 99 |
-
{%- endif %}
|
|
|
|
| 1 |
+
{%- set system_message = 'Task: Translation\n' -%}
|
| 2 |
+
{%- set loop_messages = messages[1:] if messages[0]['role'] == 'system' else messages -%}
|
| 3 |
+
{{ system_message }}
|
| 4 |
+
{%- for message in loop_messages %}
|
| 5 |
+
{%- if message['role'] == 'user' -%}
|
| 6 |
+
<|im_start|>source
|
| 7 |
+
{{ message['content'].strip() }}<|im_end|>
|
| 8 |
+
{%- elif message['role'] == 'assistant' -%}
|
| 9 |
+
<|im_start|>target
|
| 10 |
+
{{ message['content'] }}<|im_end|>
|
| 11 |
+
{%- endif %}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
{%- endfor %}
|
| 13 |
{%- if add_generation_prompt %}
|
| 14 |
+
|
| 15 |
+
<|im_start|>target
|
| 16 |
+
{%- endif -%}
|
|
|
|
|
|