update image generation params

Author: MathieuBsqt

examples/images/create_image/src/main.rs

@@ -1,7 +1,8 @@
 use openai_dive::v1::api::Client;
 use openai_dive::v1::models::ImageModel;
 use openai_dive::v1::resources::image::{
-    CreateImageParametersBuilder, ImageQuality, ImageSize, ImageStyle, ResponseFormat,
+    BackgroundStyle, CreateImageParametersBuilder, ImageQuality, ImageSize, ImageStyle,
+    ModerationLevel, OutputFormat, ResponseFormat,
 };
 
 #[tokio::main]
@@ -10,9 +11,14 @@ async fn main() {
 
     let parameters = CreateImageParametersBuilder::default()
         .prompt("A cute dog in the park")
+        .background(BackgroundStyle::Auto)
         .model(ImageModel::DallE3.to_string())
+        .moderation(ModerationLevel::Auto)
         .n(1u32)
-        .quality(ImageQuality::Standard)
+        .output_compression(100u32)
+        .output_format(OutputFormat::Png)
+        .partial_images(0u32)
+        .quality(ImageQuality::Auto)
         .response_format(ResponseFormat::Url)
         .size(ImageSize::Size1024X1024)
         .style(ImageStyle::Natural)

examples/images/create_image_edit/src/main.rs

@@ -1,5 +1,9 @@
 use openai_dive::v1::api::Client;
-use openai_dive::v1::resources::image::{EditImageParametersBuilder, ImageSize, MimeType};
+use openai_dive::v1::models::ImageModel;
+use openai_dive::v1::resources::image::{
+    BackgroundStyle, EditImageParametersBuilder, ImageQuality, ImageSize, MimeType, OutputFormat,
+    ResponseFormat,
+};
 use openai_dive::v1::resources::shared::FileUpload;
 
 #[tokio::main]
@@ -11,10 +15,17 @@ async fn main() {
             "./images/image_edit_original.png".to_string(),
         ))
         .prompt("A cute baby sea otter")
+        .background(BackgroundStyle::Auto)
         .mask(FileUpload::File("./images/image_edit_mask.png".to_string()))
+        .model(ImageModel::DallE3.to_string())
         .n(1u32)
         .mime_type(MimeType::Png)
-        .size(ImageSize::Size512X512)
+        .output_compression(100u32)
+        .output_format(OutputFormat::Png)
+        .partial_images(0u32)
+        .quality(ImageQuality::Auto)
+        .response_format(ResponseFormat::Url)
+        .size(ImageSize::Size1024X1024)
         .build()
         .unwrap();
 

examples/images/create_image_variation/src/main.rs

@@ -1,5 +1,8 @@
 use openai_dive::v1::api::Client;
-use openai_dive::v1::resources::image::{CreateImageVariationParametersBuilder, ImageSize};
+use openai_dive::v1::models::ImageModel;
+use openai_dive::v1::resources::image::{
+    CreateImageVariationParametersBuilder, ImageSize, ResponseFormat,
+};
 use openai_dive::v1::resources::shared::FileUpload;
 
 #[tokio::main]
@@ -10,8 +13,10 @@ async fn main() {
         .image(FileUpload::File(
             "./images/image_edit_original.png".to_string(),
         ))
+        .model(ImageModel::DallE3.to_string())
         .n(1u32)
-        .size(ImageSize::Size256X256)
+        .response_format(ResponseFormat::Url)
+        .size(ImageSize::Size1024X1024)
         .build()
         .unwrap();
 

openai_dive/src/v1/endpoints/images.rs

@@ -92,6 +92,10 @@ impl Images<'_> {
             form = form.text("background", background.to_string());
         }
 
+        if let Some(input_fidelity) = parameters.input_fidelity {
+            form = form.text("input_fidelity", input_fidelity.to_string());
+        }
+
         if let Some(quality) = parameters.quality {
             form = form.text("quality", quality.to_string());
         }
@@ -109,10 +113,26 @@ impl Images<'_> {
             form = form.text("n", n.to_string());
         }
 
+        if let Some(output_compression) = parameters.output_compression {
+            form = form.text("output_compression", output_compression.to_string());
+        }
+
+        if let Some(output_format) = parameters.output_format {
+            form = form.text("output_format", output_format.to_string());
+        }
+
+        if let Some(partial_images) = parameters.partial_images {
+            form = form.text("partial_images", partial_images.to_string());
+        }
+
         if let Some(size) = parameters.size {
             form = form.text("size", size.to_string());
         }
 
+        if let Some(stream) = parameters.stream {
+            form = form.text("stream", stream.to_string());
+        }
+
         if let Some(response_format) = parameters.response_format {
             form = form.text("response_format", response_format.to_string());
         }