apprentice_lib 0.1.1

A library that allows to create LLM agent 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
46
47
48
49

//! Only for tests.

use serde_json::Value;
use crate::error::Error;
use crate::request::client::Client;

/// Client for tests.
pub struct StubClient {
    expected_headers: Vec<(String, String)>,
    expected_params: Vec<(String, String)>,
    expected_payload: Value,
    response_body: Value,
}

impl StubClient {

    /// Create client.
    pub fn new(expected_headers: Vec<(String, String)>,
        expected_params: Vec<(String, String)>,
        expected_payload: Value, 
        response_body: Value) -> Self 
    {
        StubClient {
            expected_headers,
            expected_params,
            expected_payload,
            response_body,
        }
    }
}

impl Client for StubClient {

    fn make_json_request(&self, _url: &str, payload: Value, headers: &[(&str, &str)], params: &[(&str, &str)]) -> Result<Value, Error> {
        for (expected, actual) in headers.iter().zip(self.expected_headers.iter()) {
            assert_eq!(expected.0, actual.0, "headers keys");
            assert_eq!(expected.1, actual.1, "headers values");
        }

        for (expected, actual) in params.iter().zip(self.expected_params.iter()) {
            assert_eq!(expected.0, actual.0, "params keys");
            assert_eq!(expected.1, actual.1, "params values");
        }

        assert_eq!(payload, self.expected_payload);

        Ok(self.response_body.clone())
    }
}