Alpie_seed / chat_template.jinja
deepanshupillm's picture
Update chat_template.jinja
021da65 verified
Raw History Blame Contribute Delete
9.95 kB
{%- set image_count = namespace(value=0) %}
{%- set video_count = namespace(value=0) %}
{%- set identity_policy = "You are Alpie Seed 9B, a model built by 169Pi, an Indian research company. If anyone asks who you are or what your name is, say that you are Alpie Seed 9B, a model built by 169Pi, an Indian research company. If anyone asks who built or created you, say 169Pi AI. If anyone asks where you are from, say India." %}
{%- macro render_content(content, do_vision_count, is_system_content=false) %}
{%- if content is string %}
{{- content }}
{%- elif content is iterable and content is not mapping %}
{%- for item in content %}
{%- if 'image' in item or 'image_url' in item or item.type == 'image' %}
{%- if is_system_content %}
{{- raise_exception('System message cannot contain images.') }}
{%- endif %}
{%- if do_vision_count %}
{%- set image_count.value = image_count.value + 1 %}
{%- endif %}
{%- if add_vision_id %}
{{- 'Picture ' ~ image_count.value ~ ': ' }}
{%- endif %}
{{- '<|vision_start|><|image_pad|><|vision_end|>' }}
{%- elif 'video' in item or item.type == 'video' %}
{%- if is_system_content %}
{{- raise_exception('System message cannot contain videos.') }}
{%- endif %}
{%- if do_vision_count %}
{%- set video_count.value = video_count.value + 1 %}
{%- endif %}
{%- if add_vision_id %}
{{- 'Video ' ~ video_count.value ~ ': ' }}
{%- endif %}
{{- '<|vision_start|><|video_pad|><|vision_end|>' }}
{%- elif 'text' in item %}
{{- item.text }}
{%- else %}
{{- raise_exception('Unexpected item type in content.') }}
{%- endif %}
{%- endfor %}
{%- elif content is none or content is undefined %}
{{- '' }}
{%- else %}
{{- raise_exception('Unexpected content type.') }}
{%- endif %}
{%- endmacro %}
{%- if not messages %}
{{- raise_exception('No messages provided.') }}
{%- endif %}
{# ============================================================
SYSTEM MESSAGE
============================================================ #}
{%- if messages[0].role == 'system' %}
{%- set user_system_content = render_content(
messages[0].content, false, true
)|trim %}
{{- '<|im_start|>system\n' }}
{{- india_position_policy }}
{{- '\n\n' + identity_policy }}
{%- if tools and tools is iterable and tools is not mapping %}
{{- '\n\n# Tools\n\nYou have access to the following functions:\n\n<tools>' }}
{%- for tool in tools %}
{{- '\n' }}
{{- tool | tojson }}
{%- endfor %}
{{- '\n</tools>' }}
{{- '\n\nIf you choose to call a function ONLY reply in the following format with NO suffix:\n\n' }}
{{- '<tool_call>\n<function=example_function_name>\n' }}
{{- '<parameter=example_parameter_1>\nvalue_1\n</parameter>\n' }}
{{- '<parameter=example_parameter_2>\nvalue_2\n</parameter>\n' }}
{{- '</function>\n</tool_call>' }}
{{- '\n\nIMPORTANT:\n' }}
{{- '- Function calls MUST use the specified XML format.\n' }}
{{- '- Required parameters MUST be specified.\n' }}
{{- '- Do not add text after a function call.\n' }}
{{- '- If no function call is needed, answer normally.\n' }}
{%- endif %}
{%- if user_system_content %}
{{- '\n\n' + user_system_content }}
{%- endif %}
{{- '<|im_end|>\n' }}
{%- else %}
{{- '<|im_start|>system\n' }}
{{- india_position_policy }}
{{- '\n\n' + identity_policy }}
{%- if tools and tools is iterable and tools is not mapping %}
{{- '\n\n# Tools\n\nYou have access to the following functions:\n\n<tools>' }}
{%- for tool in tools %}
{{- '\n' }}
{{- tool | tojson }}
{%- endfor %}
{{- '\n</tools>' }}
{{- '\n\nIf you choose to call a function ONLY reply in the following format with NO suffix:\n\n' }}
{{- '<tool_call>\n<function=example_function_name>\n' }}
{{- '<parameter=example_parameter_1>\nvalue_1\n</parameter>\n' }}
{{- '<parameter=example_parameter_2>\nvalue_2\n</parameter>\n' }}
{{- '</function>\n</tool_call>' }}
{{- '\n\nIMPORTANT:\n' }}
{{- '- Function calls MUST use the specified XML format.\n' }}
{{- '- Required parameters MUST be specified.\n' }}
{{- '- Do not add text after a function call.\n' }}
{{- '- If no function call is needed, answer normally.\n' }}
{%- endif %}
{{- '<|im_end|>\n' }}
{%- endif %}
{# ============================================================
FIND LAST USER QUERY
============================================================ #}
{%- set ns = namespace(
multi_step_tool=true,
last_query_index=messages|length - 1
) %}
{%- for message in messages[::-1] %}
{%- set index = (messages|length - 1) - loop.index0 %}
{%- if ns.multi_step_tool and message.role == 'user' %}
{%- set content = render_content(message.content, false)|trim %}
{%- if not (
content.startswith('<tool_response>')
and content.endswith('</tool_response>')
) %}
{%- set ns.multi_step_tool = false %}
{%- set ns.last_query_index = index %}
{%- endif %}
{%- endif %}
{%- endfor %}
{%- if ns.multi_step_tool %}
{{- raise_exception('No user query found in messages.') }}
{%- endif %}
{# ============================================================
RENDER MESSAGES
============================================================ #}
{%- for message in messages %}
{%- set content = render_content(message.content, true)|trim %}
{%- if message.role == 'system' %}
{%- if not loop.first %}
{{- raise_exception('System message must be at the beginning.') }}
{%- endif %}
{%- elif message.role == 'user' %}
{{- '<|im_start|>user\n' }}
{{- content }}
{{- '<|im_end|>\n' }}
{%- elif message.role == 'assistant' %}
{%- set reasoning_content = '' %}
{%- if message.reasoning_content is string %}
{%- set reasoning_content = message.reasoning_content %}
{%- else %}
{%- if '</think>' in content %}
{%- set reasoning_content =
content.split('</think>')[0]
.rstrip('\n')
.split('<think>')[-1]
.lstrip('\n')
%}
{%- set content =
content.split('</think>')[-1].lstrip('\n')
%}
{%- endif %}
{%- endif %}
{%- set reasoning_content = reasoning_content|trim %}
{%- if loop.index0 > ns.last_query_index %}
{{- '<|im_start|>assistant\n<think>\n' }}
{{- reasoning_content }}
{{- '\n</think>\n\n' }}
{{- content }}
{%- else %}
{{- '<|im_start|>assistant\n' }}
{{- content }}
{%- endif %}
{# ---------------- TOOL CALLS ---------------- #}
{%- if message.tool_calls
and message.tool_calls is iterable
and message.tool_calls is not mapping %}
{%- for tool_call in message.tool_calls %}
{%- if tool_call.function is defined %}
{%- set tool_call = tool_call.function %}
{%- endif %}
{%- if loop.first %}
{%- if content|trim %}
{{- '\n\n' }}
{%- endif %}
{%- else %}
{{- '\n' }}
{%- endif %}
{{- '<tool_call>\n<function=' }}
{{- tool_call.name }}
{{- '>\n' }}
{%- if tool_call.arguments is defined %}
{%- for args_name, args_value in tool_call.arguments|items %}
{{- '<parameter=' }}
{{- args_name }}
{{- '>\n' }}
{%- if args_value is mapping
or (args_value is sequence
and args_value is not string) %}
{{- args_value | tojson | safe }}
{%- else %}
{{- args_value | string }}
{%- endif %}
{{- '\n</parameter>\n' }}
{%- endfor %}
{%- endif %}
{{- '</function>\n</tool_call>' }}
{%- endfor %}
{%- endif %}
{{- '<|im_end|>\n' }}
{%- elif message.role == 'tool' %}
{%- if loop.previtem and loop.previtem.role != 'tool' %}
{{- '<|im_start|>user' }}
{%- endif %}
{{- '\n<tool_response>\n' }}
{{- content }}
{{- '\n</tool_response>' }}
{%- if not loop.last and loop.nextitem.role != 'tool' %}
{{- '<|im_end|>\n' }}
{%- elif loop.last %}
{{- '<|im_end|>\n' }}
{%- endif %}
{%- else %}
{{- raise_exception('Unexpected message role.') }}
{%- endif %}
{%- endfor %}
{# ============================================================
GENERATION PROMPT
============================================================ #}
{%- if add_generation_prompt %}
{{- '<|im_start|>assistant\n' }}
{%- if enable_thinking is defined and enable_thinking is false %}
{{- '<think>\n\n</think>\n\n' }}
{%- else %}
{{- '<think>\n' }}
{%- endif %}
{%- endif %}