diff --git a/src/main.rs b/src/main.rs
index 5348e5a..10d0c74 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -31,10 +31,12 @@ struct ChatPayload {
 
 #[derive(Debug, Clone, Serialize, Deserialize)]
 struct ChatChunk {
-    pub role: String,
+    pub role: Option<String>,
     pub message: String,
     pub created: u64,
     pub action: String,
+    pub id: Option<String>,
+    pub model: Option<String>
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize)]