o ϖi@sddlZddlmZmZmZmZmZer ddlmZddl m Z ddl m Z ddl mZmZGdddeZGd d d eZd eeefd efd dZdS)N) TYPE_CHECKINGAnyDictTupleUnion)ClientModel)Page) NamespaceResourcec@s:eZdZUdZeed< ejed< eed< eed<dS)Versionz A version of a model. id created_at cog_versionopenapi_schemaN)__name__ __module__ __qualname____doc__str__annotations__datetimedictrrE/home/ubuntu/.local/lib/python3.10/site-packages/replicate/version.pyr s  r cseZdZUdZeeefed<dddeeeeefdfddffdd Zd ede fd d Z d ede fd dZ de e fddZ de e fddZd edefddZd edefddZZS)Versionsz= Namespace for operations related to model versions. modelclientrr returnNcshtj|dddlm}t||r|j|jf|_dSt|tr/| dd\}}||f|_dS||_dS)N)rrr/) super__init__replicate.modelr isinstanceownernamerrsplit)selfrrr r&r' __class__rrr#%s    zVersions.__init__rc Cs8|jdd|jdd|jdd|}t|S) Get a specific model version. Args: id: The version ID. Returns: The model version. GET /v1/models/rr r! /versions/)_client_requestr_json_to_versionjsonr)rresprrrget4s " z Versions.getc s@|jdd|jdd|jdd|IdH}t|S)r,r-r.rr r!r/N)r0_async_requestrr2r3r4rrr async_getDs  " zVersions.async_getcCsX|jdd|jdd|jdd}|}dd|d D|d <ttd i|S) Return a list of all versions for a model. Returns: List[Version]: A list of version objects. r-r.rr r! /versionscSg|]}t|qSrr2.0resultrrr `z!Versions.list..resultsNr)r0r1rr3r r r)r5objrrrlistTs z Versions.listcs`|jdd|jdd|jddIdH}|}dd |d D|d <ttd i|S) r9r-r.rr r!r:NcSr;rr<r=rrrr@prAz'Versions.async_list..rBr)r0r7rr3r r rCrrr async_listds zVersions.async_listc Cs6|jdd|jdd|jdd|}|jdkS) Delete a model version and all associated predictions, including all output files. Model version deletion has some restrictions: * You can only delete versions from models you own. * You can only delete versions from private models. * You cannot delete a version if someone other than you has run predictions with it. Args: id: The version ID. DELETEr.rr r!r/)r0r1r status_coder4rrrdeletets" zVersions.deletec s>|jdd|jdd|jdd|IdH}|jdkS) rGrHr.rr r!r/NrI)r0r7rrJr4rrr async_deletes " zVersions.async_delete)rrrrrrrrr#r r6r8r rErFboolrKrL __classcell__rrr*rrs rr3rcCstdi|S)Nr)r )r3rrrr2sr2)rtypingrrrrrreplicate.clientrr$r replicate.paginationr replicate.resourcer r r rrr2rrrrs