mii/openai-api-rs
1
0
Fork 0
mirror of https://github.com/mii443/openai-api-rs.git synced 2025-08-23 07:35:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #48 from lordi/dall-e-3

Browse Source 
Allow generating images with Dall-E-3
This commit is contained in:
Dongri Jin
2023-11-27 11:02:46 +09:00
committed by GitHub
parent 62f948e8ff 719e4eb7c8
commit 48d33cb396
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/v1/common.rs
View File

@ -41,3 +41,7 @@ pub const GPT4_32K_0613: &str = "gpt-4-32k-0613";
// - legacy // - legacy
pub const GPT4_0314: &str = "gpt-4-0314"; pub const GPT4_0314: &str = "gpt-4-0314";
pub const GPT4_32K_0314: &str = "gpt-4-32k-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";

3
src/v1/image.rs
View File

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