bitrouter-guardrails 0.33.0

BitRouter guardrails — local firewall for AI agent traffic at the proxy layer
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use std::sync::Arc;

use bitrouter_core::models::language::language_model::DynLanguageModel;

use crate::engine::Guardrail;

/// A model wrapper that holds an inner [`DynLanguageModel`] and a shared
/// [`Guardrail`] engine. The [`LanguageModel`](bitrouter_core::models::language::language_model::LanguageModel)
/// implementation lives in `router.rs` so that all trait methods are co-located.
pub struct GuardedModel {
    pub(crate) inner: Box<DynLanguageModel<'static>>,
    pub(crate) guardrail: Arc<Guardrail>,
}

impl GuardedModel {
    pub fn new(inner: Box<DynLanguageModel<'static>>, guardrail: Arc<Guardrail>) -> Self {
        Self { inner, guardrail }
    }
}