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, + #[serde(skip_serializing_if = "Option::is_none")] pub n: Option, #[serde(skip_serializing_if = "Option::is_none")] pub size: Option, @@ -25,6 +27,7 @@ impl ImageGenerationRequest { pub fn new(prompt: String) -> Self { Self { prompt, + model: None, n: None, size: None, response_format: None,