}%dZddlmZddlmZddlmZddlmZddlmZdZ ejfdZ d Z d Z d Z dd Z ddZddZdZy )z$Utilities for validating parameters.)absolute_import)division)unicode_literals) exceptions) constantsc:||stjddyy)zValidates the display name.Nz--display-namezDisplay name can not be empty.rInvalidArgumentException) display_names /lib/googlecloudsdk/command_lib/ai/validation.pyValidateDisplayNamer s+l  - -: '3cv||vr5tjddjdj||y)z=Validates whether a given region is among the available ones.regionz*Available values are [{}], but found [{}]., N)rr formatjoin)ravailable_regionss r ValidateRegionr!sC $$  - -4;; II' (&  %rct|jdra|jjj}|r|j SdD]%}t ||dst jddyy)zFParse CMEK resource arg, and check if the arg was partially specified.kms_key)r kms_keyring kms_location kms_projectNz --kms-keyz#Encryption key not fully specified.)hasattrCONCEPTSrParse RelativeNamegetattrrr )argskms_refkeywords r GetAndValidateKmsKeyr#,so T]]I&mm##))+G  ! ! ##N' 4$ '33@  O 'rc|y|jD]\}}|tjvrvtjddj dj ttjjDcgc]}dj |c}|dk(r|dkstjdd|dks|d kDstjdd j ||ycc}w) zDValue validation for autoscaling metric specs target name and value.N--autoscaling-metric-specsz=Autoscaling metric name can only be one of the following: {}.rz'{}'zrequest-counts-per-minuterzEMetric target for request-counts-per-minute must be a positive value.dz7Metric target value {} for {} is not between 0 and 100.) itemsr!OP_AUTOSCALING_METRIC_NAME_MAPPERrr rrsortedkeys)specskeyvaluecs r ValidateAutoscalingMetricSpecsr/:s ] KKMjc5 )===  / / & M T Tii!AAFFHa--"    )) !11 (   !us{  / / & C J JS  -" s;C6cL|"|xsd}||kDrtjddyy)z,Value validation for required replica count.N--required-replica-countz6Value must be less than or equal to min-replica-count.r )required_replica_countmin_replica_counts r ValidateRequiredReplicaCountr5]s?').Q 11  / / $ B 2(rNc~|P|tjdd|dkDrtjdd|||krtjdd|3|tjdd|dkDrtjdd |4|tjd d |dkDrtjd d yy) z(Value validation for scale-to-zero args.Nz--initial-replica-countzuCannot set initial-replica-count without explicitly setting min-replica-count to 0 to enable scale-to-zero.rzkCannot set initial-replica-count when min-replica-count > 0 as scale-to-zero will not be enabled.z=Initial-replica-count must be smaller than max replica count.z--min-scaleup-periodzrCannot set min-scaleup-period without explicitly setting min-replica-count to 0 to enable scale-to-zero.zhCannot set min-scaleup-period when min-replica-count > 0 as scale-to-zero will not be enabled.z--idle-scaledown-periodzuCannot set idle-scaledown-period without explicitly setting min-replica-count to 0 to enable scale-to-zero.zkCannot set idle-scaledown-period when min-replica-count > 0 as scale-to-zero will not be enabled.r )r4initial_replica_countmax_replica_countmin_scaleup_periodidle_scaledown_periods r ValidateScaleToZeroArgsr;hs &  / / # =  1  / / # 0  %  5 5  / / # M  #  / /  =  1  / /  0  &  / / # =  1  / / # 0 'rc*|y|tjdd|tjdd|tjdd|tjdd |tjd d |tjd d y)aValue validation for dedicated resource args while making a shared resource command call. Args: shared_resources_ref: str or None, the shared deployment resource pool full name the model should use, formatted as the full URI machine_type: str or None, the type of the machine to serve the model. accelerator_dict: dict or None, the accelerator attached to the deployed model from args. min_replica_count: int or None, the minimum number of replicas the deployed model will be always deployed on. max_replica_count: int or None, the maximum number of replicas the deployed model may be deployed on. required_replica_count: int or None, the required number of replicas the deployed model will be considered successfully deployed. autoscaling_metric_specs: dict or None, the metric specification that defines the target resource utilization for calculating the desired replica count. Nz--machine-typezECannot use machine type and shared resources in the same command.z --acceleratorzDCannot use accelerator and shared resources in the same command.z--min-replica-countzJCannot use min replica count and shared resources in the same command.z--max-replica-countzJCannot use max replica count and shared resources in the same command.r2zOCannot use required replica count and shared resources in the same command.r%zUCannot use autoscaling metric specs and shared resources in the same command.r )shared_resources_ref machine_typeaccelerator_dictr4r8r3autoscaling_metric_specss r ValidateSharedResourceArgsrAs6!   - - >  !  - - A  "  - - G  "  - - G  '  - -" L  )  - -$ 5 *rc:||rtjddyy)z2Validates the network and public_endpoint_enabled.NzRPlease either set --network for private endpoint, or set --public-endpoint-enabledzfor public enpdoint.r )networkpublic_endpoint_enableds r ValidateEndpointArgsrEs. 4  - - % 5rcL|j"|jstjddt|jj ddk7s9t|jj ddj ddkDrtjddy) z&Validates the model garden model args.Nz--modelzModel name should not be empty./r1@a4Model name should be in the format of Model Garden, e.g. `{publisher_name}/{model_name}@{model_version_name}, e.g. `google/gemma2@gemma-2-2b` or in the format of Hugging Face convention, e.g. `meta-llama/Meta-Llama-3-8B`. You can use the `gcloud ai model-garden models list` command to find supported models.)modelrr lensplit)r s r ValidateModelGardenModelArgsrMs ZZDJJ  - -)   $**  3  A% TZZ  c "1 % + +C 0 1A 5  - -   6r)NNNNN)NNNNNNN)NN)__doc__ __future__rrrgooglecloudsdk.calliopergooglecloudsdk.command_lib.airr SUPPORTED_REGIONrr#r/r5r;rArErMrr rTs}+&'.3.7-G-G   F 9z!AHr