Surface openai error (mlflow#10176)

bbqiu · Oct 27, 2023 · f10fbd3 · f10fbd3
1 parent d9b8d17
commit f10fbd3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mlflow/metrics/genai/genai_metric.py b/mlflow/metrics/genai/genai_metric.py
@@ -281,6 +281,7 @@ def score_model_on_one_payload(
                     if e.error_code in [
                         ErrorCode.Name(BAD_REQUEST),
                         ErrorCode.Name(UNAUTHENTICATED),
+                        ErrorCode.Name(INVALID_PARAMETER_VALUE)
                     ]:
                         raise MlflowException(e)
                 return None, f"Failed to score model on payload. Error: {e!s}"