Update chat_template.jinja

chat_template.jinja

@@ -1,6 +1,7 @@
 {%- set thinking_prompt = 'You are a deep thinking AI, you may use extremely long chains of thought to deeply consider the problem and deliberate with yourself via systematic reasoning processes to help come to a correct solution prior to answering. You should enclose your thoughts and internal monologue inside <think> </think> tags, and then provide your solution or response to the problem.' %}
 {%- set standard_prompt = 'You are Hermes, created by Nous Research.' %}
 {%- if not thinking is defined %}{% set thinking = false %}{% endif %}
+{%- if not keep_cots is defined %}{% set keep_cots = false %}{% endif %}
 {%- if thinking %}{%- set system_prompt = thinking_prompt %}{%- else %}{%- set system_prompt = standard_prompt %}{%- endif %}
 {%- if tools %}
     {{- bos_token + '<|start_header_id|>system<|end_header_id|>\n' }}
@@ -23,12 +24,29 @@
     {%- endif %}
 {%- endif %}
 {%- for message in messages %}
-    {%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
+    {%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
         {{- '<|start_header_id|>' + message.role + '<|end_header_id|>\n\n' + message.content + '<|eot_id|>' }}
+    {%- elif (message.role == "assistant" and not message.tool_calls) %}
+        {{- '<|start_header_id|>' + message.role + '<|end_header_id|>\n' }}
+        {%- if message.content %}
+            {%- set content = message['content'] -%}
+            {%- if thinking %}
+                {%- if not keep_cots %}
+                    {%- set content = '<think> </think>' + content.split('</think>', 1)[1] -%}
+                {%- endif %}
+            {%- endif %}
+            {{- '\n' + content + '<|eot_id|>' }}
+        {%- endif %}
     {%- elif message.role == "assistant" %}
         {{- '<|start_header_id|>' + message.role + '<|end_header_id|>\n' }}
         {%- if message.content %}
-            {{- '\n' + message.content }}
+            {%- set content = message['content'] -%}
+            {%- if thinking %}
+                {%- if not keep_cots %}
+                    {%- set content = '<think> </think>' + content.split('</think>', 1)[1] -%}
+                {%- endif %}
+            {%- endif %}
+            {{- '\n' + content }}
         {%- endif %}
         {%- for tool_call in message.tool_calls %}
             {%- if tool_call.function is defined %}
@@ -55,7 +73,4 @@
 {%- endfor %}
 {%- if add_generation_prompt %}
     {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
-    {%- if thinking %}
-        {{- '<think>' }}
-    {%- endif %}
-{%- endif %}
+{%- endif %}