o ϖi;@sRddlmZmZmZmZmZmZmZddlm Z m Z ddl m Z ddl mZmZddlmZddlmZmZmZmZddlmZmZzddlmZWn eyWdd lZYnwerfdd lmZdd lmZGd d d eZ GdddeZ!dddee"efde fddZ#GdddeZ$GdddeZ%dee"ee"e"fe fde"fddZ&d S)) TYPE_CHECKINGAnyDictOptionalTuple TypedDictUnion)Unpack deprecated)Account)async_encode_json encode_json)Page) Prediction_create_prediction_body!_create_prediction_request_params_json_to_prediction) NamespaceResource)v1N)Client) Predictionsc@seZdZUdZeZded<eed< eed< Gddde Z e e ed< e e d d efd d Ze d efd dZe dddZdS) Deploymentz6 A deployment of a model hosted on Replicate. r_clientownernamec@s`eZdZUdZeed< eed< eed< eed< eeed< Gddde Z e ed <d S) zDeployment.Releasez, A release of a deployment. numbermodelversion created_at created_byc@s.eZdZUdZeed< eed< eed<dS)z Deployment.Release.Configurationz; A configuration for a deployment. hardware min_instances max_instancesN__name__ __module__ __qualname____doc__str__annotations__intr,r,H/home/ubuntu/.local/lib/python3.10/site-packages/replicate/deployment.py ConfigurationJs r. configurationN) r%r&r'r(r+r*r)rr rr.r,r,r,r-Release,s  r0current_releasezUse `deployment.owner` instead.returncCs|jS)z The name of the user or organization that owns the deployment. This attribute is deprecated and will be removed in future versions. )rselfr,r,r-usernamehszDeployment.usernamecCs|jd|jS)zS Return the qualified deployment name, in the format `owner/name`. /)rrr3r,r,r-idqsz Deployment.idDeploymentPredictionscCst|j|dS)z: Get the predictions for this deployment. )client deployment)r8rr3r,r,r- predictionsxszDeployment.predictionsN)r2r8)r%r&r'r(pydantic PrivateAttrrr*r)rr0rpropertyr r5r7r;r,r,r,r-rs"  7rc@sLeZdZUdZded< d*deeddfdeefd d Z d*deeddfdeefd d Z d edefddZ d edefddZ Gddde ZdeedefddZdeedefddZGddde ddZdededeedefdd Zdededeedefd!d"Zdededefd#d$Zdededefd%d&Zed+d(d)ZdS), Deploymentsz: Namespace for operations related to deployments. rr.cursorellipsisNr2cs^|durtdjd|durdn|}|}fdd|dD|d<ttd i|S) \ List all deployments. Returns: A page of Deployments. Ncursor cannot be NoneGET./v1/deploymentscg|]}tj|qSr,_json_to_deploymentr.0resultr3r,r-  z$Deployments.list..resultsr,) ValueErrorr_requestjsonrrr4r@respobjr,r3r-lists   zDeployments.listcsf|dur tdjd|durdn|IdH}|}fdd|dD|d<ttd i|S) rBNrCrD.rEcrFr,rGrIr3r,r-rLrMz*Deployments.async_list..rNr,)rOr_async_requestrQrrrRr,r3r- async_lists    zDeployments.async_listrcCs:|dd\}}|jdd|d|}t|j|S) Get a deployment by name. Args: name: The name of the deployment, in the format `owner/model-name`. Returns: The model. r6rD/v1/deployments/)splitrrPrHrQr4rrrSr,r,r-gets  zDeployments.getcsB|dd\}}|jdd|d|IdH}t|j|S)rXr6rYrDrZN)r[rrVrHrQr\r,r,r- async_gets  zDeployments.async_getc@sLeZdZUdZeed< eed< eed< eed< eed< eed<dS) z"Deployments.CreateDeploymentParamsz; Parameters for creating a new deployment. rrrr!r"r#Nr$r,r,r,r-CreateDeploymentParamss r_paramscKsR|dd}rd|vr|dd\}}||d<|jjdd|d}t|j|Sz Create a new deployment. Args: params: Configuration for the new deployment. Returns: The newly created Deployment. rNr6rYPOSTrErQ)r]r[rrPrHrQr4r`r_rSr,r,r-creates zDeployments.createcsZ|dd}rd|vr|dd\}}||d<|jjdd|dIdH}t|j|Sra)r]r[rrVrHrQrdr,r,r- async_create s  zDeployments.async_createc@s8eZdZUdZeed< eed< eed< eed<dS)z"Deployments.UpdateDeploymentParamszA Parameters for updating an existing deployment. rr!r"r#Nr$r,r,r,r-UpdateDeploymentParams&s rhF)totaldeployment_ownerdeployment_namecKs.|jjdd|d||d}t|j|S)& Update an existing deployment. Args: deployment_owner: The owner of the deployment. deployment_name: The name of the deployment. params: Configuration updates for the deployment. Returns: The updated Deployment. PATCHrZr6rc)rrPrHrQr4rjrkr`rSr,r,r-update7s zDeployments.updatecs6|jjdd|d||dIdH}t|j|S)rlrmrZr6rcN)rrVrHrQrnr,r,r- async_updatePs zDeployments.async_updatecCs$|jdd|d|}|jdkS)z Delete an existing deployment. Args: deployment_owner: The owner of the deployment. deployment_name: The name of the deployment. DELETErZr6)rrP status_coder4rjrkrSr,r,r-deleteis  zDeployments.deletecs,|jdd|d|IdH}|jdkS)z Delete an existing deployment asynchronously. Args: deployment_owner: The owner of the deployment. deployment_name: The name of the deployment. rqrZr6Nrr)rrVrsrtr,r,r- async_deletews  zDeployments.async_deleteDeploymentsPredictionscCs t|jdS)z2 Get predictions for deployments. r9)rwrr3r,r,r-r;s zDeployments.predictions).)r2rw)r%r&r'r(r*rr)rrrUrWr]r^rr_r rfrgrhrorpboolrurvr>r;r,r,r,r-r?s^        r?r9rrQr2cCstdi|}||_|S)Nr,)rr)r9rQr:r,r,r-rHsrHcs|eZdZUdZeed<dddeddffdd Zd eee fd e d de fd dZ d eee fd e d de fddZ ZS)r8zJ Namespace for operations related to predictions in a deployment. _deploymentr9rr:r2Ncstj|d||_dS)Nrx)super__init__rz)r4r9r: __class__r,r-r|s zDeploymentPredictions.__init__inputr`"Predictions.CreatePredictionParamscKs|dd}|dd}|durt||j|d}td d|d|}t|d}|jjdd|jjd |jjd fd |i|}t |j| S > Create a new prediction with the deployment. waitNfile_encoding_strategyr9rrrrrbrZr6 /predictionsrQr,) popr rrrrPrzrrrrQr4rr`rrbodyextrasrSr,r,r-rfs* zDeploymentPredictions.createcs|dd}|dd}|durt||j|dIdH}td d|d|}t|d}|jjdd|jjd |jjd fd |i|IdH}t |j| Sr) rr rrrrVrzrrrrQrr,r,r-rgs,   z"DeploymentPredictions.async_create)r%r&r'r(rr*r|rr)rr rrfrg __classcell__r,r,r}r-r8s$    r8c@seZdZdZdeeeeefefdeee fde dde fddZ deeeeefefdeee fde dde fd d Z d S) rwzI Namespace for operations related to predictions in deployments. r:rr`rr2c Ks|dd}|dd}t|}|durt||j|d}td d|d|}t|d}|jjd|fd|i|} t|j| S rrNrrrrrbrQr,) r&_create_prediction_url_from_deploymentr rrrrPrrQ r4r:rr`rrurlrrrSr,r,r-rfs  zDeploymentsPredictions.createc s|dd}|dd}t|}|dur t||j|dIdH}td d|d|}t|d}|jjd|fd|i|IdH} t|j| Sr) rrr rrrrVrrQrr,r,r-rgs    z#DeploymentsPredictions.async_createN)r%r&r'r(rr)rrrrr rrfrgr,r,r,r-rws(   rwr:cCsd\}}t|tr|j|j}}nt|tr |d|d}}n t|tr-|dd\}}|dus5|dur9tdd|d|dS)N)NNrrYr6zadeployment must be a Deployment, a tuple of (owner, name), or a string in the format 'owner/name'rZr) isinstancerrrtupler)r[rO)r:rrr,r,r-rs   r)'typingrrrrrrrtyping_extensionsr r replicate.accountr replicate.helpersr r replicate.paginationrreplicate.predictionrrrrreplicate.resourcerrr<r ImportErrorreplicate.clientrrrr?r)rHr8rwrr,r,r,r-s6$      fK<