From 719e4eb7c87f990063233c5703565c6470b09422 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Hannes=20Gr=C3=A4uler?= <hannes@smasi.de>
Date: Sun, 26 Nov 2023 15:06:14 +0100
Subject: [PATCH] Allow generating images with Dall-E-3

---
 src/v1/common.rs | 4 ++++
 src/v1/image.rs  | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/src/v1/common.rs b/src/v1/common.rs
index 7db9999..fe4561a 100644
--- a/src/v1/common.rs
+++ b/src/v1/common.rs
@@ -41,3 +41,7 @@ pub const GPT4_32K_0613: &str = "gpt-4-32k-0613";
 // - legacy
 pub const GPT4_0314: &str = "gpt-4-0314";
 pub const GPT4_32K_0314: &str = "gpt-4-32k-0314";
+
+// https://platform.openai.com/docs/api-reference/images/object
+pub const DALL_E_2: &str = "dall-e-2";
+pub const DALL_E_3: &str = "dall-e-3";
diff --git a/src/v1/image.rs b/src/v1/image.rs
index c51505c..8932d97 100644
--- a/src/v1/image.rs
+++ b/src/v1/image.rs
@@ -12,6 +12,8 @@ pub struct ImageData {
 pub struct ImageGenerationRequest {
     pub prompt: String,
     #[serde(skip_serializing_if = "Option::is_none")]
+    pub model: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
     pub n: Option<i32>,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub size: Option<String>,
@@ -25,6 +27,7 @@ impl ImageGenerationRequest {
     pub fn new(prompt: String) -> Self {
         Self {
             prompt,
+            model: None,
             n: None,
             size: None,
             response_format: None,