o xiK@sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZe rLdd lmZdd lmZdd lmZed dd0ddZed dd1ddZed dd2ddZd3d!d"Zd4d$d%Zed&d'Gd(d)d)Z *d5d6d.d/Zd*S)7) annotations)suppress) dataclass) lru_cache) TYPE_CHECKING)gql)one) ServerFeature)SERVER_FEATURES_QUERY_GQLServerFeaturesQuery)RetryingClient)TypeInfoFragment)FetchOrgInfoFromEntityEntity)maxsizeclientr typenamestrreturnTypeInfoFragment | NonecCs2ddlm}m}|jt|d|id}||jS)z/Returns the type info for a given GraphQL type.) TYPE_INFO_GQLTypeInfonamevariable_values) _generatedrrexecutermodel_validatetype)rrrrdatar!Q/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/artifacts/_gqlutils.py type_infos r#entity#FetchOrgInfoFromEntityEntity | NonecCs6ddlm}m}t|}|j|d|id}||jS)z1Returns the organization info for a given entity.r)FETCH_ORG_INFO_FROM_ENTITY_GQLFetchOrgInfoFromEntityr$r)rr&r'rrrr$)rr$r&r'gql_opr r!r!r"org_info_from_entitys r)dict[str, bool]c Cszz |tt}Wnty$}zdt|vriWYd}~Sd}~wwt|}|j}r;|j}r;dd|DSiS)zzReturns a mapping of `{server_feature_name (str) -> is_enabled (bool)}`. Results are cached per client instance. z3Cannot query field "features" on type "ServerInfo".NcSsi|] }|r|j|jqSr!)r is_enabled).0featr!r!r" <sz#server_features..) rrr Exceptionrr r server_infofeatures)rresponseeresultr0r1r!r!r"server_features+s  r5feature str | intboolcCsDzt|tr t|n|}Wn tyYdSwt||p!dS)zReturn whether the current server supports the given feature. Good to use for features that have a fallback mechanism for older servers. F) isinstanceintr Name ValueErrorr5get)rr6rr!r!r"server_supports@s   r>set[str]cCs*t||}|r|jrdd|jDStS)z9Returns the allowed field names for a given GraphQL type.cSsh|]}|jqSr!)r)r,fr!r!r" Tsz!allowed_fields..)r#fieldsset)rrtypr!r!r"allowed_fieldsQs  rET)frozenc@s(eZdZUded<ded<d ddZd S) OrgInfororg_name entity_nameotherrr8cCs||j|jhvSNrHrI)selfrJr!r!r" __contains__\szOrgInfo.__contains__N)rJrrr8)__name__ __module__ __qualname____annotations__rNr!r!r!r"rGWs rGNnon_org_entity str | None org_or_entityc s4|stdt||}|r(|j}r(|jr(t|jjd}r%|vr(jS|r|j}r|j}rfdd|D}rztt t fdd|DWdS1sYwYt |dkrrtd|d j d d td d t dd|Dtd|dtd|ddStd|d)Nz3Entity name is required to resolve org entity name.rLcs&g|]}|jrt|jjdqS)rL) org_entityrGrr,org)rVr!r" s z+resolve_org_entity_name..c3s|] }|vr|jVqdSrKrI)r,info)rUr!r" sz*resolve_org_entity_name..rz8Expecting the organization name or entity name to match rz# and cannot be linked/fetched with zC. Please update the target path with the correct organization name.zTPersonal entity belongs to multiple organizations and cannot be linked/fetched with zo. Please update the target path with the correct organization name or use a team entity in the entity settings.css|]}|jVqdSrKrZrWr!r!r"r\szCUnable to resolve an organization associated with personal entity: z. This could be because its a personal entity that doesn't belong to any organizations. Please specify the organization in the Registry path or use a team entity in the entity settings.zPersonal entity z belongs to multiple organizations and cannot be used without specifying the organization name. Please specify the organization in the Registry path or use a team entity in the entity settings.) too_shorttoo_longz'Unable to find organization for entity .)r<r) organizationrVrGruser organizationsr StopIterationnextlenrHr) rrSrUr$rXorg_inforaorgs org_infosr!)rVrUr"resolve_org_entity_name`sL         ri)rr rrrr)rr r$rrr%)rr rr*)rr r6r7rr8)rr rrrr?rK)rr rSrTrUrTrr) __future__r contextlibr dataclassesr functoolsrtypingr wandb_gqlrwandb._iterutilsrwandb.proto.wandb_internal_pb2r wandb.sdk.internal._generatedr r wandb.apis.publicr wandb.sdk.artifacts._generatedr 9wandb.sdk.artifacts._generated.fetch_org_info_from_entityrr#r)r5r>rErGrir!r!r!r"s2