GabrielTofvesson/OpenAI-Rust
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implement translation test

Browse Source
This commit is contained in:
Gabriel Tofvesson 2023-03-18 12:48:37 +01:00
parent 82bfadc2fe
commit c4ae754bca
No known key found for this signature in database
GPG Key ID: 6F1345DF28EDA13E
2 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
french.mp3 Normal file
View File

Binary file not shown.

24
src/lib.rs
View File

@ -26,6 +26,7 @@ mod tests {
use crate::image_variation::ImageVariationRequestBuilder; use crate::image_variation::ImageVariationRequestBuilder;
use crate::embedding::EmbeddingRequestBuilder; use crate::embedding::EmbeddingRequestBuilder;
use crate::transcription::{TranscriptionRequestBuilder, AudioFile}; use crate::transcription::{TranscriptionRequestBuilder, AudioFile};
use crate::translation::TranslationRequestBuilder;
fn get_api() -> anyhow::Result<Context> { fn get_api() -> anyhow::Result<Context> {
Ok(Context::new(std::fs::read_to_string(std::path::Path::new("apikey.txt"))?.trim().to_string())) Ok(Context::new(std::fs::read_to_string(std::path::Path::new("apikey.txt"))?.trim().to_string()))
@ -128,7 +129,6 @@ mod tests {
#[tokio::test] #[tokio::test]
async fn test_image_edit() { async fn test_image_edit() {
const IMAGE_PROMPT: &str = "A real ginger cat gracefully walking along a real, thin brick wall";
let ctx = get_api(); let ctx = get_api();
assert!(ctx.is_ok(), "Could not load context"); assert!(ctx.is_ok(), "Could not load context");
let ctx = ctx.unwrap(); let ctx = ctx.unwrap();
@ -136,7 +136,7 @@ mod tests {
let image = ctx.create_image_edit( let image = ctx.create_image_edit(
ImageEditRequestBuilder::default() ImageEditRequestBuilder::default()
.image(File::open("clown.png").await.unwrap()) .image(File::open("clown.png").await.unwrap())
.prompt("Blue nose") .prompt("Clown with a round, blue nose")
.build() .build()
.unwrap() .unwrap()
).await; ).await;
@ -144,7 +144,6 @@ mod tests {
assert!(image.is_ok(), "Could not get image: {}", image.unwrap_err()); assert!(image.is_ok(), "Could not get image: {}", image.unwrap_err());
assert!(image.as_ref().unwrap().data.len() == 1, "No image found"); assert!(image.as_ref().unwrap().data.len() == 1, "No image found");
assert!(matches!(image.as_ref().unwrap().data[0], Image::URL(_)), "No image found"); assert!(matches!(image.as_ref().unwrap().data[0], Image::URL(_)), "No image found");
println!("Image prompt: {IMAGE_PROMPT}");
match image.unwrap().data[0] { match image.unwrap().data[0] {
Image::URL(ref url) => { Image::URL(ref url) => {
println!("Generated edited image URL: {url}"); println!("Generated edited image URL: {url}");
@ -221,5 +220,22 @@ mod tests {
println!("Transcription: {:?}", transcription.unwrap().text); println!("Transcription: {:?}", transcription.unwrap().text);
} }
// TODO: Add translation test #[tokio::test]
async fn test_translation() {
let ctx = get_api();
assert!(ctx.is_ok(), "Could not load context");
let ctx = ctx.unwrap();
let translation = ctx.create_translation(
TranslationRequestBuilder::default()
.model("whisper-1")
.prompt("[English]") // Without this, Whisper just responds with the French transcript?
.file(AudioFile::MP3(File::open("french.mp3").await.unwrap()))
.build()
.unwrap()
).await;
assert!(translation.is_ok(), "Could not get translation: {}", translation.unwrap_err());
println!("Translation: {:?}", translation.unwrap().text);
}
} }