o ϖi6:@sddlmZmZmZmZmZmZmZmZddl m Z m Z m Z m Z ddlmZddlmZmZddlmZddlmZddlmZmZmZmZddlmZmZdd lm Z m!Z!zdd l"m#Z"Wn e$ykdd l"Z"Ynwerzdd l%m&Z&dd lm'Z'GdddeZ(GdddeZ)GdddeZ*d d d d d dde+de+de+de+dee+dee+dee+dee+dee+dee+effdd Z,de+fd!d"Z-de+fd#d$Z.d%d&d'ee+efde(fd(d)Z/d*ee+ee+e+fdfde+fd+d,Z0d S)-) TYPE_CHECKINGAnyDictLiteralOptionalTupleUnionoverload) NotRequired TypedDictUnpack deprecatedReplicateException)async_encode_json encode_json)ModelVersionIdentifier)Page) Prediction_create_prediction_body!_create_prediction_request_params_json_to_prediction) NamespaceResource)VersionVersions)v1N)Client) Predictionsc@s,eZdZUdZeZded<eed< eed< eed< e eed< e ded < e eed < e eed < e eed < e ed < e eed< e e ed< e e ed< edefddZeeddefddZejeddeddfddZd ddZedefddZd ddZdS)!Modelz7 A machine learning model hosted on Replicate. r_clienturlownername descriptionpublicprivate visibility github_url paper_url license_url run_countcover_image_urldefault_examplelatest_versionreturncCs|jd|jS)zN Return the qualified model name, in the format `owner/name`. /)r"r#selfr4C/home/ubuntu/.local/lib/python3.10/site-packages/replicate/model.pyid`szModel.idzUse `model.owner` instead.cCs|jS)z The name of the user or organization that owns the model. This attribute is deprecated and will be removed in future versions. r"r2r4r4r5usernamegszModel.usernamevalueNcCs ||_dSNr7)r3r9r4r4r5r8ps cOstd)z< DEPRECATED: Use `replicate.run()` instead. aThe `model.predict()` method has been removed, because it's unstable: if a new version of the model you're using is pushed and its API has changed, your code may break. Use `replicate.run()` instead. See https://github.com/replicate/replicate-python#readmer)r3argskwargsr4r4r5predictusz Model.predictcCst|j|dS)z1 Get the versions of this model. )clientmodel)rr r2r4r4r5versions~szModel.versionscCsB|jj|jd|j}|D] \}}t|||qdS)z3 Load this object from the server. r1N)r modelsgetr"r#dictitemssetattr)r3objr#r9r4r4r5reloadsz Model.reload)r0N)__name__ __module__ __qualname____doc__pydantic PrivateAttrr __annotations__strrrintrrpropertyr6r r8setterr=rr@rGr4r4r4r5rsN           rc@seZdZdZeZed.ddZd/dee dd fde efd d Z d/dee dd fde efd d Z de de efddZ de de efddZede defddZede de defddZdefddZede defddZede de defddZdefddZede defddZede de defd dZdefd!dZede defd"d#Zede de defd$d#Zdefd%d#ZGd&d'd'eZde de d(ed)defd*d+Zde de d(ed)defd,d-Zd S)0Modelsz5 Namespace for operations related to models. r0ModelsPredictionscCs t|jdS)zS Get a namespace for operations related to predictions on a model. )r>)rTr r2r4r4r5 predictionss zModels.predictions.cursorellipsisNcs^|durtdjd|durdn|}|}fdd|dD|d<ttd i|S) # List all public models. Parameters: cursor: The cursor to use for pagination. Use the value of `Page.next` or `Page.previous`. Returns: Page[Model]: A page of of models. Raises: ValueError: If `cursor` is `None`. Ncursor cannot be NoneGET. /v1/modelscg|]}tj|qSr4_json_to_modelr .0resultr2r4r5  zModels.list..resultsr4) ValueErrorr _requestjsonrrr3rVresprFr4r2r5lists   z Models.listcsf|dur tdjd|durdn|IdH}|}fdd|dD|d<ttd i|S) rXNrYrZ.r[cr\r4r]r_r2r4r5rbrcz%Models.async_list..rdr4)rer _async_requestrgrrrhr4r2r5 async_lists   zModels.async_listquerycsLjjdd|ddid}|}fdd|dD|d<ttd i|S) z Search for public models. Parameters: query: The search query. Returns: Page[Model]: A page of models matching the search query. QUERYr[ Content-Type text/plaincontentheaderscr\r4r]r_r2r4r5rbrcz!Models.search..rdNr4)r rfrgrrr3rmrirFr4r2r5searchs   z Models.searchcsTjjdd|ddidIdH}|}fdd|d D|d <ttd i|S) z Asynchronously search for public models. Parameters: query: The search query. Returns: Page[Model]: A page of models matching the search query. rnr[rorprqNcr\r4r]r_r2r4r5rbrcz'Models.async_search..rdr4)r rkrgrrrtr4r2r5 async_searchs    zModels.async_searchkeycCdSr:r4r3rwr4r4r5rBz Models.getr"r#cCrxr:r4r3r"r#r4r4r5rBrzcOs,t|i|}|jd|}t|j|S)z& Get a model by name. rZ)_get_model_urlr rfr^rgr3r;r<r!rir4r4r5rBscdSr:r4ryr4r4r5 async_get zModels.async_getcr~r:r4r{r4r4r5rrcs4t|i|}|jd|IdH}t|j|S)z Get a model by name. Args: key: The qualified name of the model, in the format `owner/name`. Returns: The model. rZN)r|r rkr^rgr}r4r4r5rs cCrxr:r4ryr4r4r5delete!rzz Models.deletecCrxr:r4r{r4r4r5r$rzcOs&t|i|}|jd|}|jdkS)z} Delete a model by name. Returns: `True` if deletion was successful, otherwise `False`. DELETE)_delete_model_urlr rf status_coder}r4r4r5r's cr~r:r4ryr4r4r5 async_delete2rzModels.async_deletecr~r:r4r{r4r4r5r5rcs.t|i|}|jd|IdH}|jdkS)z Asynchronously delete a model by name. Returns: `True` if deletion was successful, otherwise `False`. rNr)rr rkrr}r4r4r5r8s c@sneZdZUdZeed< eded< eeed< eeed< eeed< eeed< eeed <d S) Models.CreateModelParamsz Parameters for creating a model.hardwarer%r(r$r)r*r+r-N)rHrIrJrKrOrNrr r4r4r4r5CreateModelParamsCs       rparamsrcKs4t||fi|}|jjdd|d}t|j|S)! Create a model. POSTr[rg)_create_model_bodyr rfr^rgr3r"r#rbodyrir4r4r5create]s z Models.createcs@t||fi|}}|jjdd|dIdH}t|j|S)rrr[rN)rr rkr^rgrr4r4r5 async_createlszModels.async_create)r0rT).)rHrIrJrKrr?rQrUrrOrrjrlrurvr rBrboolrrr rr rrr4r4r4r5rSsj "      rSc@seZdZdZdeeeeefdfdeeefde dde fdd Z deeeeefdfdeeefde dde fd d Z d S) rTzJ Namespace for operations related to predictions in a deployment. r?rinputrz"Predictions.CreatePredictionParamsr0c Ks|dd}|dd}t|}|durt||j|d}td d|d|}t|d}|jjd|fd|i|} t|j| S z> Create a new prediction with the deployment. waitNfile_encoding_strategy)r>r)versionr)rrrgr4) pop"_create_prediction_path_from_modelrr rrrfrrg r3r?rrrrpathrextrasrir4r4r5r~s  zModelsPredictions.createc s|dd}|dd}t|}|dur t||j|dIdH}td d|d|}t|d}|jjd|fd|i|IdH} t|j| Sr) rrrr rrrkrrgrr4r4r5rs    zModelsPredictions.async_createN) rHrIrJrKrrOrrrr rrrr4r4r4r5rTys(   rT)r$r)r*r+r-r"r#r(rr$r)r*r+r-r0c Csb||||d} |dur|| d<|dur|| d<|dur|| d<|dur'|| d<|dur/|| d<| S)N)r"r#r(rr$r)r*r+r-r4) r"r#r(rr$r)r*r+r-rr4r4r5rs  rcOst|dkrt|dkrtd|dd}|dd}|dd}|r,|s(|r,td|rHt|dkr9|d}nt|dkrD|\}}ntd |sY|rN|sRtd |d |}d |S) Nrz+Cannot mix positional and keyword argumentsr"r#rwzYMust specify exactly one of 'owner' and 'name' or single 'key' in the format 'owner/name'zInvalid number of argumentszCBoth 'owner' and 'name' must be provided if 'key' is not specified.r1 /v1/models/)lenrerB)r;r<r"r#rwr4r4r5r|s,         r|cOst|i|Sr:)r|)r;r<r4r4r5rsrr>rrgcCs*tdi|}||_|jdur||j_|S)Nr4)rr r.)r>rgr?r4r4r5r^s  r^r?cCsd\}}t|tr|j|j}}n(t|tr |d|d}}nt|tr9t|\}}}|dur9td|d|dusA|durEtdd|d|d S) N)NNrrz$Invalid reference to model version: z6. Expected model or reference in the format owner/namezWmodel must be a Model, a tuple of (owner, name), or a string in the format 'owner/name'rr1z /predictions) isinstancerr"r#tuplerOrparsere)r?r"r# version_idr4r4r5rs     r)1typingrrrrrrrr typing_extensionsr r r r replicate.exceptionsrreplicate.helpersrrreplicate.identifierrreplicate.paginationrreplicate.predictionrrrrreplicate.resourcerrreplicate.versionrrrLr ImportErrorreplicate.clientrrrrSrTrOrr|rr^rr4r4r4r5sn(       sjC   %