o -wi%@s dZddlmZddlmZmZddlmZmZm Z m Z m Z ddl Z ddl mZmZmZmZmZmZddgZGd d d eeZGd d d eeZGd ddeeZGdddeeZGdddeZGdddeZGdddeZGdddeZdedefddZdS)z9Definition of the config object used by the Launch agent.)Enum)ListOptional) BaseModelFieldValidationErrorroot_validator validatorN)AZURE_BLOB_REGEX"AZURE_CONTAINER_REGISTRY_URI_REGEX$ELASTIC_CONTAINER_REGISTRY_URI_REGEXGCP_ARTIFACT_REGISTRY_URI_REGEX GCS_URI_RE S3_URI_REr AgentConfigc@eZdZdZdZdZdZdS)EnvironmentTypez Enum of valid environment types.awsgcpazureN)__name__ __module__ __qualname____doc__rrrrrZ/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/launch/agent/config.pyr rc@r) RegistryTypezEnum of valid registry types.ecracrgcrN)rrrrrrr rrrrr'rrc@r) BuilderTypezEnum of valid builder types.dockerkanikonoopN)rrrrr"r#r$rrrrr!/rr!c@seZdZdZdZdZdS)TargetPlatformzEnum of valid target platforms.z linux/amd64z linux/arm64N)rrrr linux_amd64 linux_arm64rrrrr%7sr%c@s`eZdZUdZedddZeeed<edddZ ee ed<e de de de fd d Z dS) RegistryConfiga4Configuration for registry block. Note that we don't forbid extra fields here because: - We want to allow all fields supported by each registry - We will perform validation on the registry object itself later - Registry block is being deprecated in favor of destination field in builder NzThe type of registry to use. descriptiontypezThe URI of the registry.urireturncCst|S)Nvalidate_registry_uri)clsr,rrr validate_uriPszRegistryConfig.validate_uri)rrrrrr+rr__annotations__r,strr classmethodr1rrrrr(>s r(c@speZdZUdZedddZeeed<edddZ ee ed<Gd d d Z e d d e d edefddZdS)EnvironmentConfigz(Configuration for the environment block.NzThe type of environment to use.r)r+.zThe region to use.regionc@eZdZdZdS)zEnvironmentConfig.ConfigallowNrrrextrarrrrConfig_r;Tprevaluesr-cCs(|D]}|dvrtd|dq|S)z+Check for extra fields and print a warning.)r+r6zUnrecognized field z5 in environment block. Please check your config file.)wandbtermwarn)r0r?keyrrrcheck_extra_fieldsbs z$EnvironmentConfig.check_extra_fields)rrrrrr+rrr2r6r3r;rr4dictrCrrrrr5Vs r5c@sDeZdZUedddZeeed<edddZee ed<edddZ ee ed<edd d d Z ee ed <ed ddd Z ee ed<edddd Zee ed<edddd Zee ed<edddd Zee ed<ed ed ee dee fddZedded edefd!d"Zededee dee fd#d$ZdS)% BuilderConfigNzThe type of builder to use.r)r+zfThe destination to use for the built image. If not provided, the image will be pushed to the registry. destinationzfThe platform to use for the built image. If not provided, the platform will be detected automatically.platformz;The build context store to use. Required for kaniko builds.zbuild-context-store)r*aliasbuild_context_storezwandb-launch-container-buildzName prefix of the build job.zbuild-job-namebuild_job_namez0The name of the secret to use for the build job.z secret-name secret_namez/The key of the secret to use for the build job.z secret-key secret_keyz%gcr.io/kaniko-project/executor:latestz)The image to use for the kaniko executor.z kaniko-image kaniko_imager-cCs6|durdStttfD] }||r|Sq td)zHValidate that the build context store is a valid container registry URI.NzkInvalid build context store. Build context store must be a URI for an S3 bucket, GCS bucket, or Azure blob.)rrr match ValueError)r0rIregexrrrvalidate_build_context_stores z*BuilderConfig.validate_build_context_storeTr=r?cCs|S)z9Right now there are no required fields for docker builds.r)r0r?rrrvalidate_dockerszBuilderConfig.validate_dockercCs|durdSt|S)z@Validate that the destination is a valid container registry URI.Nr.)r0rFrrrvalidate_destinationsz"BuilderConfig.validate_destination)rrrrr+rr!r2rFr3rGr%rIrJrKrLrMr r4rQrrDrRrSrrrrrEnsf "rEc@seZdZUdZegddZeeed<eddZ e eed<edd dZ e e ed <edd dZ e e ed <ed ddZe eed<edddZe eed<edddZe eed<edddZe eed<edddZe e ed<edddZe e ed<GdddZdS)rz#Configuration for the Launch agent.z!The queues to use for this agent.)defaultr*queuesz%The W&B entity to use for this agent.r)entityz/The maximum number of jobs to run concurrently.max_jobsz;The maximum number of sweep schedulers to run concurrently.max_schedulersFzWhether to use secure mode for this agent. If True, the agent will reject runs that attempt to override the entrypoint or image. secure_modeNzThe registry to use.registryzThe environment to use. environmentzThe builder to use.builderrz@How verbose to print, 0 = default, 1 = verbose, 2 = very verbose verbosity<z[How many seconds to wait after receiving the stop command before forcibly cancelling a run.stopped_run_timeoutc@r7)zAgentConfig.ConfigforbidNr9rrrrr;r<r;)rrrrrrUrr3r2rVrrXintrYrZboolr[r(r\r5r]rEr^r`r;rrrrrsT r,r-cCsd}|dr |dd}t|}|r/|dr$t|j|dd|ds-td|St|}|rQ|drFt|j|dd|d sOtd |St|}|rs|drht|j|dd|d sqtd |St d |d |S)avValidate that the registry URI is a valid container registry URI. The URI should resolve to an image name in a container registry. The recognized formats are for ECR, ACR, and GCP Artifact Registry. If the URI does not match any of these formats, a warning is printed indicating the registry type is not recognized and the agent can't guarantee that images can be pushed. If the format is recognized but does not resolve to an image name, an error is raised. For example, if the URI is an ECR URI but does not include an image name or includes a tag as well as an image name, an error is raised. zDestination for built images may not include a tag, but the URI provided includes the suffix '{tag}'. Please remove the tag and try again. The agent will automatically tag each image with a unique hash of the source code.zhttps://Ntag)re image_namezAn image name must be specified in the URI for a GCP Artifact Registry. Please provide a uri with the format 'https://-docker.pkg.dev///'. repositoryzA repository name must be specified in the URI for an Azure Container Registry. Please provide a uri with the format 'https://.azurecr.io/'.zA repository name must be specified in the URI for an Elastic Container Registry. Please provide a uri with the format 'https://.dkr.ecr..amazonaws.com/'.z)Unable to recognize registry type in URI zN. You are responsible for ensuring the agent can push images to this registry.) startswithr rNgrouprOformatr r r@rA)r,tag_msgrNrrrr/sF             r/)renumrtypingrrpydanticrrrrr r@wandb.sdk.launch.utilsr r r r rr__all__r3rrr!r%r(r5rErr/rrrrs$   N0