diff --git a/Runtime/DataTypes.cs b/Runtime/DataTypes.cs
index 4d7c670..5e932f3 100644
--- a/Runtime/DataTypes.cs
+++ b/Runtime/DataTypes.cs
@@ -225,6 +225,7 @@ public class CreateImageRequestBase
     public sealed class CreateImageRequest: CreateImageRequestBase
     {
         public string Prompt { get; set; }
+        public string Model { get; set; }
     }
     
     public sealed class CreateImageEditRequest: CreateImageRequestBase
@@ -251,6 +252,8 @@ public struct ImageData
     {
         public string Url { get; set; }
         public string B64Json { get; set; }
+        [JsonProperty(NullValueHandling = NullValueHandling.Ignore)]
+        public string revised_prompt { get; set; }
     }
     #endregion