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
58
59
60
61
62
63
64
65
66
67
68

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/triage.questions.json");

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

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

#[derive(Debug, Deserialize)]
struct RawCriterion {
    label: String,
    description: String,
}

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