Update chat_template.jinja

chat_template.jinja

@@ -41,32 +41,29 @@
   {%- else -%}
     {{- "You are a helpful AI assistant named SmolLM, trained by Hugging Face.\n\n" -}}
   {%- endif -%}
-  
-  {{- "## Tools\n\n" -}}
-  {{- "### XML Tools\n\n" -}}
-  {%- if tools -%}
-    {%- set ns = namespace(xml_tool_string="You 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\n<tools>\n") -%}
-    {%- for tool in tools -%}
-      {%- set ns.xml_tool_string = ns.xml_tool_string ~ (tool | tojson) ~ "\n" -%}
-    {%- endfor -%}
-    {%- set xml_tools = ns.xml_tool_string + "</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags." -%}
-  {%- endif -%}
-  {%- if xml_tools -%}
-    {{- xml_tools -}}
-  {%- else -%}
-    {{- "None"  -}}
-  {%- endif -%}
-  {{- "\n\n" -}}
-  {{- "### Python Tools\n\n" -}}
-  {%- if python_tools -%}
-    {{- python_tools -}}
-  {%- else -%}
-    {{- "None"  -}}
+
+  {%- if xml_tools or python_tools -%}
+    {{- "### Tools\n\n" -}}
+    {%- if xml_tools -%}
+      {%- set ns = namespace(xml_tool_string="You may call one or more functions to assist with the user query.\nYou are provided with function signatures within <tools></tools> XML tags:\n\n<tools>\n") -%}
+      {%- for tool in xml_tools[:] -%} {# The slicing makes sure that xml_tools is a list #}
+        {%- set ns.xml_tool_string = ns.xml_tool_string ~ (tool | string) ~ "\n" -%}
+      {%- endfor -%}
+      {%- set xml_tool_string = ns.xml_tool_string + "</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>" -%}
+      {{- xml_tool_string -}}
+    {%- endif -%}
+    {%- if python_tools -%}
+      {%- set ns = namespace(python_tool_string="When you send a message containing Python code between '<code>' and '</code>' tags, it will be executed in a stateful Jupyter notebook environment, and you will then be given the output to continued reasoning in an agentic loop.\n\nYou can use the following tools in your python code like regular functions:\n<tools>\n") -%}
+      {%- for tool in python_tools[:] -%} {# The slicing makes sure that python_tools is a list #}
+        {%- set ns.python_tool_string = ns.python_tool_string ~ (tool | string) ~ "\n" -%}
+      {%- endfor -%}
+      {%- set python_tool_string = ns.python_tool_string + "</tools>\n\nThe state persists between code executions: so variables that you define in one step are still available thereafter." -%}
+      {{- python_tool_string -}}
+    {%- endif -%}
+    {{- "\n\n" -}}
+    {{- "<|im_end|>\n" -}}
   {%- endif -%}
-  {{- "\n\n" -}}
-  {{- "<|im_end|>\n" -}}
 {%- endif -%}
-
 {# ───── main loop ───── #}
 {%- for message in messages -%}
     {%- set content = message.content if message.content is string else "" -%}
@@ -80,12 +77,10 @@
             {{ "<|im_start|>assistant\n" + "<think>\n\n</think>\n" + content.lstrip("\n") + "<|im_end|>\n" }}
         {%- endif -%}
         {% endgeneration %}
-
     {%- elif message.role == "tool" -%}
     {{ "<|im_start|>" + "user\n"  + content + "<|im_end|>\n" }}
     {%- endif -%}
 {%- endfor -%}
-
 {# ───── generation prompt ───── #}
 {%- if add_generation_prompt -%}
     {%- if reasoning_mode == "/think" -%}