mod: add chat_template.jinja

chat_template.jinja

@@ -0,0 +1,168 @@
+{#- Harmony Chat Template for Multistral -#}
+{#- Enhanced version with support for both toon and json formats -#}
+
+{#- Configuration Variables -#}
+{%- set use_toon = use_toon | default(true) -%}
+
+{#- Tool Definition Rendering -#}
+{%- macro render_typescript_type(param_spec, required_params) -%}
+    {%- if param_spec.type == "array" -%}
+        {%- if param_spec.items and param_spec.items.type == "string" -%}
+            string[]
+        {%- elif param_spec.items and param_spec.items.type in ["number", "integer"] -%}
+            number[]
+        {%- elif param_spec.items and param_spec.items.type == "boolean" -%}
+            boolean[]
+        {%- else -%}
+            any[]
+        {%- endif -%}
+    {%- elif param_spec.type == "string" -%}
+        {%- if param_spec.enum -%}
+            "{{ param_spec.enum | join('" | "') }}"
+        {%- else -%}
+            string
+        {%- endif -%}
+    {%- elif param_spec.type in ["number", "integer"] -%}
+        number
+    {%- elif param_spec.type == "boolean" -%}
+        boolean
+    {%- elif param_spec.type == "object" -%}
+        {%- if param_spec.properties -%}
+            {
+{%- for prop_name, prop_spec in param_spec.properties.items() -%}
+{{ prop_name }}{% if prop_name not in (param_spec.required or []) %}?{% endif %}: {{ render_typescript_type(prop_spec, param_spec.required or []) }}{% if not loop.last %}, {% endif %}
+{%- endfor %}
+}
+        {%- else -%}
+            object
+        {%- endif -%}
+    {%- elif param_spec.oneOf -%}
+        {%- for variant in param_spec.oneOf -%}
+            {{ render_typescript_type(variant, required_params) }}{% if not loop.last %} | {% endif %}
+        {%- endfor -%}
+    {%- else -%}
+        any
+    {%- endif -%}
+{%- endmacro -%}
+
+{%- macro render_tool_namespace(namespace_name, tools) -%}
+## {{ namespace_name }}
+
+namespace {{ namespace_name }} {
+
+{%- for tool in tools %}
+{%- set tool = tool.function if tool.function is defined else tool %}
+// {{ tool.description }}
+type {{ tool.name }} = {% if tool.parameters and tool.parameters.properties %}(_: {
+{%- for param_name, param_spec in tool.parameters.properties.items() %}
+{%- if param_spec.description %}
+// {{ param_spec.description }}
+{%- endif %}
+{{ param_name }}{% if param_name not in (tool.parameters.required or []) %}?{% endif %}: {{ render_typescript_type(param_spec, tool.parameters.required or []) }}{% if param_spec.default is defined %}, // default: {{ param_spec.default | tojson }}{% endif %}{% if not loop.last %},{% endif %}
+{%- endfor %}
+}) => any{% else %}() => any{% endif %};
+
+{%- endfor %}
+} // namespace {{ namespace_name }}
+{%- endmacro -%}
+
+{#- Format tool call arguments based on the selected format -#}
+{%- macro format_tool_call_arguments(tool_call, message) -%}
+    {%- if use_toon -%}
+        {%- if tool_call.arguments -%}
+            {%- if tool_call.arguments is string -%}
+                {{- tool_call.arguments | fromjson | to_toon -}}
+            {%- else -%}
+                {{- tool_call.arguments | to_toon -}}
+            {%- endif -%}
+        {%- else -%}
+            {{- {} | to_toon -}}
+        {%- endif -%}
+    {%- else -%}
+        {%- if tool_call.arguments -%}
+            {%- if tool_call.arguments is string -%}
+                {{- tool_call.arguments -}}
+            {%- else -%}
+                {{- tool_call.arguments | tojson -}}
+            {%- endif -%}
+        {%- else -%}
+            {{- {} | tojson -}}
+        {%- endif -%}
+    {%- endif -%}
+{%- endmacro -%}
+
+{#- System Message -#}
+<|start|>system<|message|>{{ model_identity or "You are Aizia, a helpful assistant trained by damfle." }}
+Knowledge cutoff: 2025-12
+Current date: {{ strftime_now("%Y-%m-%d") }}
+
+Reasoning: {{ reasoning_effort or "none" }}
+
+# Valid channels: analysis, commentary, final. Channel must be included for every message.
+{%- if tools %}
+Calls to these tools must go to the commentary channel: 'functions'.
+{%- endif %}<|end|>
+
+{#- Extract developer message -#}
+{%- if messages and messages[0].role in ["developer", "system"] %}
+    {%- set developer_message = messages[0].content %}
+    {%- set loop_messages = messages[1:] %}
+{%- else %}
+    {%- set developer_message = "" %}
+    {%- set loop_messages = messages %}
+{%- endif %}
+
+{#- Render developer message and tools -#}
+{%- if developer_message or tools %}
+<|start|>developer<|message|>
+{%- if developer_message %}
+# Instructions
+
+{{ developer_message }}
+
+{%- endif %}
+{%- if tools %}
+# Tools
+
+{{ render_tool_namespace("functions", tools) }}
+{%- endif %}
+<|end|>
+{%- endif %}
+
+{#- Process messages -#}
+{%- set last_tool_call = namespace(name=none) %}
+{%- for message in loop_messages %}
+    {%- if message.role == 'assistant' %}
+        {%- if "tool_calls" in message and message.tool_calls %}
+            {#- Assistant message with tool calls -#}
+            {%- if message.reasoning %}
+<|start|>assistant<|channel|>analysis<|message|>{{ message.reasoning }}<|end|>
+            {%- endif %}
+
+            {%- for tool_call in message.tool_calls %}
+                {%- set tool_call = tool_call.function if tool_call.function else tool_call %}
+<|start|>assistant to=functions.{{ tool_call.name }}<|channel|>commentary<|constrain|>{{ "toon" if use_toon else "json" }}<|message|>{{ format_tool_call_arguments(tool_call, message) }}<|call|>
+                {%- set last_tool_call.name = tool_call.name %}
+            {%- endfor %}
+        {%- else %}
+            {#- Regular assistant message -#}
+            {%- if message.reasoning %}
+<|start|>assistant<|channel|>analysis<|message|>{{ message.reasoning }}<|end|>
+            {%- endif %}
+<|start|>assistant<|channel|>final<|message|>{{ message.content }}<|end|>
+            {%- set last_tool_call.name = none %}
+        {%- endif %}
+    {%- elif message.role == 'tool' %}
+        {%- if last_tool_call.name is none %}
+            {{ raise_exception("Message has tool role, but there was no previous assistant message with a tool call!") }}
+        {%- endif %}
+<|start|>functions.{{ last_tool_call.name }} to=assistant<|channel|>commentary<|message|>{{ message.content | tojson }}<|end|>
+    {%- elif message.role == 'user' %}
+<|start|>user<|message|>{{ message.content }}<|end|>
+    {%- endif %}
+{%- endfor %}
+
+{#- Generation prompt -#}
+{%- if add_generation_prompt %}
+<|start|>assistant
+{%- endif %}