o -wi@sddlmZddlmZddlmZmZmZmZm Z m Z m Z ddl m Z mZer.ddlmZddlmZGdddeeZ dd e eefd d Zd ededfddZd eddefddZdddZedddddedefddZd S) )Enum) lru_cache) TYPE_CHECKINGAnyListLiteralMappingOptionalSequence)REGISTRY_PREFIXvalidate_artifact_types_list)Client)gqlc@s*eZdZdZdZededefddZdS) VisibilityPRIVATE RESTRICTEDvaluereturncstfdd|DdS)Nc3s|] }|jkr|VqdSNname.0er`/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/apis/public/registries/_utils.py sz'Visibility._missing_..)next)clsrrrr _missing_szVisibility._missing_N) __name__ __module__ __qualname__ organization restricted classmethodobjectrr rrrrrs rNartifact_typescCs"|durgSt|}dd|DS)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. NcSsg|]}d|iqSrr)rtyperrr ,sz3format_gql_artifact_types_input..)r )r( new_typesrrrformat_gql_artifact_types_inputs r, visibilityr)r$r%cCs,zt|jWStytd|dw)zConvert the GQL visibility to the registry visibility. Args: visibility: The GQL visibility. Returns: The registry visibility. Invalid visibility: z from backend)rr ValueErrorr-rrrgql_to_registry_visibility/s   r1c CsFzt|jWSty"td|ddttddtDw)z6Convert the registry visibility to the GQL visibility.r.z. Must be one of: z, css|]}|jVqdSrrrrrrrIsz-registry_visibility_to_gql..)rrKeyErrorr/joinmapreprr0rrrregistry_visibility_to_gql@s  r6Fcst|}trr|tr|St|S|St|}trJi}|D]"\}}|dkr6t|dd||<q%|dkr?|||<q%t|d||<q%|St|}tr\tt fdd|S|S)zTraverse the filter to prepend the `name` key value with the registry prefix unless the value is a regex. - in_name: True if we are under a "name" key (or propagating from one). EX: {"name": "model"} -> {"name": "wandb-registry-model"} rTin_namez$regexcs t|dS)Nr7)ensure_registry_prefix_on_names)xr7rrfs z1ensure_registry_prefix_on_names..) isinstancestr startswithr ritemsr9r listr4)queryr8txtdctnew_dictkeyobjobjsrr7rr9Ms$ r9 )maxsizeclientr r$c Cstd}z |j|d|id}Wnty#}ztd||d}~ww|d}r6|d}r6|d}s>td|d |S) zFetch the org entity from the organization. Args: client (Client): Graphql client. organization (str): The organization to fetch the org entity for. z query FetchOrgEntityFromOrganization($organization: String!) { organization(name: $organization) { orgEntity { name } } } r$)variable_valuesz,Error fetching org entity for organization: N orgEntityrzOrganization entity for z not found.)rexecute Exceptionr/)rJr$rAresponserorg org_entityorg_namerrr"fetch_org_entity_from_organizationks*    rSr)F)enumr functoolsrtypingrrrrrr r wandb.sdk.artifacts._validatorsr r wandb_gqlr rr=rr,r1r6r9rSrrrrs2 $