googleapis
diff --git a/‎vertexai/language_models/_language_models.py‎
Lines changed: 1 addition & 0 deletions b/‎vertexai/language_models/_language_models.py‎
Lines changed: 1 addition & 0 deletions
@@ -3448,6 +3448,7 @@ class _PreviewCodeGenerationModel(CodeGenerationModel, _CountTokensCodeGeneratio
 _SUPPORTED_RLHF_MODELS = {
  "text-bison@001",
  "chat-bison@001",
+ "text-bison@002",
 }
Original file line number	Diff line number	Diff line change
`@@ -3448,6 +3448,7 @@ class _PreviewCodeGenerationModel(CodeGenerationModel, _CountTokensCodeGeneratio`
`3448`	`3448`	`_SUPPORTED_RLHF_MODELS = {`
`3449`	`3449`	`"text-bison@001",`
`3450`	`3450`	`"chat-bison@001",`
	`3451`	`+ "text-bison@002",`
`3451`	`3452`	`}`
`3452`	`3453`
`3453`	`3454`