fix: Add MAX_TOKENS to the list of successful finish reasons for Rapid Evaluation SDK

jsondai · copybara-github · commit 195c77ed7320 · 2024-05-06T11:58:56.000-07:00
PiperOrigin-RevId: 631138372
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,
 ]

Original file line number	Diff line number	Diff line change
`@@ -82,6 +82,7 @@`
`82`	`82`	`}`
`83`	`83`	`_SUCCESSFUL_FINISH_REASONS = [`
`84`	`84`	`gapic_content_types.Candidate.FinishReason.STOP,`
	`85`	`+ gapic_content_types.Candidate.FinishReason.MAX_TOKENS,`
`85`	`86`	`# Many responses have this finish reason`
`86`	`87`	`gapic_content_types.Candidate.FinishReason.FINISH_REASON_UNSPECIFIED,`
`87`	`88`	`]`