8f86289373
This is a huge patch which does some major refactoring like: - renaming "Image Generation" to "Image Creation" in most places, to better match its new command (`!bai image create`) - relocating image creation command (`!bai image` -> `!bai image create`), so it wouldn't conflict with the new image editing command (`!bai image edit`) - introducing a new image editing command (`!bai image edit`), which is meant to work only with the OpenAI provider, but doesn't fully work yet due to https://github.com/64bit/async-openai/issues/364, though a next patch will fix it - adding support for reading images off of Matrix conversations and forwarding them to text conversations. Works for OpenAI, but not for Anthropic yet (requires custom patches) and not for OpenAI-Compat (no support for images there) - relocating some utils around (base64, mime)
35 lines
1.2 KiB
Rust
35 lines
1.2 KiB
Rust
use mxlink::mime;
|
|
|
|
pub fn get_file_extension(mime_type: &mime::Mime) -> String {
|
|
match (mime_type.type_(), mime_type.subtype()) {
|
|
(mime::AUDIO, mime::BASIC) => "au",
|
|
(mime::AUDIO, mime::MPEG) => "mp3",
|
|
(mime::AUDIO, mime::MP4) => "m4a",
|
|
(mime::AUDIO, mime::OGG) => "ogg",
|
|
(mime::IMAGE, mime::BMP) => "bmp",
|
|
(mime::IMAGE, mime::GIF) => "gif",
|
|
(mime::IMAGE, mime::JPEG) => "jpg",
|
|
(mime::IMAGE, mime::PNG) => "png",
|
|
(mime::IMAGE, mime::SVG) => "svg",
|
|
_ => "bin",
|
|
}
|
|
.to_string()
|
|
}
|
|
|
|
pub fn get_mime_type_from_file_name(file_name: &str) -> mime::Mime {
|
|
let extension = file_name.rsplit('.').next().unwrap_or("");
|
|
|
|
match extension.to_lowercase().as_str() {
|
|
"jpg" | "jpeg" => mime::IMAGE_JPEG,
|
|
"png" => mime::IMAGE_PNG,
|
|
"gif" => mime::IMAGE_GIF,
|
|
"webp" => "image/webp".parse().unwrap(),
|
|
"svg" => mime::IMAGE_SVG,
|
|
"tiff" | "tif" => "image/tiff".parse().unwrap(),
|
|
"bmp" => "image/bmp".parse().unwrap(),
|
|
"heic" | "heif" => "image/heic".parse().unwrap(),
|
|
"avif" => "image/avif".parse().unwrap(),
|
|
_ => mime::APPLICATION_OCTET_STREAM,
|
|
}
|
|
}
|