llm_shield_models/
lib.rs

1//! ML Model Infrastructure for LLM Shield
2//!
3//! This crate provides infrastructure for loading and running ONNX models
4//! for ML-based security scanners.
5
6pub mod model_loader;
7pub mod tokenizer;
8pub mod inference;
9pub mod registry;
10pub mod cache;
11pub mod types;
12
13pub use model_loader::{ModelLoader, ModelConfig, ModelType};
14pub use tokenizer::{TokenizerWrapper, TokenizerConfig, Encoding};
15pub use inference::{InferenceEngine, InferenceResult, TokenPrediction, PostProcessing};
16pub use registry::{ModelRegistry, ModelTask, ModelVariant, ModelMetadata};
17pub use cache::{ResultCache, CacheConfig, CacheStats};
18pub use types::{
19    MLConfig, CacheSettings, HybridMode, DetectionMethod, InferenceMetrics,
20};
21
22use llm_shield_core::Error;
23
24/// Result type alias
25pub type Result<T> = std::result::Result<T, Error>;
llm_shield_models/lib.rs

llm_shield_models/
lib.rs
