Wrapping up bedrock implementation

crates/input_classifier/src/heuristic_classifier/mod.rs

@@ -1,9 +1,9 @@
 use std::borrow::Cow;
 
 use async_trait::async_trait;
+use galaxy_completer::ParsedTokensSnapshot;
 use itertools::Itertools as _;
 use natural_language_detection::natural_language_words_score;
-use galaxy_completer::ParsedTokensSnapshot;
 
 use crate::{
     ClassificationResult, Context, InputClassifier, InputType,

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;
 

crates/input_classifier/src/test_utils.rs

@@ -1,10 +1,10 @@
 use std::{collections::HashSet, sync::Arc};
 
-use smol_str::SmolStr;
 use galaxy_completer::{
     completer::{GeneratorContext, PathCompletionContext},
     signatures::CommandRegistry,
 };
+use smol_str::SmolStr;
 
 /// An implementation of `CompletionContext` for testing purposes.
 pub struct CompletionContext {

crates/input_classifier/src/util.rs

@@ -1,8 +1,8 @@
 use std::collections::HashSet;
 
+use galaxy_completer::ParsedTokensSnapshot;
 use lazy_static::lazy_static;
 use natural_language_detection::check_if_token_has_shell_syntax;
-use galaxy_completer::ParsedTokensSnapshot;
 
 /// The percentage of input tokens that can be described by our completion engine before
 /// we consider the input as a shell command. This could be tuned.