ValidationError was not correctly handled

e86ecbac · Olivier Dehaene · 4c693e65 · e86ecbac
Commit e86ecbac authored Oct 11, 2022 by Olivier Dehaene
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 3 deletions

router/src/server.rs router/src/server.rs +5 -3

No files found.
--- a/router/src/server.rs
+++ b/router/src/server.rs
@@ -69,13 +69,15 @@ async fn generate(
 ) -> poem::Result<Json<serde_json::Value>> {
    let start = Instant::now();
-    let (input_length, validated_request) = validation
+    let (input_length, validated_request) = match validation
        .validate(GenerateRequest {
            inputs: req.inputs.clone(),
            parameters: req.parameters.clone(),
        })
-        .await
+        .await {
-        .unwrap();
+        Ok(result) => result,
+        Err(_) => return Err(poem::Error::from_status(StatusCode::INTERNAL_SERVER_ERROR))
+    };
    let output = infer.infer(input_length, validated_request).await;