xz-embed 0.1.0

Text embedding and vector storage abstraction layer
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

use std::collections::HashMap;
use xz_embed::*;

#[test]
fn test_metadata_filter_combinators() {
    let filter = MetadataFilter::and([
        MetadataFilter::eq("source", "docs"),
        MetadataFilter::in_values("lang", &["rust", "go"]),
        MetadataFilter::or([
            MetadataFilter::eq("status", "active"),
            MetadataFilter::eq("status", "draft"),
        ]),
    ]);

    match &filter {
        MetadataFilter::And(inner) => assert_eq!(inner.len(), 3),
        _ => panic!("expected And"),
    }
}

#[test]
fn test_metadata_filter_not() {
    let filter = MetadataFilter::not(MetadataFilter::eq("status", "archived"));

    match &filter {
        MetadataFilter::Not(inner) => match inner.as_ref() {
            MetadataFilter::Eq { key, value } => {
                assert_eq!(key, "status");
                assert_eq!(value, "archived");
            }
            _ => panic!("expected Eq"),
        },
        _ => panic!("expected Not"),
    }
}

#[test]
fn test_filter_from_str_auto() {
    let f = MetadataFilter::eq("key", "val");
    match f {
        MetadataFilter::Eq { key, value } => {
            assert_eq!(key, "key");
            assert_eq!(value, "val");
        }
        _ => panic!(),
    }
}