File size: 2,328 Bytes
70bf47a | 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 | {%- macro flint_text(t) -%}
{%- if t is string -%}
{{- t | replace("\x00", "") | replace("<think>", "") | replace("</think>", "") | replace("<|begin_of_text|>", "<|begin_of_text|>") | replace("<|end_of_text|>", "<|end_of_text|>") | replace("<|system|>", "<|system|>") | replace("<|developer|>", "<|developer|>") | replace("<|user|>", "<|user|>") | replace("<|assistant|>", "<|assistant|>") | replace("<|observation|>", "<|observation|>") -}}
{%- else -%}
{{- t -}}
{%- endif -%}
{%- endmacro -%}
{%- set tool_preamble -%}
{%- if tools %}<|system|>
# Tools
You may call one or more functions to assist with the user query.
You are provided with function signatures within <tools></tools> XML tags:
<tools>
{%- for tool in tools %}
{%- set fn = tool.function if tool.function is defined else tool %}
{{ fn | tojson(ensure_ascii=False) }}
{%- endfor %}
</tools>
For each function call, output the function name and arguments within the following XML format:
<tool_call>{function-name}<arg_key>{arg-key-1}</arg_key><arg_value>{arg-value-1}</arg_value>...</tool_call>
{% endif -%}
{%- endset -%}
{%- set ns = namespace(tools_pending=true) -%}
{%- for m in messages -%}
{%- if ns.tools_pending and m.role != 'system' -%}
{{ tool_preamble }}
{%- set ns.tools_pending = false -%}
{%- endif -%}
{%- if m.role == 'system' -%}
<|system|>{{ flint_text(m.content) }}
{%- elif m.role == 'developer' -%}
<|developer|>{{ flint_text(m.content) }}
{%- elif m.role == 'user' -%}
<|user|>{{ flint_text(m.content) }}
{%- elif m.role == 'assistant' -%}
<|assistant|>
{%- if m.reasoning_content %}<think>{{ m.reasoning_content }}</think>{% endif -%}
{%- if m.content %}{{ m.content }}{% endif -%}
{%- if m.tool_calls -%}
{%- for tc in m.tool_calls -%}
{%- set fn = tc.function if tc.function is defined else tc -%}
<tool_call>{{ fn.name }}
{%- for k, v in fn.arguments.items() -%}
<arg_key>{{ k }}</arg_key><arg_value>{{ v | tojson(ensure_ascii=False) if v is not string else v }}</arg_value>
{%- endfor -%}
</tool_call>
{%- endfor -%}
{%- endif -%}
{%- elif m.role == 'tool' -%}
<|observation|><tool_response>{{ flint_text(m.content) }}</tool_response>
{%- endif -%}
{%- endfor -%}
{%- if ns.tools_pending -%}
{{ tool_preamble }}
{%- endif -%}
{%- if add_generation_prompt -%}
<|assistant|>
{%- endif -%}
|