Upload chat_template.jinja

chat_template.jinja

@@ -1,47 +1,51 @@
-{{ bos_token }}
-{%- if messages[0]['role'] == 'system' -%}
-    {%- if messages[0]['content'] is string -%}
-        {%- set first_user_prefix = messages[0]['content'] + '
+{%- if messages[0].role == "system" -%}
+  {%- set system_message = messages[0].content -%}
+  {%- set custom_instructions = system_message.replace("/no_think", "").replace("/think", "").rstrip() -%}
+{%- endif -%}
 
-' -%}
-    {%- else -%}
-        {%- set first_user_prefix = messages[0]['content'][0]['text'] + '
+{%- if custom_instructions -%}
+  {{- custom_instructions + "\n\n" -}}
+{%- endif -%}
 
-' -%}
+{%- if xml_tools or python_tools or tools -%}
+  {{- "### Tools\n\n" -}}
+  {%- if xml_tools or tools -%}
+    {%- if tools -%}
+      {%- set xml_tools = tools -%}
     {%- endif -%}
-    {%- set loop_messages = messages[1:] -%}
-{%- else -%}
-    {%- set first_user_prefix = "" -%}
-    {%- set loop_messages = messages -%}
+    {%- 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" -}}
 {%- endif -%}
-{%- for message in loop_messages -%}
-    {%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
-        {{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
-    {%- endif -%}
-    {%- if (message['role'] == 'assistant') -%}
-        {%- set role = "model" -%}
-    {%- else -%}
-        {%- set role = message['role'] -%}
-    {%- endif -%}
-    {{ '<start_of_turn>' + role + '
-' + (first_user_prefix if loop.first else "") }}
-    {%- if message['content'] is string -%}
-        {{ message['content'] | trim }}
-    {%- elif message['content'] is iterable -%}
-        {%- for item in message['content'] -%}
-            {%- if item['type'] == 'image' -%}
-                {{ '<start_of_image>' }}
-            {%- elif item['type'] == 'text' -%}
-                {{ item['text'] | trim }}
-            {%- endif -%}
-        {%- endfor -%}
-    {%- else -%}
-        {{ raise_exception("Invalid content type") }}
-    {%- endif -%}
-    {{ '<end_of_turn>
-' }}
+
+
+{# ───── main loop ───── #}
+{%- for message in messages -%}
+  {%- set content = message.content if message.content is string else "" -%}
+  {%- if message.role == "user" -%}
+    {{- message.role + ": "  + content + "\n" -}}
+  {%- elif message.role == "assistant" -%}
+  {% generation %}
+    {{- "assistant: " + content.lstrip("\n") + "\n" -}}
+  {% endgeneration %}
+  {%- elif message.role == "tool" -%}
+  {{- "tool: "  + content + "\n" -}}
+  {%- endif -%}
 {%- endfor -%}
+{# ───── generation prompt ───── #}
 {%- if add_generation_prompt -%}
-    {{'<start_of_turn>model
-'}}
-{%- endif -%}
+    {{- "assistant: " -}}
+{%- endif -%}