Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions core/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,8 @@ pub enum TextEmbeddingsError {
Tokenizer(#[from] tokenizers::Error),
#[error("Input validation error: {0}")]
Validation(String),
#[error("Input validation error: {0}")]
Empty(String),
#[error("Model is overloaded")]
Overloaded(#[from] TryAcquireError),
#[error("Backend error: {0}")]
Expand Down
6 changes: 3 additions & 3 deletions core/src/tokenization.rs
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ impl Tokenization {
) -> Result<ValidEncoding, TextEmbeddingsError> {
// Check if inputs is empty
if inputs.is_empty() {
return Err(TextEmbeddingsError::Validation(
return Err(TextEmbeddingsError::Empty(
"`inputs` cannot be empty".to_string(),
));
}
Expand Down Expand Up @@ -107,7 +107,7 @@ impl Tokenization {
) -> Result<(Option<String>, RawEncoding), TextEmbeddingsError> {
// Check if inputs is empty
if inputs.is_empty() {
return Err(TextEmbeddingsError::Validation(
return Err(TextEmbeddingsError::Empty(
"`inputs` cannot be empty".to_string(),
));
}
Expand Down Expand Up @@ -140,7 +140,7 @@ impl Tokenization {
) -> Result<String, TextEmbeddingsError> {
// Check if inputs is empty
if ids.is_empty() {
return Err(TextEmbeddingsError::Validation(
return Err(TextEmbeddingsError::Empty(
"`input_ids` cannot be empty".to_string(),
));
}
Expand Down
1 change: 1 addition & 0 deletions router/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -552,6 +552,7 @@ impl From<TextEmbeddingsError> for ErrorResponse {
let error_type = match err {
TextEmbeddingsError::Tokenizer(_) => ErrorType::Tokenizer,
TextEmbeddingsError::Validation(_) => ErrorType::Validation,
TextEmbeddingsError::Empty(_) => ErrorType::Empty,
TextEmbeddingsError::Overloaded(_) => ErrorType::Overloaded,
TextEmbeddingsError::Backend(_) => ErrorType::Backend,
};
Expand Down