rig-core 0.44.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

use assert_fs::prelude::{FileTouch, FileWriteStr, PathChild};

use super::FileLoader;

#[test]
fn test_file_loader_bytes() {
    let temp = assert_fs::TempDir::new().expect("Failed to create temp dir");
    let foo_file = temp.child("foo.txt");
    let bar_file = temp.child("bar.txt");

    foo_file.touch().expect("Failed to create foo.txt");
    bar_file.touch().expect("Failed to create bar.txt");

    foo_file.write_str("foo").expect("Failed to write to foo");
    bar_file.write_str("bar").expect("Failed to write to bar");

    let foo_bytes = std::fs::read(foo_file.path()).unwrap();
    let bar_bytes = std::fs::read(bar_file.path()).unwrap();

    let loader = FileLoader::from_bytes_multi(vec![foo_bytes, bar_bytes]);
    let mut actual = loader
        .read()
        .ignore_errors()
        .into_iter()
        .collect::<Vec<_>>();
    let mut expected = vec!["foo".to_string(), "bar".to_string()];

    actual.sort();
    expected.sort();

    assert!(!actual.is_empty());
    assert!(expected == actual);
}