File size: 5,831 Bytes
a37151b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 | {#-
LLaDA22 chat template — NO-THINK variant.
Usage:
start_header: "<role>assistant</role>"
end_header: "<|role_end|>"
end_learnable_inclusive: true
#}
{#- Message Content Rendering ============================================== #}
{%- macro render_content(content) -%}
{%- if content is string -%}
{{- content -}}
{%- elif content is iterable and content is not mapping -%}
{%- for item in content -%}
{%- if item is mapping -%}
{%- if 'text' in item -%}
{{- item.text -}}
{%- elif 'image' in item or 'image_url' in item or item.get('type') == 'image' -%}
{{- raise_exception('Constraint Violation: Image data detected.') -}}
{%- elif 'video' in item or item.get('type') == 'video' -%}
{{- raise_exception('Constraint Violation: Video data detected.') -}}
{%- else -%}
{{- raise_exception('Invalid mapping structure: Missing "text" key.') -}}
{%- endif -%}
{%- elif item is string -%}
{{- item -}}
{%- else -%}
{{- raise_exception('Invalid item type: Must be string or mapping.') -}}
{%- endif -%}
{%- endfor -%}
{%- elif content is none or content is undefined -%}
{{- '' -}}
{%- else -%}
{{- raise_exception('Fatal error: Content must be a string, iterable, or None.') -}}
{%- endif -%}
{%- endmacro -%}
{%- macro render_toolcalls(message, ns_tool) -%}
{{- '<|tool_calls_section_begin|>' -}}
{%- for tool_call in message.get('tool_calls', []) -%}
{%- set original_id = tool_call.get('id', '') -%}
{%- set func_name = tool_call.get('function', {}).get('name', 'unknown') -%}
{%- set tool_id = 'functions.' + func_name + ':' + ns_tool.index|string -%}
{%- set ns_tool.id_map = ns_tool.id_map + [[original_id, tool_id]] -%}
{%- set ns_tool.index = ns_tool.index + 1 -%}
{%- set args = tool_call.get('function', {}).get('arguments', '') -%}
{{- '<|tool_call_begin|>' + tool_id + '<|tool_call_argument_begin|>' -}}
{%- if args is string -%}
{{- args -}}
{%- else -%}
{{- args | tojson -}}
{%- endif -%}
{{- '<|tool_call_end|>' -}}
{%- endfor -%}
{{- '<|tool_calls_section_end|>' -}}
{%- endmacro -%}
{%- if not messages -%}
{{- raise_exception('No messages provided.') -}}
{%- endif -%}
{# 1. System Instruction, Model Identity & Tools Injection #}
{%- set ns_sys = namespace(has_system=false, content='') -%}
{%- if messages[0].role == 'system' -%}
{%- set ns_sys.has_system = true -%}
{%- if ns_sys.content -%}
{%- set ns_sys.content = ns_sys.content + '\n\n' -%}
{%- endif -%}
{%- set ns_sys.content = ns_sys.content + render_content(messages[0].content) -%}
{%- endif -%}
{%- if tools or ns_sys.content -%}
{{- '<role>system</role>' -}}
{%- if ns_sys.content -%}
{{- ns_sys.content -}}
{%- if tools -%}{{ '\n\n' }}{%- endif -%}
{%- endif -%}
{#- Tool Definition Rendering -#}
{%- if tools and tools is iterable and tools is not mapping -%}
{{- "<tools>\n" -}}
{%- if tools_ts_str -%}
{{- tools_ts_str -}}
{%- else -%}
{%- for tool in tools -%}
{{- tool | tojson(ensure_ascii=False) -}}{{ '\n' -}}
{%- endfor -%}
{%- endif -%}
{{- "</tools>\n" -}}
{%- endif -%}
{{- '<|role_end|>' -}}
{%- endif -%}
{# 2. Main Message Rendering Loop #}
{%- set ns_tool = namespace(index=0, id_map=[]) -%}
{%- for message in messages -%}
{# Skip the first System message as it is pre-rendered above #}
{%- if not (loop.first and message.role == 'system') -%}
{%- set content_text = render_content(message.content) -%}
{%- set role_name = message.get('name') or message.role -%}
{%- if message.role == 'user' or message.role == 'system' -%}
{{- '<role>' + role_name + '</role>' -}}
{{- content_text + '<|role_end|>' -}}
{%- elif message.role == 'assistant' -%}
{{- '<role>' + role_name + '</role>' -}}
{#- NO-THINK: completely ignore reasoning_content, strip <think> from content -#}
{%- set final_content = content_text -%}
{#- Strip embedded <think>...</think> from content if present -#}
{%- if '<think>' in final_content and '</think>' in final_content -%}
{%- set parts = final_content.split('</think>', 1) -%}
{%- set final_content = parts[1].strip() -%}
{%- endif -%}
{#- Output content directly, no think tags -#}
{%- if final_content -%}
{{- final_content -}}
{%- endif -%}
{%- if message.tool_calls -%}
{{- render_toolcalls(message, ns_tool) -}}
{%- endif -%}
{{- '<|role_end|>' -}}
{%- elif message.role == 'tool' -%}
{{- '<role>tool</role>' -}}
{%- set original_id = message.get('tool_call_id', '') -%}
{%- set ns_lookup = namespace(mapped_id=original_id) -%}
{%- for pair in ns_tool.id_map -%}
{%- if pair[0] == original_id -%}
{%- set ns_lookup.mapped_id = pair[1] -%}
{%- endif -%}
{%- endfor -%}
{{- '## Return of ' + ns_lookup.mapped_id + '\n' -}}
{{- content_text + '\n<|role_end|>' -}}
{%- endif -%}
{%- endif -%}
{%- endfor -%}
{# 3. Generation Prompt Suffix — no think tag #}
{%- if add_generation_prompt -%}
{{- '<role>assistant</role>' -}}
{%- endif -%}
|