Surface openai error (mlflow#10176)

prithvikannan · web-flow · commit f10fbd3e6b9a · 2023-10-26T18:30:58.000-07:00
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}"
