Files
rust-genai/tests/support/data.rs
2025-01-01 10:36:38 -08:00

23 lines
777 B
Rust

use base64::engine::general_purpose;
use base64::Engine;
use std::fs::File;
use std::io::Read;
pub const IMAGE_URL_JPG_DUCK: &str = "https://upload.wikimedia.org/wikipedia/commons/thumb/b/bf/Bucephala-albeola-010.jpg/440px-Bucephala-albeola-010.jpg";
/// Get the base64 of the image above (but resized/lower to fit 5kb)
pub fn get_b64_duck() -> Result<String, Box<dyn std::error::Error>> {
// Path to the local image file
let image_path = "./tests/data/duck-small.jpg";
// Open the file and read its contents into a buffer
let mut file = File::open(image_path)?;
let mut buffer = Vec::new();
file.read_to_end(&mut buffer)?;
// Use the general-purpose Base64 engine for encoding
let base64_encoded = general_purpose::STANDARD.encode(&buffer);
Ok(base64_encoded)
}