61 lines
2.5 KiB
Django/Jinja
61 lines
2.5 KiB
Django/Jinja
{% macro render_content(msg) -%}
|
|
{%- set c = msg.get('content') -%}
|
|
{%- if c is string -%}
|
|
{{ c }}
|
|
{%- elif c is not none -%}
|
|
{% for content in c -%}
|
|
{% if content['type'] == 'image' or 'image' in content or 'image_url' in content -%}
|
|
<|media_start|>image<|media_content|><|media_pad|><|media_end|>
|
|
{% else -%}
|
|
{{ content['text'] }}
|
|
{%- endif -%}
|
|
{%- endfor -%}
|
|
{%- endif -%}
|
|
{%- endmacro %}
|
|
|
|
{%- set tool_response_queue = namespace(ids=[]) -%}
|
|
{%- set tool_call_counter = namespace(value=0) -%}
|
|
|
|
{%- if tools -%}
|
|
<|im_system|>tool_declare<|im_middle|>{{ tools | tojson }}<|im_end|>
|
|
{%- endif -%}
|
|
{% for message in messages %}
|
|
{%- if loop.first and messages[0]['role'] != 'system' -%}
|
|
<|im_system|>system<|im_middle|>You are Kimi, an AI assistant created by Moonshot AI.<|im_end|>
|
|
{% endif %}
|
|
|
|
{%- set role_name = message.get('name') or message['role'] -%}
|
|
{%- if message['role'] == 'user' -%}
|
|
<|im_user|>{{role_name}}<|im_middle|>
|
|
{%- elif message['role'] == 'assistant' -%}
|
|
<|im_assistant|>{{role_name}}<|im_middle|>
|
|
{%- else -%}
|
|
<|im_system|>{{role_name}}<|im_middle|>
|
|
{%- endif -%}
|
|
|
|
{%- if message['role'] == 'assistant' and message.get('tool_calls') -%}
|
|
{{render_content(message)}}<|tool_calls_section_begin|>
|
|
{%- for tool_call in message['tool_calls'] -%}
|
|
{%- set formatted_id = 'functions.' + tool_call['function']['name'] + ':' + (tool_call_counter.value | string) -%}
|
|
{%- set tool_call_counter.value = tool_call_counter.value + 1 -%}
|
|
{%- set _ = tool_response_queue.ids.append(formatted_id) -%}
|
|
<|tool_call_begin|>{{ formatted_id }}<|tool_call_argument_begin|>{% if tool_call['function']['arguments'] is string %}{{ tool_call['function']['arguments'] }}{% else %}{{ tool_call['function']['arguments'] | tojson }}{% endif %}<|tool_call_end|>
|
|
{%- endfor -%}
|
|
<|tool_calls_section_end|>
|
|
{%- elif message['role'] == 'tool' -%}
|
|
{%- if tool_response_queue.ids -%}
|
|
{%- set tool_call_id = tool_response_queue.ids.pop(0) -%}
|
|
{%- else -%}
|
|
{%- set tool_call_id = 'functions.' + message.get('name', 'unknown') + ':' + (tool_call_counter.value | string) -%}
|
|
{%- endif -%}
|
|
## Return of {{ tool_call_id }}
|
|
{{render_content(message)}}
|
|
{%- elif message['content'] is not none -%}
|
|
{{render_content(message)}}
|
|
{%- endif -%}
|
|
<|im_end|>
|
|
{%- endfor -%}
|
|
{%- if add_generation_prompt -%}
|
|
<|im_assistant|>assistant<|im_middle|>
|
|
{%- endif -%}
|