Update chat_template.jinja
#1
by
baseten-admin - opened
- chat_template.jinja +6 -9
chat_template.jinja
CHANGED
|
@@ -1,12 +1,10 @@
|
|
| 1 |
{%- if tools -%}
|
| 2 |
-
<|im_system|>tool_declare<|im_middle|>
|
| 3 |
-
# Tools
|
| 4 |
-
{{ tools | tojson }}<|im_end|>
|
| 5 |
{%- endif -%}
|
| 6 |
-
{
|
| 7 |
{%- if loop.first and messages[0]['role'] != 'system' -%}
|
| 8 |
-
|
| 9 |
-
{
|
| 10 |
|
| 11 |
{%- set role_name = message.get('name') or message['role'] -%}
|
| 12 |
{%- if message['role'] == 'user' -%}
|
|
@@ -15,8 +13,7 @@
|
|
| 15 |
<|im_assistant|>{{role_name}}<|im_middle|>
|
| 16 |
{%- else -%}
|
| 17 |
<|im_system|>{{role_name}}<|im_middle|>
|
| 18 |
-
{
|
| 19 |
-
|
| 20 |
{%- if message['role'] == 'assistant' and message.get('tool_calls') -%}
|
| 21 |
{%- if message['content'] -%}{{ message['content'] }}{%- endif -%}
|
| 22 |
<|tool_calls_section_begin|>
|
|
@@ -27,7 +24,7 @@
|
|
| 27 |
<|tool_calls_section_end|>
|
| 28 |
{%- elif message['role'] == 'tool' -%}
|
| 29 |
## Return of {{ message.tool_call_id }}
|
| 30 |
-
|
| 31 |
{%- elif message['content'] is string -%}
|
| 32 |
{{ message['content'] }}
|
| 33 |
{%- elif message['content'] is not none -%}
|
|
|
|
| 1 |
{%- if tools -%}
|
| 2 |
+
<|im_system|>tool_declare<|im_middle|>{{ tools | tojson(separators=(',', ':')) }}<|im_end|>
|
|
|
|
|
|
|
| 3 |
{%- endif -%}
|
| 4 |
+
{% for message in messages %}
|
| 5 |
{%- if loop.first and messages[0]['role'] != 'system' -%}
|
| 6 |
+
<|im_system|>system<|im_middle|>You are Kimi, an AI assistant created by Moonshot AI.<|im_end|>
|
| 7 |
+
{% endif %}
|
| 8 |
|
| 9 |
{%- set role_name = message.get('name') or message['role'] -%}
|
| 10 |
{%- if message['role'] == 'user' -%}
|
|
|
|
| 13 |
<|im_assistant|>{{role_name}}<|im_middle|>
|
| 14 |
{%- else -%}
|
| 15 |
<|im_system|>{{role_name}}<|im_middle|>
|
| 16 |
+
{%- endif -%}
|
|
|
|
| 17 |
{%- if message['role'] == 'assistant' and message.get('tool_calls') -%}
|
| 18 |
{%- if message['content'] -%}{{ message['content'] }}{%- endif -%}
|
| 19 |
<|tool_calls_section_begin|>
|
|
|
|
| 24 |
<|tool_calls_section_end|>
|
| 25 |
{%- elif message['role'] == 'tool' -%}
|
| 26 |
## Return of {{ message.tool_call_id }}
|
| 27 |
+
{{ message['content'] }}
|
| 28 |
{%- elif message['content'] is string -%}
|
| 29 |
{{ message['content'] }}
|
| 30 |
{%- elif message['content'] is not none -%}
|