diff --git a/src/generation/chat/mod.rs b/src/generation/chat/mod.rs
index 7fdceb0..09d628e 100644
--- a/src/generation/chat/mod.rs
+++ b/src/generation/chat/mod.rs
@@ -297,4 +297,6 @@ pub enum MessageRole {
     Assistant,
     #[serde(rename = "system")]
     System,
+    #[serde(rename = "tool")]
+    Tool,
 }