rig-core 0.35.0

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
33
34
35
36
37
38
39
40
41
42
43
44
45

//! Migrated from `examples/loaders.rs`.

use rig::loaders::FileLoader;

#[test]
fn file_loader_reads_manifest() {
    let pattern = concat!(env!("CARGO_MANIFEST_DIR"), "/Cargo.toml");
    let results: Vec<_> = FileLoader::with_glob(pattern)
        .expect("glob should parse")
        .read()
        .into_iter()
        .collect();

    assert_eq!(results.len(), 1, "expected exactly one manifest");
    let content = results
        .into_iter()
        .next()
        .expect("expected a result")
        .expect("reading Cargo.toml should succeed");
    assert!(
        content.contains("[package]"),
        "manifest content should include the package section"
    );
}

#[test]
fn loader_fixture_glob_resolves_from_manifest_dir() {
    let results: Vec<_> = FileLoader::with_glob(concat!(
        env!("CARGO_MANIFEST_DIR"),
        "/tests/data/loaders/*.rs"
    ))
    .expect("glob should parse")
    .read_with_path()
    .ignore_errors()
    .into_iter()
    .collect();

    assert_eq!(results.len(), 3, "expected every loader fixture to resolve");
    assert!(
        results
            .iter()
            .any(|(path, _)| path.ends_with("agent_with_loaders.rs")),
        "expected the agent_with_loaders fixture to be included"
    );
}