Ba2han commited on
Commit
cca3e64
·
verified ·
1 Parent(s): 83f40db

Update chat_template.jinja

Browse files
Files changed (1) hide show
  1. chat_template.jinja +14 -97
chat_template.jinja CHANGED
@@ -1,99 +1,16 @@
1
- {%- if tools %}
2
- {{- '<|im_start|>system\n' }}
3
- {%- if messages[0].role == 'system' %}
4
- {{- messages[0].content + '\n\n' }}
5
- {%- endif %}
6
- {{- "# 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>" }}
7
- {%- for tool in tools %}
8
- {{- "\n" }}
9
- {{- tool | tojson }}
10
- {%- endfor %}
11
- {{- "\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" }}
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
- {{- '<|im_start|>assistant\n' }}
96
- {%- if enable_thinking is defined and enable_thinking is false %}
97
- {{- '<think>\n\n</think>\n\n' }}
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 -%}