assay-core 2.17.0

High-performance evaluation framework for LLM agents (Core)
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

use crate::storage::store::Store;

#[derive(Debug, Clone, Copy)]
pub enum QuarantineMode {
    Off,
    Warn,
    Strict,
}

impl QuarantineMode {
    pub fn parse(s: &str) -> Self {
        match s {
            "off" => Self::Off,
            "strict" => Self::Strict,
            _ => Self::Warn,
        }
    }
}

#[derive(Clone)]
pub struct QuarantineService {
    store: Store,
}

impl QuarantineService {
    pub fn new(store: Store) -> Self {
        Self { store }
    }

    pub fn is_quarantined(&self, suite: &str, test_id: &str) -> anyhow::Result<Option<String>> {
        self.store.quarantine_get_reason(suite, test_id)
    }

    pub fn add(&self, suite: &str, test_id: &str, reason: &str) -> anyhow::Result<()> {
        self.store.quarantine_add(suite, test_id, reason)
    }

    pub fn remove(&self, suite: &str, test_id: &str) -> anyhow::Result<()> {
        self.store.quarantine_remove(suite, test_id)
    }
}