hrm-text-code-tools-sft / chat_template.jinja
vonjack's picture
Upload 12 files
57462bd verified
Raw
History Blame Contribute Delete
1.05 kB
{{- '<|im_start|><|object_ref_start|>' -}}
{%- if tools -%}
{{- '<tools>' -}}
{{- tools | tojson -}}
{{- '</tools>\n' -}}
{%- endif -%}
{%- for message in messages -%}
{%- if message['role'] == 'system' -%}
{{- '<user>\n' + message['content'] + '\n</user>\n' -}}
{%- elif message['role'] == 'user' -%}
{{- '<user>\n' + message['content'] + '\n</user>\n' -}}
{%- elif message['role'] == 'assistant' -%}
{{- '<assistant>\n' -}}
{%- if message.get('content') -%}
{{- message['content'] -}}
{%- endif -%}
{%- for tool_call in message.get('tool_calls', []) -%}
{{- '<tool_call>' -}}
{{- {'arguments': tool_call['function']['arguments'], 'name': tool_call['function']['name']} | tojson -}}
{{- '</tool_call>' -}}
{%- endfor -%}
{{- '\n' -}}
{%- elif message['role'] == 'tool' -%}
{{- '<tool_result>' + message['content'] + '</tool_result>\n' -}}
{%- else -%}
{{- raise_exception('Unsupported message role: ' + message['role']) -}}
{%- endif -%}
{%- endfor -%}
{%- if add_generation_prompt -%}
{{- '<assistant>\n' -}}
{%- endif -%}
{{- '<|im_end|>' -}}