[data] fix qwen tool template (#6796)

* Update tool_utils.py * fix unittest --------- Co-authored-by: hoshi-hiyouga <[email protected]>
hiyouga · Feb 4, 2025 · dd6b7d2 · dd6b7d2
1 parent ab9bd06
commit dd6b7d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/llamafactory/data/tool_utils.py b/src/llamafactory/data/tool_utils.py
@@ -56,7 +56,7 @@ class FunctionCall(NamedTuple):
     "You are provided with function signatures within <tools></tools> XML tags:\n<tools>{tool_text}"
     "\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"""
+    """"arguments": <args-json-object>}}\n</tool_call>"""
 )
 
 

diff --git a/tests/data/test_formatter.py b/tests/data/test_formatter.py
@@ -224,7 +224,7 @@ def test_qwen_tool_formatter():
         f"\n{json.dumps(wrapped_tool, ensure_ascii=False)}"
         "\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"""
+        """"arguments": <args-json-object>}\n</tool_call>"""
     ]