fix: Add MAX_TOKENS to the list of successful finish reasons for Rapi…

…d Evaluation SDK PiperOrigin-RevId: 631138372
googleapis · May 6, 2024 · 195c77e · 195c77e
1 parent b0c5eda
commit 195c77e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,
 ]