omni_search 0.2.6

A unified Rust SDK for multimodal embedding and similarity search.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

use std::path::Path;

#[cfg(feature = "avif")]
pub const SUPPORTED_IMAGE_EXTENSIONS: &[&str] = &["jpg", "jpeg", "png", "webp", "bmp", "avif"];
#[cfg(not(feature = "avif"))]
pub const SUPPORTED_IMAGE_EXTENSIONS: &[&str] = &["jpg", "jpeg", "png", "webp", "bmp"];

#[must_use]
pub fn is_supported_image_path(path: &Path) -> bool {
    path.extension()
        .and_then(|ext| ext.to_str())
        .is_some_and(|ext| SUPPORTED_IMAGE_EXTENSIONS.contains(&ext.to_ascii_lowercase().as_str()))
}

#[cfg(test)]
mod tests {
    use std::path::Path;

    use super::is_supported_image_path;

    #[test]
    fn recognizes_supported_extensions_case_insensitively() {
        assert!(is_supported_image_path(Path::new("sample.jpg")));
        assert!(is_supported_image_path(Path::new("sample.webp")));
        #[cfg(feature = "avif")]
        assert!(is_supported_image_path(Path::new("sample.AVIF")));
        #[cfg(not(feature = "avif"))]
        assert!(!is_supported_image_path(Path::new("sample.AVIF")));
        assert!(!is_supported_image_path(Path::new("sample.gif")));
        assert!(!is_supported_image_path(Path::new("sample")));
    }
}