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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

//! Tests of the first `.ask` with a bad variable in the environment.

mod support;

use std::{env, fs};
use support::{INSTRUCTIONS, STATE, cache_path, noul};
use system2::prelude::*;
use system2::{AskError, CacheError, Client, EnvError, NoulAnswer};

#[expect(unsafe_code, reason = "the test sets the environment")]
fn set_var(name: &str, value: &str) {
    // SAFETY: this test is alone in its binary, and it starts no thread, as each ask fails
    // before a request
    unsafe { env::set_var(name, value) }
}

// the first `.ask` with one variable
async fn ask_with(name: &str, value: &str) -> Result<NoulAnswer, AskError> {
    set_var(name, value);
    let answer = noul(INSTRUCTIONS).ask(STATE).await;
    set_var(name, "");
    answer
}

#[tokio::test]
async fn ask_makes_the_global_client_again_after_a_bad_variable() {
    let folder =
        cache_path("ask_makes_the_global_client_again_after_a_bad_variable").with_extension("");
    let _ = fs::remove_dir_all(&folder);
    let path = folder.join("cache.jsonl");
    for name in ["TYPESAFE_API_KEY", "TYPESAFE_BASE_URL", "SYSTEM2_CACHE"] {
        set_var(name, "");
    }
    set_var("SYSTEM2_OFFLINE", "0");

    set_var("TYPESAFE_API_KEY", "first\nsecond");
    let global = Client::global().unwrap_err();
    set_var("TYPESAFE_API_KEY", "");

    assert!(matches!(global, EnvError::InvalidApiKey), "{global:?}");

    let key = ask_with("TYPESAFE_API_KEY", "first\nsecond")
        .await
        .unwrap_err();
    let url = ask_with("TYPESAFE_BASE_URL", "api.typesafe.ai")
        .await
        .unwrap_err();
    let cache = ask_with("SYSTEM2_CACHE", path.to_str().unwrap())
        .await
        .unwrap_err();

    assert!(matches!(key, AskError::InvalidApiKey), "{key:?}");
    assert!(matches!(url, AskError::InvalidBaseUrl { .. }), "{url:?}");
    assert!(
        matches!(cache, AskError::Cache(CacheError::Read { .. })),
        "{cache:?}"
    );

    // the cache opens now, and the client is online, so it needs a key
    fs::create_dir(&folder).unwrap();
    set_var("SYSTEM2_CACHE", path.to_str().unwrap());
    let online = noul(INSTRUCTIONS).ask(STATE).await.unwrap_err();

    assert!(matches!(online, AskError::MissingApiKey), "{online:?}");
    assert!(path.exists());

    // the global client stays, so a later bad variable does not change it
    set_var("TYPESAFE_API_KEY", "first\nsecond");
    let kept = noul(INSTRUCTIONS).ask(STATE).await.unwrap_err();

    assert!(matches!(kept, AskError::MissingApiKey), "{kept:?}");
}