chore: LLM - Added private preview launch stage

Ark-kun · copybara-github · commit 970970e8c759 · 2023-06-29T13:16:54.000-07:00
PiperOrigin-RevId: 544442705
diff --git a/vertexai/_model_garden/_model_garden_models.py b/vertexai/_model_garden/_model_garden_models.py
@@ -34,6 +34,13 @@
     "text-bison": "https://us-kfp.pkg.dev/vertex-ai/large-language-model-pipelines/tune-large-model/sdk-1-25"
 }
 
+_SDK_PRIVATE_PREVIEW_LAUNCH_STAGE = frozenset(
+    [
+        gca_publisher_model.PublisherModel.LaunchStage.PRIVATE_PREVIEW,
+        gca_publisher_model.PublisherModel.LaunchStage.PUBLIC_PREVIEW,
+        gca_publisher_model.PublisherModel.LaunchStage.GA,
+    ]
+)
 _SDK_PUBLIC_PREVIEW_LAUNCH_STAGE = frozenset(
     [
         gca_publisher_model.PublisherModel.LaunchStage.PUBLIC_PREVIEW,

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,13 @@`
`34`	`34`	`"text-bison": "https://us-kfp.pkg.dev/vertex-ai/large-language-model-pipelines/tune-large-model/sdk-1-25"`
`35`	`35`	`}`
`36`	`36`
	`37`	`+_SDK_PRIVATE_PREVIEW_LAUNCH_STAGE = frozenset(`
	`38`	`+ [`
	`39`	`+ gca_publisher_model.PublisherModel.LaunchStage.PRIVATE_PREVIEW,`
	`40`	`+ gca_publisher_model.PublisherModel.LaunchStage.PUBLIC_PREVIEW,`
	`41`	`+ gca_publisher_model.PublisherModel.LaunchStage.GA,`
	`42`	`+ ]`
	`43`	`+)`
`37`	`44`	`_SDK_PUBLIC_PREVIEW_LAUNCH_STAGE = frozenset(`
`38`	`45`	`[`
`39`	`46`	`gca_publisher_model.PublisherModel.LaunchStage.PUBLIC_PREVIEW,`