add chat_template

2025-10-28 09:40:37 +08:00
parent 93a1a8df02
commit 03f1343d46
3 changed files with 17 additions and 1 deletions
--- a/chat_template.jinja
+++ b/chat_template.jinja
@@ -0,0 +1,16 @@
+{%- for message in messages %}
+{%- if message['role'] == 'system' -%}
+<|im_start|>system
+{{ message['content'] }}<|im_end|>
+{%- elif message['role'] == 'user' -%}
+<|im_start|>user
+{{ message['content'] }}<|im_end|>
+{%- elif message['role'] == 'assistant' -%}
+<|im_start|>assistant
+{{ message['content'] }}<|im_end|>
+{%- elif message['role'] == 'tool' -%}
+<|im_start|>tool
+{{ message['content'] }}<|im_end|>
+{%- endif -%}
+{%- endfor %}
+<|im_start|>assistant