From 253a98078a21a014c263bea9f99ae9234a263670 Mon Sep 17 00:00:00 2001
From: Dinghow Yang <DinghowYang@gmail.com>
Date: Fri, 15 Mar 2024 14:19:22 +0800
Subject: [PATCH] Add chat templates for ChatGLM (#3418)

---
 examples/template_chatglm.jinja  | 18 ++++++++++++++++++
 examples/template_chatglm2.jinja | 18 ++++++++++++++++++
 2 files changed, 36 insertions(+)
 create mode 100644 examples/template_chatglm.jinja
 create mode 100644 examples/template_chatglm2.jinja

diff --git a/examples/template_chatglm.jinja b/examples/template_chatglm.jinja
new file mode 100644
index 00000000..bf26f272
--- /dev/null
+++ b/examples/template_chatglm.jinja
@@ -0,0 +1,18 @@
+{%- set counter = namespace(index=0) -%}
+{%- for message in messages -%}
+    {%- if message['role'] == 'user' -%}
+        {{- '[Round ' + counter.index|string + ']\n问：' + message['content'] -}}
+        {%- set counter.index = counter.index + 1 -%}
+    {%- endif -%}
+    {%- if message['role'] == 'assistant' -%}
+        {{- '\n答：' + message['content'] -}}
+        {%- if (loop.last and add_generation_prompt) or not loop.last -%}
+            {{- '\n' -}}
+        {%- endif -%}
+    {%- endif -%}
+{%- endfor -%}
+
+
+{%- if add_generation_prompt and messages[-1]['role'] != 'assistant' -%}
+    {{- '\n答：' -}}
+{%- endif -%}
\ No newline at end of file
diff --git a/examples/template_chatglm2.jinja b/examples/template_chatglm2.jinja
new file mode 100644
index 00000000..c155b7c2
--- /dev/null
+++ b/examples/template_chatglm2.jinja
@@ -0,0 +1,18 @@
+{%- set counter = namespace(index=1) -%}
+{%- for message in messages -%}
+    {%- if message['role'] == 'user' -%}
+        {{- '[Round ' + counter.index|string + ']\n\n问：' + message['content'] -}}
+        {%- set counter.index = counter.index + 1 -%}
+    {%- endif -%}
+    {%- if message['role'] == 'assistant' -%}
+        {{- '\n\n答：' + message['content'] -}}
+        {%- if (loop.last and add_generation_prompt) or not loop.last -%}
+            {{- '\n\n' -}}
+        {%- endif -%}
+    {%- endif -%}
+{%- endfor -%}
+
+
+{%- if add_generation_prompt and messages[-1]['role'] != 'assistant' -%}
+    {{- '\n\n答：' -}}
+{%- endif -%}
\ No newline at end of file