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

use serde_json::{Value, json};
use snapif::{Client, Decision, FakeBackend, Policy, QuestionId, State};

snapif::choice! {
    enum Department {
        Billing = "billing" => "Payments, invoices, refunds",
        Technical = "technical" => "Bugs, outages, integrations",
        Sales = "sales" => "Pricing, upgrades",
    }
}

fn main() {
    let backend = FakeBackend::new()
        .on_choice("department", "billing", 0.91)
        .on_noul("wants_refund", 0.05);
    let client = Client::new(backend).policy(Policy::shipped("triage").expect("shipped triage"));
    let out = pollster::block_on(client.ask(
        State {
            trusted: json!({"user_request": "invoice"}),
            untrusted: Value::Null,
        },
        snapif::triage::questions(),
    ))
    .expect("ask");
    let decision = out
        .choice::<Department>(&QuestionId::new("department"))
        .expect("choice");
    match decision {
        Decision::Known(Department::Billing) => {
            println!("billing");
        }
        Decision::Known(Department::Technical) => {
            eprintln!("technical");
            std::process::exit(1);
        }
        Decision::Known(Department::Sales) => {
            eprintln!("sales");
            std::process::exit(1);
        }
        Decision::Unsure { .. } => {
            eprintln!("unsure");
            std::process::exit(1);
        }
    }
}