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

//! Tests for `#[derive(ContextValue)]`.

#![allow(clippy::indexing_slicing, clippy::unwrap_used)]

use rig_core::tool::{ContextValue, ToolContext};

#[derive(Debug, PartialEq, serde::Serialize, serde::Deserialize, rig_core::ContextValue)]
struct SessionId(String);

#[derive(Debug, PartialEq, serde::Serialize, serde::Deserialize, rig_core::ContextValue)]
#[context(key = "session.tenant")]
struct Tenant {
    name: String,
}

#[test]
fn default_key_is_the_type_name() {
    assert_eq!(<SessionId as ContextValue>::KEY, "SessionId");
}

#[test]
fn context_attribute_overrides_the_key() {
    assert_eq!(<Tenant as ContextValue>::KEY, "session.tenant");
}

#[test]
fn derived_values_round_trip_through_a_context_under_their_key() {
    let mut context = ToolContext::new();
    context.insert(SessionId("abc".into())).unwrap();
    context
        .insert(Tenant {
            name: "acme".into(),
        })
        .unwrap();

    assert_eq!(
        context.get::<SessionId>().unwrap(),
        Some(SessionId("abc".into()))
    );
    assert_eq!(
        context.get::<Tenant>().unwrap(),
        Some(Tenant {
            name: "acme".into()
        })
    );

    let json = serde_json::to_value(&context).unwrap();
    assert_eq!(json["inbound"]["SessionId"], serde_json::json!("abc"));
    assert_eq!(
        json["inbound"]["session.tenant"],
        serde_json::json!({"name": "acme"})
    );
}

#[test]
fn a_bare_value_is_rejected_with_the_fix_named() {
    let tests = trybuild::TestCases::new();
    tests.compile_fail("tests/ui/tool_context/fail_bare_value_declares_no_key.rs");
}