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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

use std::path::Path;

use image::{DynamicImage, ImageReader};

use crate::error::Error;

pub(crate) fn load_image_from_path(path: &Path) -> Result<DynamicImage, Error> {
    let reader = ImageReader::open(path)
        .map_err(|error| Error::image_preprocess(format!("{}: {error}", path.display())))?;

    let reader = reader
        .with_guessed_format()
        .map_err(|error| Error::image_preprocess(format!("{}: {error}", path.display())))?;

    reader
        .decode()
        .map_err(|error| Error::image_preprocess(format!("{}: {error}", path.display())))
}

#[cfg(test)]
mod tests {
    use std::fs;
    use std::io::Cursor;

    use image::{DynamicImage, ImageFormat, Rgb, RgbImage};
    use tempfile::tempdir;

    use super::load_image_from_path;

    fn sample_image() -> DynamicImage {
        DynamicImage::ImageRgb8(RgbImage::from_pixel(3, 2, Rgb([12, 34, 56])))
    }

    #[test]
    fn decodes_webp_bytes_from_png_path() {
        let dir = tempdir().unwrap();
        let path = dir.path().join("mislabeled.png");
        let image = sample_image();
        let mut bytes = Cursor::new(Vec::new());

        image.write_to(&mut bytes, ImageFormat::WebP).unwrap();
        fs::write(&path, bytes.into_inner()).unwrap();

        let decoded = load_image_from_path(&path).unwrap();
        assert_eq!(decoded.width(), 3);
        assert_eq!(decoded.height(), 2);
    }

    #[test]
    fn decodes_png_bytes_from_png_path() {
        let dir = tempdir().unwrap();
        let path = dir.path().join("normal.png");
        let image = sample_image();
        let mut bytes = Cursor::new(Vec::new());

        image.write_to(&mut bytes, ImageFormat::Png).unwrap();
        fs::write(&path, bytes.into_inner()).unwrap();

        let decoded = load_image_from_path(&path).unwrap();
        assert_eq!(decoded.width(), 3);
        assert_eq!(decoded.height(), 2);
    }

    #[cfg(feature = "avif")]
    #[test]
    fn decodes_avif_bytes_from_png_path() {
        let dir = tempdir().unwrap();
        let path = dir.path().join("mislabeled.png");
        let image = sample_image();
        let mut bytes = Cursor::new(Vec::new());

        image.write_to(&mut bytes, ImageFormat::Avif).unwrap();
        fs::write(&path, bytes.into_inner()).unwrap();

        let decoded = load_image_from_path(&path).unwrap();
        assert_eq!(decoded.width(), 3);
        assert_eq!(decoded.height(), 2);
    }
}