chat template: tool call fix?

chat_template.jinja

@@ -75,13 +75,13 @@
     {%- elif message.role == "assistant" -%}
         {% generation %}
         {%- if reasoning_mode == "/think" -%}
-            {{ "<|im_start|>assistant\n" + content.lstrip("\n") + "<|im_end|>\n" }}
+            {{ "<|im_start|>" + message.role + "\n" + content.lstrip("\n") + "<|im_end|>\n" }}
         {%- else -%}
-            {{ "<|im_start|>assistant\n" + "<think>\n\n</think>\n" + content.lstrip("\n") + "<|im_end|>\n" }}
+            {{ "<|im_start|>" + message.role + "\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" }}
+    {{ "<|im_start|>" + message.role + "\n"  + content + "<|im_end|>\n" }}
     {%- endif -%}
 {%- endfor -%}
 {# ───── generation prompt ───── #}