rig-derive 0.43.0

Internal crate that implements Rig derive macros.
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

use rig_core::embeddings::{EmbedError, TextEmbedder, to_texts};

#[derive(rig_derive::Embed)]
struct BasicTuple(u8, #[embed] String, bool, #[embed] String);

fn embed_count(embedder: &mut TextEmbedder, value: usize) -> Result<(), EmbedError> {
    embedder.embed(format!("count:{value}"));
    Ok(())
}

#[derive(rig_derive::Embed)]
struct MixedTuple<T>(
    bool,
    #[embed] T,
    u8,
    #[embed(embed_with = "embed_count")] usize,
);

#[derive(rig_derive::Embed)]
struct Named<T> {
    #[embed]
    body: T,
    #[embed(embed_with = "embed_count")]
    count: usize,
}

#[test]
fn tuple_fields_keep_their_original_indices() {
    let document = BasicTuple(7, "first".into(), false, "second".into());
    assert_eq!(document.0, 7);
    assert!(!document.2);
    assert_eq!(
        to_texts(document).ok(),
        Some(vec!["first".to_owned(), "second".to_owned()])
    );
}

#[test]
fn generic_tuple_supports_basic_and_custom_fields() {
    let document = MixedTuple(false, "body", 9, 42);
    assert!(!document.0);
    assert_eq!(document.2, 9);
    assert_eq!(
        to_texts(document).ok(),
        Some(vec!["body".to_owned(), "count:42".to_owned()])
    );
}

#[test]
fn named_fields_retain_basic_and_custom_dispatch() {
    assert_eq!(
        to_texts(Named {
            body: "body",
            count: 42
        })
        .ok(),
        Some(vec!["body".to_owned(), "count:42".to_owned()]),
    );
}