mirror of
https://github.com/vllm-project/vllm.git
synced 2026-06-06 00:16:14 +00:00
Andreas Karatzas
41 lines
1.6 KiB
Django/Jinja
41 lines
1.6 KiB
Django/Jinja
{%- if tools %}
|
|
{%- if messages and messages[0]['role'] != 'system' %}
|
|
{{- '<|start_of_role|>system<|end_of_role|>You are a helpful AI assistant with access to the following tools. When a tool is required to answer the user query, respond with <|tool_call|> followed by a JSON list of tools used.<|end_of_text|>
|
|
' }}
|
|
{%- endif %}
|
|
{{- '<|start_of_role|>available_tools<|end_of_role|>
|
|
' }}
|
|
{%- for tool in tools %}
|
|
{{- tool | tojson(indent=4) }}
|
|
{%- if not loop.last %}
|
|
{{- '
|
|
|
|
' }}
|
|
{%- endif %}
|
|
{%- endfor %}
|
|
{{- '<|end_of_text|>
|
|
' }}
|
|
{%- endif %}
|
|
|
|
{%- for message in messages %}
|
|
{%- if message['role'] == 'system' %}
|
|
{{- '<|start_of_role|>system<|end_of_role|>' + message['content'] + '<|end_of_text|>
|
|
' }}
|
|
{%- elif message['role'] == 'user' %}
|
|
{{- '<|start_of_role|>user<|end_of_role|>' + message['content'] + '<|end_of_text|>
|
|
' }}
|
|
{%- elif message['role'] == 'assistant_tool_call' or (message['role'] == 'assistant' and message.tool_calls is defined) %}
|
|
{{- '<|start_of_role|>assistant<|end_of_role|><|tool_call|>' + message.tool_calls|map(attribute='function')|list|tojson(indent=4) + '<|end_of_text|>
|
|
' }}
|
|
{%- elif message['role'] == 'assistant' %}
|
|
{{- '<|start_of_role|>assistant<|end_of_role|>' + message['content'] + '<|end_of_text|>
|
|
' }}
|
|
{%- elif message['role'] == 'tool_response' or message['role'] == 'tool' %}
|
|
{{- '<|start_of_role|>tool_response<|end_of_role|>' + message['content'] + '<|end_of_text|>
|
|
' }}
|
|
{%- endif %}
|
|
{%- if loop.last and add_generation_prompt %}
|
|
{{- '<|start_of_role|>assistant<|end_of_role|>' }}
|
|
{%- endif %}
|
|
{%- endfor %}
|