dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZdd l mZdd l mZdd l mZdd l mZddl mZddl mZddlmZdZdZe j2e j4e j6j8e j6j:e j<Gdde j>Z y)z3Vertex AI deployment resource pools create command.)absolute_import)division)unicode_literals)encoding) operations)client)base) constants)deployment_resource_pools_util) endpoint_util)flags)operations_util) region_util) validation)logc tj}|jddtj||dtj j |tj|dtjy) zAdding deployment resource pool arguments from CLI. Args: parser: argparse.ArgumentParser, cli argument parser Returns: None deployment_resource_pool_idzThe ID to use for the DeploymentResourcePool, which will become the final component of the DeploymentResourcePool's resource name. The maximum length is 63 characters, and valid characters are /^[a-z]([a-z0-9-]{0,61}[a-z0-9])?$/.)helpT)drpz"to create deployment resource pool) prompt_funcN) r BETA_VERSION add_argumentr AddPredictionResourcesArgsGetAutoscalingMetricSpecsArg AddToParserAddRegionResourceArgr.PromptForDeploymentResourcePoolSupportedRegion)parserversions 2lib/surface/ai/deployment_resource_pools/create.py _AddArgsBetar!#ss  " "'# -. ""67=$$&226: *LLNctj|j|jtj }|j jj}|jd|_tj||j5tj|}|j||j|j |j"|j|j$|j&|j(|j*|j,|j.|j }t1j2t5j6|t9j:|j<}|Jt?j@|}d|vr1tBjDjGdjI|dddd|S#1swYSxYw)z0Create a new Vertex AI deployment resource pool. locationsId)region)r) autoscaling_metric_specsaccelerator_dictmin_replica_countmax_replica_count machine_type tpu_topologymultihost_gpu_node_countreservation_affinityspotrequired_replica_countNnamez/Created Vertex AI deployment resource pool: {}.)%rValidateRequiredReplicaCountr/r(r rCONCEPTSr%ParseAsDictr AiplatformEndpointOverridesrDeploymentResourcePoolsClient CreateBetarr& acceleratorr)r*r+r,r-r.rWaitForOpMayberOperationsClientr ParseOperationr0rMessageToPyValuerstatusPrintformat)argsr region_ref deployment_resource_pools_clientop response_msgresponses r _RunBetarF;s ))$*E*E*.*@*@B  " "'}}##))+*!!#M2$+00M'-'K'K($ ) 4 4 ((!%!>!>))0000&&&&!%!>!>!66 YY#:: 5 B#11##%r&55bgg>@L**<8h 8   > VHV$ % '/N4 5N4 s D4GG c&eZdZdZedZdZy) CreateV1Beta1aCreate a new Vertex AI deployment resource pool. This command creates a new deployment resource pool with a provided deployment resource pool id (name) in a provided region (assuming that region provides support for this api). You can choose to simply provide the resource pool name and create an instance with default arguments, or you can pass in your own preferences, such as the accelerator and machine specs. Please note this functionality is not yet available in the GA track and is currently only part of the v1beta1 API version. ## EXAMPLES To create a deployment resource pool with name ``example'' in region ``us-central1'', run: $ {command} example --region=us-central1 ct|SN)r!)rs r ArgszCreateV1Beta1.Argsus  r"ct|SrJ)rF)selfr@s r RunzCreateV1Beta1.Runys D>r"N)__name__ __module__ __qualname____doc__ staticmethodrKrNr"r rHrH_s $  r"rHN)!rR __future__rrrapitools.base.pyrgooglecloudsdk.api_lib.air3googlecloudsdk.api_lib.ai.deployment_resource_poolsrgooglecloudsdk.callioper googlecloudsdk.command_lib.air r r r rrrgooglecloudsdk.corerr!rFHidden ReleaseTracks ReleaseTrackALPHABETAUniverseCompatible CreateCommandrHrTr"r rcs:&'%0F(3H7/954#N0!HD%%++T->->-C-CDD&&E r"