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

use serde_json::json;

use crate::image_generation::ImageGenerationRequestBuilder;

/// Requests match what the replaced typestate builder produced for the same
/// inputs, captured before it was removed.
#[test]
fn builds_the_requests_the_typestate_builder_built() {
    let request = ImageGenerationRequestBuilder::new("lake").build();
    assert_eq!(
        (
            request.prompt.as_str(),
            request.width,
            request.height,
            request.additional_params
        ),
        ("lake", 256, 256, None)
    );
    let request = ImageGenerationRequestBuilder::new("lake")
        .width(1024)
        .height(512)
        .build();
    assert_eq!((request.width, request.height), (1024, 512));
    let request = ImageGenerationRequestBuilder::new("lake")
        .additional_params(json!({"quality": "low"}))
        .build();
    assert_eq!(request.additional_params, Some(json!({"quality": "low"})));
}