add default chat template

ad-freiburg · Sep 27, 2024 · 270ecc5 · 270ecc5
1 parent ab8283e
commit 270ecc5
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/data/utils.rs b/src/data/utils.rs
@@ -126,6 +126,19 @@ pub struct ChatTemplate {
     pub end: Option<String>,
 }
 
+impl Default for ChatTemplate {
+    fn default() -> Self {
+        ChatTemplate {
+            start: None,
+            roles: ["user", "assistant", "system"]
+                .iter()
+                .map(|r| (r.to_string(), format!("<|{r}|>\n{{text}}\n\n")))
+                .collect(),
+            end: Some("<|assistant|>\n".to_string()),
+        }
+    }
+}
+
 impl ChatTemplate {
     pub fn new(
         start: Option<String>,