o -wiP)@s@UdZddlmZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z mZmZmZmZmZmZmZddlmZddlmZdd lmZdd lmZdd lmZd d lm Z m!Z!d dl"m#Z#m$Z$e rddl m%Z%m&Z&ddl'm(Z(ede(dZ)edZ*dZ+de,d<dZ-de,d<dZ.de,d<e/dhZ0de,d<ee Z1de,d<ee!Z2de,d <eGd!d"d"Z3d]d&d'Z4d(Z5d^d*d+Z6d_d/d0Z7d`d2d3Z8e9d4Z:d5e,d6<dad8d9Z;d:Z<d?d@dAZ>dBe,dC<dbdEdFZ?dcdJdKZ@dddNdOZAedPedQefdZB dedSdTZCdedUdVZDdfdYdZZEeGd[d\d\ZFdS)gz=Internal validation utilities that are specific to artifacts.) annotationsN)astuple dataclassfieldreplace)wraps) TYPE_CHECKINGAnyCallableDictLiteralOptionalTypeVarcast)r)Self) always_list) gql_typename)json_friendly_val)ArtifactPortfolioTypeFieldsArtifactSequenceTypeFields)ArtifactFinalizedErrorArtifactNotLoggedError) CollectionFinal)Artifact ArtifactT)boundTzwandb-registry-z Final[str]REGISTRY_PREFIXdz Final[int]MAX_ARTIFACT_METADATA_KEYSARTIFACT_NAME_MAXLEN/zFinal[frozenset[str]]ARTIFACT_NAME_INVALID_CHARSLINKED_ARTIFACT_COLLECTION_TYPESOURCE_ARTIFACT_COLLECTION_TYPEc@s~eZdZUdZded<ded<ded<ded<ded<ed d d Zd ed <ed edZded<e dddZ e dddZ dS)_LinkArtifactFieldsz\Keep this list updated with fields where the linked artifact and the source artifact differ.str entity_name project_namenameversion list[str]aliasesFT)initdefaultz Literal[True]_is_link)r0default_factorylist[Artifact]_linked_artifactsreturnboolcC|jSN)r2selfr<\/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/artifacts/_validators.pyis_link8z_LinkArtifactFields.is_linkcCr8r9)r5r:r<r<r=linked_artifacts<r?z$_LinkArtifactFields.linked_artifactsN)r6r7)r6r4) __name__ __module__ __qualname____doc____annotations__rr2listr5propertyr>r@r<r<r<r=r((s  r(r,r)r6cCs^t|tkr|dtd}tdtd|t|r-tddttd||S)z~Validate the artifact name, returning it if successful. Raises: ValueError: If the artifact name is invalid. Nz ...zArtifact name is longer than z characters: zAArtifact names must not contain any of the following characters: , z. Got: )lenr# ValueErrorr% intersectionjoinsorted)r, short_namer<r<r=validate_artifact_nameAs   rO)r$\#?%:  NonecCsd}t|dkr tdd}|tr%|ttd}t|dkr%tdt||krG|rs z#validate_aliases..z:Aliases must not contain any of the following characters: rHranyrJrL)r/ aliases_listr<rir=validate_aliasessrnartifact_typescs<t|}dtfdd|Drtddd|S)zAReturn True if the artifact types list is valid, False otherwise.rec3s.|]}D] }||vpt|dkVqqdS)r"N)rI)rftyperhrir<r=rjs z/validate_artifact_types_list..zAArtifact types must not contain any of the following characters: rHz< and must be less than equal to 128 charactersrk)ror<rir=validate_artifact_types_lists rqz^[-\w]+( +[-\w]+)*$zre.Pattern[str]_VALID_TAG_PATTERNtagscCs0t|}tdd|Drtdtt|S)aValidate the artifact tag names and return them as a deduped list. In the case of duplicates, only keep the first tag, and otherwise maintain the order of appearance. Raises: ValueError: If any of the tags contain invalid characters. css|] }t| VqdSr9)rrmatch)rftagr<r<r=rjsz validate_tags..zqInvalid tag(s). Tags must only contain alphanumeric characters separated by hyphens, underscores, and/or spaces.)rrlrJrFdictfromkeys)rs tags_listr<r<r= validate_tagss ryzwandb-RESERVED_ARTIFACT_TYPE_PREFIX)job run_tablecodezFinal[tuple[str, ...]]RESERVED_ARTIFACT_TYPESzrun-zsource-)r|r}zFinal[dict[str, str]]RESERVED_ARTIFACT_NAME_PREFIXEStypcCsR|tvrt|}|r||rtd|d|S|tr'td|d|S)z5Validate the artifact type and return it as a string.zArtifact type z; is reserved for internal use. Please use a different type.)r~rgetrYrJrz)rr,reserved_name_prefixr<r<r=validate_artifact_types    rmetadatadict[str, Any] | Nonedict[str, Any]cCsJ|duriSt|tstdt|ttttft t t |S)z7Validate the artifact metadata and return it as a dict.Nzmetadata must be dict, not ) isinstancerv TypeErrorrprr r)r jsonloadsdumpsr)rr<r<r=validate_metadatas  "rgql_ttl_duration_seconds int | NonecCs|r|dkr|SdS)zIValidate the `ttlDurationSeconds` value (if any) from a GraphQL response.rNr<)rr<r<r=validate_ttl_duration_secondss r DecoratedF.methodc(jtd fdd }tt|S) zDecorator to ensure that an Artifact method can only be called if the artifact has been logged. If the method is called on an artifact that's not logged, `ArtifactNotLoggedError` is raised. r;rargsr kwargsr6cs*|r t|d|g|Ri|SN)fullnameobj)is_draftrr;rrrmethod_fullnamer<r=wrappers zensure_logged..wrapperNr;rrr rr r6r rCrrrrrr<rr= ensure_logged rcr) zDecorator to ensure that an `Artifact` method can only be called if the artifact isn't finalized. If the method is called on an artifact that's not logged, `ArtifactFinalizedError` is raised. r;rrr rr6cs(|jr t|d|g|Ri|Sr)_finalrrrr<r=rs z%ensure_not_finalized..wrapperNrrrr<rr=ensure_not_finalizedrrprojectr7cCs |tSr9)rYr)rr<r<r=is_artifact_registry_projects rc@sVeZdZUded<dZded<dZded<edd d Zdd d Z  ddddZ dS) ArtifactPathr)r,Nz Optional[str]rprefixpathr6rcCs0t|d}dkr|t|Std|)z(Instantiate by parsing an artifact path.r$zSExpected a valid path like `name`, `project/name`, or `prefix/project/name`. Got: )rIsplitreversedrJ)clsrpartsr<r<r=from_strs  zArtifactPath.from_strcCsdtttt|S)z>Returns the slash-separated string representation of the path.r$)rLfilterr7rrr:r<r<r=to_strszArtifactPath.to_str str | NonecCst||jp||jp |dS)z@Returns this path with missing values set to the given defaults.)rr)rrr)r;rrr<r<r= with_defaults!s zArtifactPath.with_defaults)rr)r6r)r6r))NN)rrrrr6r) rArBrCrErr classmethodrrrr<r<r<r=r s     r)r,r)r6r))r,r)r6rW)r/rdr6r.)ror.r6r.)rsrdr6r.)rr)r,r)r6r))rrr6r)rrr6r)rrr6r)rr)r6r7)GrD __future__rrre dataclassesrrrr functoolsrtypingrr r r r r rrpydantic.dataclassespydantic_dataclasstyping_extensionsrwandb._iterutilsrwandb._pydanticr wandb.utilr _generatedrr exceptionsrrrrwandb.sdk.artifacts.artifactrrrrrEr!r# frozensetr%r&r'r(rOr[rcrnrqcompilerrryrzr~rrrrrrrrrr<r<r<r=sb  (             '