create sdk.rs which add SDK like functionality on types

Author: 64bit

async-openai/src/types/audio/mod.rs

@@ -1,6 +1,7 @@
 mod audio_types;
 mod form;
 mod impls;
+mod sdk;
 mod stream;
 
 pub use audio_types::*;

async-openai/src/types/audio/sdk.rs

@@ -0,0 +1,18 @@
+use crate::{error::OpenAIError, types::audio::CreateSpeechResponse, util::create_all_dir};
+use std::path::Path;
+
+impl CreateSpeechResponse {
+    pub async fn save<P: AsRef<Path>>(&self, file_path: P) -> Result<(), OpenAIError> {
+        let dir = file_path.as_ref().parent();
+
+        if let Some(dir) = dir {
+            create_all_dir(dir)?;
+        }
+
+        tokio::fs::write(file_path, &self.bytes)
+            .await
+            .map_err(|e| OpenAIError::FileSaveError(e.to_string()))?;
+
+        Ok(())
+    }
+}

async-openai/src/types/images/mod.rs

@@ -1,6 +1,7 @@
 mod form;
 mod image;
 mod impls;
+mod sdk;
 mod stream;
 
 pub use image::*;

async-openai/src/types/images/sdk.rs

@@ -0,0 +1,56 @@
+use crate::{
+    download::{download_url, save_b64},
+    error::OpenAIError,
+    types::images::{Image, ImagesResponse},
+    util::create_all_dir,
+};
+use std::path::{Path, PathBuf};
+
+impl ImagesResponse {
+    /// Save each image in a dedicated Tokio task and return paths to saved files.
+    /// For [ResponseFormat::Url] each file is downloaded in dedicated Tokio task.
+    pub async fn save<P: AsRef<Path>>(&self, dir: P) -> Result<Vec<PathBuf>, OpenAIError> {
+        create_all_dir(dir.as_ref())?;
+
+        let mut handles = vec![];
+        for id in self.data.clone() {
+            let dir_buf = PathBuf::from(dir.as_ref());
+            handles.push(tokio::spawn(async move { id.save(dir_buf).await }));
+        }
+
+        let results = futures::future::join_all(handles).await;
+        let mut errors = vec![];
+        let mut paths = vec![];
+
+        for result in results {
+            match result {
+                Ok(inner) => match inner {
+                    Ok(path) => paths.push(path),
+                    Err(e) => errors.push(e),
+                },
+                Err(e) => errors.push(OpenAIError::FileSaveError(e.to_string())),
+            }
+        }
+
+        if errors.is_empty() {
+            Ok(paths)
+        } else {
+            Err(OpenAIError::FileSaveError(
+                errors
+                    .into_iter()
+                    .map(|e| e.to_string())
+                    .collect::<Vec<String>>()
+                    .join("; "),
+            ))
+        }
+    }
+}
+
+impl Image {
+    async fn save<P: AsRef<Path>>(&self, dir: P) -> Result<PathBuf, OpenAIError> {
+        match self {
+            Image::Url { url, .. } => download_url(url, dir).await,
+            Image::B64Json { b64_json, .. } => save_b64(b64_json, dir).await,
+        }
+    }
+}

async-openai/src/types/impls.rs

@@ -1,18 +1,13 @@
 use std::path::{Path, PathBuf};
 
-use crate::{
-    download::{download_url, save_b64},
-    error::OpenAIError,
-    types::{
-        audio::{AudioInput, CreateSpeechResponse},
-        chat::{Prompt, StopConfiguration},
-        embeddings::EmbeddingInput,
-        files::FileInput,
-        images::{Image, ImageInput, ImagesResponse},
-        moderations::ModerationInput,
-        InputSource,
-    },
-    util::create_all_dir,
+use crate::types::{
+    audio::AudioInput,
+    chat::{Prompt, StopConfiguration},
+    embeddings::EmbeddingInput,
+    files::FileInput,
+    images::ImageInput,
+    moderations::ModerationInput,
+    InputSource,
 };
 
 use bytes::Bytes;
@@ -146,71 +141,6 @@ impl_input!(AudioInput);
 impl_input!(FileInput);
 impl_input!(ImageInput);
 
-impl ImagesResponse {
-    /// Save each image in a dedicated Tokio task and return paths to saved files.
-    /// For [ResponseFormat::Url] each file is downloaded in dedicated Tokio task.
-    pub async fn save<P: AsRef<Path>>(&self, dir: P) -> Result<Vec<PathBuf>, OpenAIError> {
-        create_all_dir(dir.as_ref())?;
-
-        let mut handles = vec![];
-        for id in self.data.clone() {
-            let dir_buf = PathBuf::from(dir.as_ref());
-            handles.push(tokio::spawn(async move { id.save(dir_buf).await }));
-        }
-
-        let results = futures::future::join_all(handles).await;
-        let mut errors = vec![];
-        let mut paths = vec![];
-
-        for result in results {
-            match result {
-                Ok(inner) => match inner {
-                    Ok(path) => paths.push(path),
-                    Err(e) => errors.push(e),
-                },
-                Err(e) => errors.push(OpenAIError::FileSaveError(e.to_string())),
-            }
-        }
-
-        if errors.is_empty() {
-            Ok(paths)
-        } else {
-            Err(OpenAIError::FileSaveError(
-                errors
-                    .into_iter()
-                    .map(|e| e.to_string())
-                    .collect::<Vec<String>>()
-                    .join("; "),
-            ))
-        }
-    }
-}
-
-impl CreateSpeechResponse {
-    pub async fn save<P: AsRef<Path>>(&self, file_path: P) -> Result<(), OpenAIError> {
-        let dir = file_path.as_ref().parent();
-
-        if let Some(dir) = dir {
-            create_all_dir(dir)?;
-        }
-
-        tokio::fs::write(file_path, &self.bytes)
-            .await
-            .map_err(|e| OpenAIError::FileSaveError(e.to_string()))?;
-
-        Ok(())
-    }
-}
-
-impl Image {
-    async fn save<P: AsRef<Path>>(&self, dir: P) -> Result<PathBuf, OpenAIError> {
-        match self {
-            Image::Url { url, .. } => download_url(url, dir).await,
-            Image::B64Json { b64_json, .. } => save_b64(b64_json, dir).await,
-        }
-    }
-}
-
 macro_rules! impl_from_for_integer_array {
     ($from_typ:ty, $to_typ:ty) => {
         impl<const N: usize> From<[$from_typ; N]> for $to_typ {