Commit d6d5b12e authored by OlivierDehaene's avatar OlivierDehaene
Browse files

fix(router): Handle tokenizer errors

parent feb7806c
...@@ -123,7 +123,8 @@ fn validation_worker( ...@@ -123,7 +123,8 @@ fn validation_worker(
} }
// Get the number of tokens in the input // Get the number of tokens in the input
let inputs = tokenizer.encode(request.inputs.clone(), false).unwrap(); match tokenizer.encode(request.inputs.clone(), false) {
Ok(inputs) => {
let input_length = inputs.len(); let input_length = inputs.len();
if input_length > max_input_length { if input_length > max_input_length {
...@@ -138,6 +139,11 @@ fn validation_worker( ...@@ -138,6 +139,11 @@ fn validation_worker(
response_tx.send(Ok((input_length, request))).unwrap_or(()); response_tx.send(Ok((input_length, request))).unwrap_or(());
} }
Err(err) => response_tx
.send(Err(ValidationError::Tokenizer(err.to_string())))
.unwrap_or(()),
};
}
} }
type ValidationRequest = ( type ValidationRequest = (
...@@ -157,6 +163,8 @@ pub enum ValidationError { ...@@ -157,6 +163,8 @@ pub enum ValidationError {
MaxNewTokens, MaxNewTokens,
#[error("inputs must have less than {1} tokens. Given: {0}")] #[error("inputs must have less than {1} tokens. Given: {0}")]
InputLength(usize, usize), InputLength(usize, usize),
#[error("tokenizer error {0}")]
Tokenizer(String),
} }
impl From<ValidationError> for (StatusCode, Json<ErrorResponse>) { impl From<ValidationError> for (StatusCode, Json<ErrorResponse>) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment