Rebrand to Galaxy, major improvements to Bedrock support, still needs some TLC though
This commit is contained in:
@@ -5,7 +5,7 @@ use candle_core::{IndexOp as _, Tensor};
|
||||
use candle_onnx::onnx::ModelProto;
|
||||
use prost::Message as _;
|
||||
use tokenizers::Tokenizer;
|
||||
use warp_completer::ParsedTokensSnapshot;
|
||||
use galaxy_completer::ParsedTokensSnapshot;
|
||||
|
||||
use super::ClassificationResult;
|
||||
|
||||
|
||||
@@ -8,7 +8,7 @@ use std::borrow::Cow;
|
||||
use anyhow::Result;
|
||||
use async_trait::async_trait;
|
||||
use rust_embed::RustEmbed;
|
||||
use warp_completer::ParsedTokensSnapshot;
|
||||
use galaxy_completer::ParsedTokensSnapshot;
|
||||
|
||||
use crate::{
|
||||
ClassificationResult, Context, InputClassifier, InputType,
|
||||
@@ -119,7 +119,7 @@ impl InputClassifier for OnnxClassifier {
|
||||
|
||||
async fn classify_input(
|
||||
&self,
|
||||
input: warp_completer::ParsedTokensSnapshot,
|
||||
input: galaxy_completer::ParsedTokensSnapshot,
|
||||
_context: &Context,
|
||||
) -> anyhow::Result<ClassificationResult> {
|
||||
// If we ever panicked while running inference, we should fall back to the heuristic classifier.
|
||||
|
||||
@@ -6,7 +6,7 @@ use ort::{
|
||||
};
|
||||
use parking_lot::Mutex;
|
||||
use tokenizers::Tokenizer;
|
||||
use warp_completer::ParsedTokensSnapshot;
|
||||
use galaxy_completer::ParsedTokensSnapshot;
|
||||
|
||||
use super::ClassificationResult;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user