Wrapping up bedrock implementation

crates/input_classifier/src/onnx/candle.rs

@@ -3,9 +3,9 @@ use std::collections::HashMap;
 use anyhow::{Context as _, Result, ensure};
 use candle_core::{IndexOp as _, Tensor};
 use candle_onnx::onnx::ModelProto;
+use galaxy_completer::ParsedTokensSnapshot;
 use prost::Message as _;
 use tokenizers::Tokenizer;
-use galaxy_completer::ParsedTokensSnapshot;
 
 use super::ClassificationResult;
 

crates/input_classifier/src/onnx/mod.rs

@@ -7,8 +7,8 @@ use std::borrow::Cow;
 
 use anyhow::Result;
 use async_trait::async_trait;
-use rust_embed::RustEmbed;
 use galaxy_completer::ParsedTokensSnapshot;
+use rust_embed::RustEmbed;
 
 use crate::{
     ClassificationResult, Context, InputClassifier, InputType,

crates/input_classifier/src/onnx/ort.rs

@@ -1,4 +1,5 @@
 use anyhow::{Result, ensure};
+use galaxy_completer::ParsedTokensSnapshot;
 use itertools::Itertools as _;
 use ort::{
     execution_providers::CPUExecutionProvider, session::Session, tensor::ArrayExtensions as _,
@@ -6,7 +7,6 @@ use ort::{
 };
 use parking_lot::Mutex;
 use tokenizers::Tokenizer;
-use galaxy_completer::ParsedTokensSnapshot;
 
 use super::ClassificationResult;