rig 0.37.1

An opinionated library for building LLM powered applications.
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

#[derive(serde::Deserialize, rig::schemars::JsonSchema)]
struct FacadeCoordinates {
    /// Latitude
    lat: f64,
    /// Longitude
    lng: f64,
}

/// Find nearby places
#[rig::tool_macro]
fn facade_find_nearby(
    /// Location to search from
    location: FacadeCoordinates,
) -> Result<String, rig::tool::ToolError> {
    Ok(format!("{},{}", location.lat, location.lng))
}

#[tokio::test]
async fn test_tool_macro_accepts_facade_schemars_reexport() {
    use rig::tool::Tool;

    let definition = FacadeFindNearby.definition(String::default()).await;
    let schema = serde_json::to_string(&definition.parameters).unwrap();

    assert!(schema.contains("lat"), "expected lat in schema: {schema}");
    assert!(schema.contains("lng"), "expected lng in schema: {schema}");
    assert!(
        schema.contains("Latitude"),
        "expected Latitude description in schema: {schema}"
    );
    assert_eq!(definition.description, "Find nearby places");
}