o wi@sddlmZddlmZddlmZddlmZmZddl m Z m Z ddl m Z ddlmZe r6ddlmZGd d d eeZd!ddZd"d#ddZeddd$ddZd S)%) annotations) Collection)Enum) lru_cachepartial) TYPE_CHECKINGAny)gql) ensureprefix)RetryingClientc@s>eZdZdZdZedddZedd d Zedd dZdS) VisibilityPRIVATE RESTRICTEDvalueobjectreturnrcCs |j|SN) __members__get)clsrrW/home/ubuntu/.local/lib/python3.10/site-packages/wandb/apis/public/registries/_utils.py _missing_s zVisibility._missing_strcCDz||WSty!ddd|D}td|d|dw)z:Convert a GraphQL `visibility` value to a Visibility enum.,cs|]}t|jVqdSr)reprr.0errr !z&Visibility.from_gql..Invalid visibility z from backend. Expected one of: N ValueErrorjoin)rrexpectedrrrfrom_gql  zVisibility.from_gqlnamecCr)z3Convert a visibility string to a `Visibility` enum.rcsrr)rr*rrrrr!,r"z)Visibility.from_python..r#z. Expected one of: Nr$)rr*r'rrr from_python&r)zVisibility.from_pythonN)rrrr)rrrr )r*rrr ) __name__ __module__ __qualname__ organization restricted classmethodrr(r+rrrrr s   r artifact_typesCollection[str] | Nonerlist[dict[str, str]] | NonecCs&ddlm}|rdd||DSdS)zFormat the artifact types for the GQL input. Args: artifact_types: The artifact types to add to the registry. Returns: The artifact types for the GQL input. r)validate_artifact_typescSsg|]}d|iqS)r*r)rtyprrr @sz0prepare_artifact_types_input..N)wandb.sdk.artifacts._validatorsr5)r2r5rrrprepare_artifact_types_input2s r9Fqueryrin_nameboolc Csddlm}t|}tr|rt||S|St|}trHi}|D]"\}}|dkr0|||<q#|dkr=t|dd||<q#t||d||<q#|St|}tt fr\tt t t|d|S|S)zRecursively the registry prefix to values under "name" keys, excluding regex ops. - in_name: True if we are under a "name" key (or propagating from one). EX: {"name": "model"} -> {"name": "wandb-registry-model"} r)REGISTRY_PREFIXz$regexr*T)r;) r8r= isinstancerr dictitemsensure_registry_prefix_on_nameslisttuplemapr) r:r;r=txtdctnew_dictkeyobjseqrrrrADs  rA )maxsizeclientr r/rc Csddlm}m}t|}z |j|d|id}Wnty-}z d|}t||d}~ww||}|j} rB| j } rB| j } sJtd|d| S) zFetch the org entity from the organization. Args: client (Client): Graphql client. organization (str): The organization to fetch the org entity for. r)&FETCH_ORG_ENTITY_FROM_ORGANIZATION_GQLFetchOrgEntityFromOrganizationr/)variable_valuesz,Error fetching org entity for organization: NzOrganization entity for z not found.) wandb.sdk.artifacts._generatedrNrOr execute Exceptionr%model_validater/ org_entityr*) rMr/rNrOgql_opdatar msgresultorgrUorg_namerrr"fetch_org_entity_from_organization`s$    r\N)r2r3rr4)F)r:rr;r<rr)rMr r/rrr) __future__rcollections.abcrenumr functoolsrrtypingrr wandb_gqlr wandb._strutilsr wandb.apis.public.apir rr r9rAr\rrrrs      "