diff --git a/vertexai/preview/evaluation/_evaluation.py b/vertexai/preview/evaluation/_evaluation.py index 476f9663e1..c0e5464f8d 100644 --- a/vertexai/preview/evaluation/_evaluation.py +++ b/vertexai/preview/evaluation/_evaluation.py @@ -82,6 +82,7 @@ } _SUCCESSFUL_FINISH_REASONS = [ gapic_content_types.Candidate.FinishReason.STOP, + gapic_content_types.Candidate.FinishReason.MAX_TOKENS, # Many responses have this finish reason gapic_content_types.Candidate.FinishReason.FINISH_REASON_UNSPECIFIED, ]