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

use indexmap::IndexMap;
use serde::Deserialize;
use serde_json::Value;

use crate::ids::QuestionId;
use crate::question::{ChoiceQ, NoulQ, Question};

const RAW: &str = include_str!("../policies/tool-gate.battery.json");

#[derive(Debug, Deserialize)]
struct BatteryFile {
    questions: Vec<BatteryQuestion>,
}

#[derive(Debug, Deserialize)]
struct BatteryQuestion {
    id: String,
    #[serde(rename = "type")]
    kind: String,
    instructions: String,
    #[serde(default)]
    criteria: Vec<String>,
}

pub fn shipped_questions() -> Vec<Question> {
    let file: BatteryFile = serde_json::from_str(RAW).expect("shipped battery json");
    file.questions
        .into_iter()
        .map(|question| match question.kind.as_str() {
            "noul" => Question::Noul(NoulQ {
                id: QuestionId::new(question.id),
                instructions: Value::String(question.instructions),
                criteria: None,
            }),
            "choice" => {
                let criteria = question
                    .criteria
                    .into_iter()
                    .map(|label| (label.clone(), Value::String(label)))
                    .collect::<IndexMap<_, _>>();
                Question::Choice(ChoiceQ {
                    id: QuestionId::new(question.id),
                    instructions: Value::String(question.instructions),
                    criteria,
                })
            }
            other => panic!("unknown battery type {other}"),
        })
        .collect()
}