o
    xi*                     @  s  d dl mZ d dlmZmZmZ d dlmZ d dlm	Z	m
Z
mZ ddlmZ G dd de
ZG d	d
 d
e
ZG dd de
ZG dd de
ZG dd de
ZG dd de
ZG dd de
ZG dd de
ZG dd de
ZG dd de
ZG dd de
ZG dd de
ZG dd  d eZG d!d" d"e
ZG d#d$ d$e
ZG d%d& d&e
ZG d'd( d(e
ZG d)d* d*e
ZG d+d, d,e
Z G d-d. d.e
Z!G d/d0 d0e
Z"G d1d2 d2e
Z#G d3d4 d4e
Z$G d5d6 d6e
Z%G d7d8 d8e
Z&G d9d: d:e
Z'G d;d< d<e
Z(G d=d> d>e
Z)G d?d@ d@e
Z*G dAdB dBe
Z+G dCdD dDe
Z,G dEdF dFe
Z-G dGdH dHe
Z.G dIdJ dJe
Z/G dKdL dLe
Z0G dMdN dNe
Z1G dOdP dPe
Z2G dQdR dRe
Z3G dSdT dTe
Z4G dUdV dVe
Z5G dWdX dXe
Z6G dYdZ dZe
Z7G d[d\ d\e
Z8G d]d^ d^e
Z9e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e:  e :  e!:  e":  e#:  e$:  e%:  e&:  e:  e':  e(:  e):  e:  e*:  e+:  e,:  e-:  e.:  e/:  e0:  e1:  e:  e2:  e3:  e2:  e4:  e3:  e0:  e5:  e6:  e7:  e8:  e9:  e0:  d_S )`    )annotations)ListLiteralOptional)Field)GQLId	GQLResultTypename   )ArtifactStatec                   @  *   e Zd ZU dZded< ded< ded< dS )	ArtifactAliasFragmentArtifactAliasz"Typename[Literal['ArtifactAlias']]
typename__r   idstraliasN__name__
__module____qualname__r   __annotations__ r   r   \/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/artifacts/_generated/fragments.pyr         
 r   c                   @     e Zd ZU ded< ded< dS )ProjectInfoFragmentr   nameProjectInfoFragmentEntityentityNr   r   r   r   r   r   r   r   r         
 r   c                   @     e Zd ZU ded< dS )r   r   r   Nr    r   r   r   r   r         
 r   c                   @  r   )	TagFragmentTagzTypename[Literal['Tag']]r   r   r   r   r   Nr   r   r   r   r   r$      r   r$   c                   @  j   e Zd ZU ded< ded< ded< ded< ed	d
Zded< edd
Zded< ded< ded< ded< dS )ArtifactCollectionFragment:Typename[Literal['ArtifactPortfolio', 'ArtifactSequence']]r   r   r   r   r   Optional[str]description	createdAtr   
created_at	updatedAt
updated_atOptional[ProjectInfoFragment]projectArtifactCollectionFragmentTypetypeArtifactCollectionFragmentTagstagsNr   r   r   r   r   r-   r/   r   r   r   r   r'   $      
 r'   c                   @  r"   )r2   r   r   Nr    r   r   r   r   r2   0   r#   r2   c                   @  r"   )r4   z)List[ArtifactCollectionFragmentTagsEdges]edgesNr    r   r   r   r   r4   4   r#   r4   c                   @  r"   )#ArtifactCollectionFragmentTagsEdgesr$   nodeNr    r   r   r   r   r9   8   r#   r9   c                   @  s&   e Zd ZU ded< ded< ded< dS )CollectionInfoFragmentr(   r   r   r   r0   r1   Nr    r   r   r   r   r;   <   s   
 r;   c                   @  r   )	SourceCollectionInfoFragmentArtifactSequence%Typename[Literal['ArtifactSequence']]r   r   r   r0   r1   Nr   r   r   r   r   r<   B   r   r<   c                   @  s  e Zd ZU dZded< ded< eddZded	< ed
dZded< eddZded< ded< ded< eddZ	ded< eddZ
ded< ded< eddZded< ded< ded< d ed!< ed"dZded#< ed$dZded%< ed&dZd ed'< ed(dZded)< d*Zd+ed,< d*S )-ArtifactFragmentArtifactzTypename[Literal['Artifact']]r   r   r   artifactSequencer,   r<   artifact_sequenceversionIndexOptional[int]version_indexartifactTypeArtifactFragmentArtifactTypeartifact_typer)   r*   metadatattlDurationSecondsintttl_duration_secondsttlIsInheritedboolttl_is_inheritedzList[TagFragment]r5   historyStephistory_stepr   statesizer   digest
commitHashcommit_hash	fileCount
file_countr+   r-   r.   r/   Nz'Optional[List[ArtifactFragmentAliases]]aliases)r   r   r   r   r   r   rB   rE   rH   rL   rO   rQ   rV   rX   r-   r/   rY   r   r   r   r   r?   H   s(   
 r?   c                   @  r"   )rG   r   r   Nr    r   r   r   r   rG   ^   r#   rG   c                   @      e Zd ZU eddZded< dS )ArtifactFragmentAliasesartifactCollectionr,    Optional[CollectionInfoFragment]artifact_collectionN)r   r   r   r   r^   r   r   r   r   r   r[   b   s   
 r[   c                   @  sV   e Zd ZU dZded< ded< eddZded	< d
ed< eddZded< ded< dS )ArtifactMembershipFragmentArtifactCollectionMembershipz1Typename[Literal['ArtifactCollectionMembership']]r   r   r   rC   r,   rD   rE   zList[ArtifactAliasFragment]rY   r\   r]   r^   zOptional[ArtifactFragment]artifactN)r   r   r   r   r   r   rE   r^   r   r   r   r   r_   h   s   
 
r_   c                   @  r   )	ArtifactPortfolioTypeFieldsArtifactPortfolioz&Typename[Literal['ArtifactPortfolio']]r   r   r   r   r   Nr   r   r   r   r   rb   u   r   rb   c                   @  r   )	ArtifactSequenceTypeFieldsr=   r>   r   r   r   r   r   Nr   r   r   r   r   rd   {   r   rd   c                   @  sD   e Zd ZU dZded< ded< ded< ded	< ed
dZded< dS )ArtifactTypeFragmentArtifactTypez!Typename[Literal['ArtifactType']]r   r   r   r   r   r)   r*   r+   r,   r-   N)r   r   r   r   r   r   r-   r   r   r   r   re      s   
 re   c                   @  r"   )DeferredManifestFragmentDeferredManifestFragmentFilefileNr    r   r   r   r   rg      r#   rg   c                   @  rZ   )rh   	directUrlr,   r   
direct_urlN)r   r   r   r   rk   r   r   r   r   r   rh      s   
 rh   c                   @  s   e Zd ZU dZded< ded< ded< ded	< ed
dZded< eddZded< ded< eddZded< ded< eddZ	ded< eddZ
ded< dS )FileFragmentFileTypename[Literal['File']]r   r   r   r   r   r)   url	sizeBytesr,   rK   
size_bytesstoragePathstorage_pathmimetyper.   r/   rT   md5md_5rj   rk   N)r   r   r   r   r   r   rq   rs   r/   rv   rk   r   r   r   r   rl      s   
 rl   c                   @  s4   e Zd ZU dZded< ded< eddZded< d	S )
FileWithUrlFragmentrm   rn   r   r   r   rj   r,   rk   N)r   r   r   r   r   r   rk   r   r   r   r   rw      s   
 rw   c                   @  s(   e Zd ZU ded< eddZded< dS )OrgInfoFragmentr   r   	orgEntityr,   z"Optional[OrgInfoFragmentOrgEntity]
org_entityN)r   r   r   r   r   rz   r   r   r   r   rx      s   
 rx   c                   @  r"   )OrgInfoFragmentOrgEntityr   r   Nr    r   r   r   r   r{      r#   r{   c                   @  s>   e Zd ZU dZded< eddZded< eddZd	ed
< dS )PageInfoFragmentPageInfozTypename[Literal['PageInfo']]r   	endCursorr,   r)   
end_cursorhasNextPagerN   has_next_pageN)r   r   r   r   r   r   r   r   r   r   r   r   r|      s   
 r|   c                   @  r&   )RegistryCollectionFragmentr(   r   r   r   r   r   r)   r*   r+   r,   r-   r.   r/   r0   r1   RegistryCollectionFragmentTyper3   RegistryCollectionFragmentTagsr5   Nr6   r   r   r   r   r      r7   r   c                   @  r"   )r   r   r   Nr    r   r   r   r   r      r#   r   c                   @  r"   )r   z)List[RegistryCollectionFragmentTagsEdges]r8   Nr    r   r   r   r   r      r#   r   c                   @  r"   )#RegistryCollectionFragmentTagsEdgesr$   r:   Nr    r   r   r   r   r      r#   r   c                   @  s   e Zd ZU dZded< ded< ded< ded	< d
ed< eddZded< eddZd
ed< d
ed< eddZded< eddZ	ded< dS )RegistryFragmentProjectzTypename[Literal['Project']]r   r   r   r   r   RegistryFragmentEntityr   r)   r*   r+   r,   r-   r.   r/   accessallowAllArtifactTypesrN   allow_all_artifact_typesartifactTypesRegistryFragmentArtifactTypesartifact_typesN)
r   r   r   r   r   r   r-   r/   r   r   r   r   r   r   r      s   
 r   c                   @  r   )r   r   r   z,Optional[RegistryFragmentEntityOrganization]organizationNr    r   r   r   r   r      r!   r   c                   @  r"   )"RegistryFragmentEntityOrganizationr   r   Nr    r   r   r   r   r      r#   r   c                   @  r"   )r   z(List[RegistryFragmentArtifactTypesEdges]r8   Nr    r   r   r   r   r      r#   r   c                   @  r"   )"RegistryFragmentArtifactTypesEdgesz0Optional[RegistryFragmentArtifactTypesEdgesNode]r:   Nr    r   r   r   r   r      r#   r   c                   @  r"   )&RegistryFragmentArtifactTypesEdgesNoder   r   Nr    r   r   r   r   r      r#   r   c                   @  r"   )RegistryRoleFragmentr   r   Nr    r   r   r   r   r      r#   r   c                   @  s2   e Zd ZU dZded< ded< ded< ded	< d
S )RunInfoFragmentRunzTypename[Literal['Run']]r   r   r   r   r   r0   r1   Nr   r   r   r   r   r      s
   
 r   c                   @  s   e Zd ZU dZded< ded< ded< ded< ded	< ded
< ded< eddZded< eddZded< eddZded< dS )TeamMemberFragmentMemberzTypename[Literal['Member']]r   r)   r   roleOptional[bool]pendingemailusernamer   r   photoUrlr,   	photo_urlaccountTypeaccount_typeapiKeyapi_keyN)	r   r   r   r   r   r   r   r   r   r   r   r   r   r      s   
 r   c                   @  s   e Zd ZU dZded< ded< ded< ded	< ed
dZded< eddZded< eddZded< eddZ	ded< eddZ
ded< eddZded< eddZded< eddZded< eddZded< d ed!< d"S )#TeamFragmentEntityzTypename[Literal['Entity']]r   r   r   r   r   r   	availabler   r,   r)   r   readOnly	read_onlyreadOnlyAdminrN   read_only_adminisTeamis_teamprivateOnlyprivate_onlystorageBytesrK   storage_bytescodeSavingEnabledcode_saving_enableddefaultAccessdefault_accessisPaidis_paidzList[TeamMemberFragment]membersN)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r     s   
 r   c                   @  r   )TeamRegistryMemberFragmentr   teamr   r   Nr    r   r   r   r   r     r!   r   c                   @  s0   e Zd ZU ded< ded< eddZded< d	S )
TypeInfoFragmentr)   r   z&Optional[List[TypeInfoFragmentFields]]fieldsinputFieldsr,   z+Optional[List[TypeInfoFragmentInputFields]]input_fieldsN)r   r   r   r   r   r   r   r   r   r   r     s   
 r   c                   @  r   )TypeInfoFragmentFieldsr   r   z List[TypeInfoFragmentFieldsArgs]argsNr    r   r   r   r   r   $  r!   r   c                   @  r"   )TypeInfoFragmentFieldsArgsr   r   Nr    r   r   r   r   r   )  r#   r   c                   @  r"   )TypeInfoFragmentInputFieldsr   r   Nr    r   r   r   r   r   -  r#   r   c                   @  s6   e Zd ZU ded< ded< ded< ded< ded< d	S )
UserRegistryMemberFragmentr   r   r)   r   r   r   r   r   Nr    r   r   r   r   r   1  s   
 r   N);
__future__r   typingr   r   r   pydanticr   wandb._pydanticr   r   r	   enumsr   r   r   r   r$   r'   r2   r4   r9   r;   r<   r?   rG   r[   r_   rb   rd   re   rg   rh   rl   rw   rx   r{   r|   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   model_rebuildr   r   r   r   <module>   s   