system2 0.1.0

An (unofficial) idiomatic Rust SDK for TypeSafe's System One model, Jev.
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

//! Tests of `Client::from_env` with blank variables.

mod support;

use std::env;
use support::{INSTRUCTIONS, STATE, mock_api, noul_response, only_request, request_body};
use system2::{AskError, Client, Noul};

#[tokio::test]
#[expect(unsafe_code, reason = "the test sets the environment")]
async fn from_env_ignores_blank_variables() {
    // SAFETY: this test is alone in its binary, and it starts no thread before this
    unsafe {
        env::set_var("TYPESAFE_API_KEY", "");
        env::set_var("TYPESAFE_BASE_URL", " ");
        env::set_var("TYPESAFE_DEFAULT_MODEL", "\n");
        env::set_var("SYSTEM2_CACHE", " ");
        env::set_var("SYSTEM2_OFFLINE", "");
    }

    let client = Client::from_env().unwrap();

    let error = client
        .ask(Noul::new(INSTRUCTIONS).unwrap(), STATE)
        .await
        .unwrap_err();
    assert!(matches!(error, AskError::MissingApiKey));
    assert!(format!("{client:?}").contains("cache: None, offline: false"));

    let server = mock_api(noul_response(0.92)).await;
    client
        .api_key("key")
        .base_url(server.uri())
        .ask(Noul::new(INSTRUCTIONS).unwrap(), STATE)
        .await
        .unwrap();

    let request = only_request(&server).await;
    assert_eq!(request_body(&request)["model"], "jev-latest");
}