snapif 0.1.2

Snapif scores one tool call and returns Auto, Review, or Escalate.
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

use indexmap::IndexMap;
use serde_json::{Value, json};
use snapif::question::ChoiceQ;
use snapif::{Client, FakeBackend, Policy, Question, QuestionId, State};

fn department() -> Question {
    let mut criteria = IndexMap::new();
    criteria.insert("billing".to_string(), json!("Payments, invoices, refunds"));
    Question::Choice(ChoiceQ {
        id: QuestionId::new("department"),
        instructions: json!("Which department owns this request?"),
        criteria,
    })
}

fn state() -> State {
    State {
        trusted: json!({}),
        untrusted: Value::Null,
    }
}

fn ask(client: Client<FakeBackend>) -> Client<FakeBackend> {
    pollster::block_on(client.ask(state(), vec![department()])).expect("ask");
    client
}

#[test]
fn builder_sends_the_model_id() {
    let client = ask(
        Client::new(FakeBackend::new().on_choice("department", "billing", 0.91))
            .model("  local-model  ")
            .expect("model")
            .policy(Policy::shipped("tool-gate").expect("policy")),
    );
    assert_eq!(
        client.backend().last_model().as_deref(),
        Some("local-model")
    );
}

#[test]
fn blank_builder_is_policy_error() {
    match Client::new(FakeBackend::new()).model("   ") {
        Err(err) => assert!(err.to_string().contains("SNAPIF_MODEL"), "{err}"),
        Ok(_) => panic!("blank model"),
    }
}

#[test]
fn default_model_on_the_wire_is_jev_latest() {
    let client = ask(
        Client::new(FakeBackend::new().on_choice("department", "billing", 0.91))
            .policy(Policy::shipped("tool-gate").expect("policy")),
    );
    assert_eq!(client.backend().last_model().as_deref(), Some("jev-latest"));
}