
    ni,                         d dl mZ d dlmZ d dlmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'  G d de(          Z)dS )    )HttpRedirect)KaggleHttpClient)#ApiCreateModelInstanceRequest$ApiCreateModelInstanceVersionRequestApiCreateModelRequestApiCreateModelResponseApiDeleteModelInstanceRequest$ApiDeleteModelInstanceVersionRequestApiDeleteModelRequestApiDeleteModelResponse&ApiDownloadModelInstanceVersionRequestApiGetModelInstanceRequestApiGetModelRequest%ApiListModelGatingUserConsentsRequest&ApiListModelGatingUserConsentsResponseApiListModelInstancesRequestApiListModelInstancesResponse'ApiListModelInstanceVersionFilesRequest(ApiListModelInstanceVersionFilesResponse#ApiListModelInstanceVersionsRequest$ApiListModelInstanceVersionsResponseApiListModelsRequestApiListModelsResponseApiModelApiModelInstance&ApiReviewModelGatingUserConsentRequestApiUpdateModelInstanceRequestApiUpdateModelRequestApiUpdateModelResponseApiUploadModelFileRequestApiUploadModelFileResponseCreateModelSigningTokenRequestCreateModelSigningTokenResponseKeysRequestKeysResponseWellKnowEndpointRequestWellKnowEndpointResponsec                      e Zd ZdefdZddedefdZddedefdZ	dde
defdZddedefd	Zddedefd
ZddedefdZddedefdZddedefdZddedefdZddedefdZddede fdZ!dde"de#fdZ$dde%de&fdZ'dde(de)fdZ*dde+de)fdZ,dde-de.fdZ/dde0de1fdZ2dde3de4fdZ5dde6de7fdZ8dde9de:fdZ;dde<fdZ=dS )ModelApiClientclientc                     || _         d S N)_client)selfr*   s     h/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/kagglesdk/models/services/model_api_service.py__init__zModelApiClient.__init__   s    DLLL    Nrequestreturnc                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiCreateModelRequest):
        The request object; initialized to empty instance if not specified.
    Nmodels.ModelApiServiceCreateModel)r   r-   callr   r.   r2   s     r/   create_modelzModelApiClient.create_model
   2     %''g<5}gOefffr1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiCreateModelInstanceRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   CreateModelInstance)r   r-   r7   r   r8   s     r/   create_model_instancez$ModelApiClient.create_model_instance   3     -//g<57LgWmnnnr1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiCreateModelInstanceVersionRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   CreateModelInstanceVersion)r   r-   r7   r   r8   s     r/   create_model_instance_versionz,ModelApiClient.create_model_instance_version"   4     466g<57SU\^tuuur1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiDeleteModelRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   DeleteModel)r   r-   r7   r   r8   s     r/   delete_modelzModelApiClient.delete_model.   r:   r1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiDeleteModelInstanceRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   DeleteModelInstance)r	   r-   r7   r   r8   s     r/   delete_model_instancez$ModelApiClient.delete_model_instance:   r>   r1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiDeleteModelInstanceVersionRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   DeleteModelInstanceVersion)r
   r-   r7   r   r8   s     r/   delete_model_instance_versionz,ModelApiClient.delete_model_instance_versionF   rB   r1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiGetModelRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   GetModel)r   r-   r7   r   r8   s     r/   	get_modelzModelApiClient.get_modelR   s1     "$$g<5z7HUUUr1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiGetModelInstanceRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   GetModelInstance)r   r-   r7   r   r8   s     r/   get_model_instancez!ModelApiClient.get_model_instance^   s3     *,,g<57I7Tdeeer1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiDownloadModelInstanceVersionRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   DownloadModelInstanceVersion)r   r-   r7   r   r8   s     r/   download_model_instance_versionz.ModelApiClient.download_model_instance_versionj   s4     688g<57UW^`lmmmr1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiListModelsRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   
ListModels)r   r-   r7   r   r8   s     r/   list_modelszModelApiClient.list_modelsv   s2     $&&g<5|WNcdddr1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiListModelInstancesRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   ListModelInstances)r   r-   r7   r   r8   s     r/   list_model_instancesz#ModelApiClient.list_model_instances   s3     ,..g<57KWVstttr1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiListModelInstanceVersionsRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   ListModelInstanceVersions)r   r-   r7   r   r8   s     r/   list_model_instance_versionsz+ModelApiClient.list_model_instance_versions   sA     355g<57RT[  ^B  C  C  Cr1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiListModelInstanceVersionFilesRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   ListModelInstanceVersionFiles)r   r-   r7   r   r8   s     r/   !list_model_instance_version_filesz0ModelApiClient.list_model_instance_version_files   sA     799g<57VX_  bJ  K  K  Kr1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiUpdateModelRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   UpdateModel)r   r-   r7   r   r8   s     r/   update_modelzModelApiClient.update_model   r:   r1   c                 f    |t                      }| j                            dd|t                    S )z
    Args:
      request (ApiUpdateModelInstanceRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   UpdateModelInstance)r   r-   r7   r   r8   s     r/   update_model_instancez$ModelApiClient.update_model_instance   r>   r1   c                 f    |t                      }| j                            dd|t                    S )z
    Deprecated. Use the new unified BlobApiService#StartBlobUpload rpc.

    Args:
      request (ApiUploadModelFileRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   UploadModelFile)r    r-   r7   r!   r8   s     r/   upload_model_filez ModelApiClient.upload_model_file   s3     )++g<57H'Smnnnr1   c                 f    |t                      }| j                            dd|t                    S )a5  
    Creates an Kaggle issued identity token. The token is signed using a
    private key held in KMS that is only accessible by Kaggle model-signer
    service account.

    Args:
      request (CreateModelSigningTokenRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   CreateModelSigningToken)r"   r-   r7   r#   r8   s     r/   create_model_signing_tokenz)ModelApiClient.create_model_signing_token   s4     .00g<57PRY[z{{{r1   c                 f    |t                      }| j                            dd|t                    S )a"  
    see spec -
    https://openid.net/specs/openid-connect-discovery-1_0.html#ProviderConfig.
    Must support CORS. The service will have a path component.

    Args:
      request (WellKnowEndpointRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   WellKnowEndpoint)r&   r-   r7   r'   r8   s     r/   well_know_endpointz!ModelApiClient.well_know_endpoint   s3     '))g<57I7Tlmmmr1   c                 f    |t                      }| j                            dd|t                    S )z
    The JWKS endpoint containing the keys to validate the signature of a kaggle
    issued signing identity token.

    Args:
      request (KeysRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   Keys)r$   r-   r7   r%   r8   s     r/   keyszModelApiClient.keys   s/     g<5vwUUUr1   c                 f    |t                      }| j                            dd|t                    S )a  
    Model gating
    List the user consents for a gated model, under the model's current active
    agreement.

    Args:
      request (ApiListModelGatingUserConsentsRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   ListModelGatingUserConsents)r   r-   r7   r   r8   s     r/   list_model_gating_user_consentsz.ModelApiClient.list_model_gating_user_consents   sA     577g<57TV]  `F  G  G  Gr1   c                 `    |t                      }| j                            dd|d           dS )z
    Review the user consents for a gated model, under the model's current
    active agreement.

    Args:
      request (ApiReviewModelGatingUserConsentRequest):
        The request object; initialized to empty instance if not specified.
    Nr5   ReviewModelGatingUserConsent)r   r-   r7   r8   s     r/    review_model_gating_user_consentz/ModelApiClient.review_model_gating_user_consent  s:     688gL.0NPWY]^^^^^r1   r,   )>__name__
__module____qualname__r   r0   r   r   r9   r   r=   r   rA   r   r   rE   r	   rH   r
   rK   r   r   rN   r   r   rQ   r   r   rT   r   r   rW   r   r   rZ   r   r   r]   r   r   r`   r   r   rc   r   rf   r    r!   ri   r"   r#   rl   r&   r'   ro   r$   r%   rr   r   r   ru   r   rx    r1   r/   r)   r)      s"       -    
g 
g"7 
gCY 
g 
g 
g 
g
o 
o+H 
oTj 
o 
o 
o 
o
v 
v3W 
vcy 
v 
v 
v 
v
g 
g"7 
gCY 
g 
g 
g 
g
o 
o+H 
oTj 
o 
o 
o 
o
v 
v3W 
vcy 
v 
v 
v 
v
V 
V1 
VX 
V 
V 
V 
V
f 
f(B 
fN^ 
f 
f 
f 
f
n 
n5[ 
ngs 
n 
n 
n 
n
e 
e!5 
eAV 
e 
e 
e 
e
u 
u*F 
uRo 
u 
u 
u 
u
C 
C2U 
C  bF 
C 
C 
C 
C
K 
K7^ 
K  kS 
K 
K 
K 
K
g 
g"7 
gCY 
g 
g 
g 
g
o 
o+H 
oTj 
o 
o 
o 
oo o'@ oLf o o o o| |0N |Zy | | | | n n(? nKc n n n n V V+ V V V V VG G5Z G  gM G G G G _ _6\ _ _ _ _ _ _r1   r)   N)*$kagglesdk.common.types.http_redirectr   kagglesdk.kaggle_http_clientr   (kagglesdk.models.types.model_api_servicer   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   objectr)   r|   r1   r/   <module>r      s-   = = = = = = 9 9 9 9 9 9 c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  c  cS_ S_ S_ S_ S_V S_ S_ S_ S_ S_r1   