*VdZddlmZddlmZddlmZGddejZ y)z7Generated client library for gkerecommender version v1.)absolute_import)base_api)gkerecommender_v1_messagesc\eZdZdZeZdZdZdZdgZ dZ dZ dZ d Z dZdZd Z dfd ZGd d ej&ZGddej&ZGddej&ZGddej&ZGddej&ZGddej&ZxZS)GkerecommenderV1z?Generated client library for service gkerecommender version v1.z&https://gkerecommender.googleapis.com/z+https://gkerecommender.mtls.googleapis.com/gkerecommenderz.https://www.googleapis.com/auth/cloud-platformv1 CLIENT_ID CLIENT_SECRETzgoogle-cloud-sdkNc j|xs |j}tt|||||||||| | |  |j ||_|j ||_|j||_ |j||_ |j||_ |j||_y)z#Create a new gkerecommender handle.) credentialsget_credentialshttpmodel log_request log_responsecredentials_argsdefault_global_paramsadditional_http_headersresponse_encodingN)BASE_URLsuperr__init__BenchmarkingDataServicebenchmarkingDataModelServerVersionsServicemodelServerVersionsModelServersService modelServers ModelsServicemodelsOptimizedManifestServiceoptimizedManifestProfilesServiceprofiles) selfurlr rrrrrrrrr __class__s Wlib/googlecloudsdk/generated_clients/apis/gkerecommender/v1/gkerecommender_v1_client.pyrzGkerecommenderV1.__init__s  C D* 'd%l)3 7++-!88>D#>>tDD006D$$T*DK!::4@D((.DMc>eZdZdZdZfdZddZde_xZS)(GkerecommenderV1.BenchmarkingDataServicez0Service class for the benchmarkingData resource.rcNttj||i|_yN)rrrr_upload_configsr&clientr(s r)rz1GkerecommenderV1.BenchmarkingDataService.__init__6s#  4 4dDVL dr*cL|jd}|j|||S)aFetches all of the benchmarking data available for a profile. Benchmarking data returns all of the performance metrics available for a given model server setup on a given instance type. Args: request: (FetchBenchmarkingDataRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (FetchBenchmarkingDataResponse) The response message. Fetch global_paramsGetMethodConfig _RunMethodr&requestr5configs r)r3z.GkerecommenderV1.BenchmarkingDataService.Fetch;1##G,f __ '88r*c @tjddgggddddd S) NPOSTz%gkerecommender.benchmarkingData.fetchzv1/benchmarkingData:fetch FetchBenchmarkingDataRequestFetchBenchmarkingDataResponseF http_method method_idordered_params path_params query_params relative_path request_fieldrequest_type_nameresponse_type_namesupports_downloadr ApiMethodInfor*r)z1GkerecommenderV1.BenchmarkingDataService.Hs-("8"891!8: #r*r. __name__ __module__ __qualname____doc___NAMErr3 method_config __classcell__r(s@r)rr,1s": E 8 Er*rc>eZdZdZdZfdZddZde_xZS)+GkerecommenderV1.ModelServerVersionsServicez3Service class for the modelServerVersions resource.rcNttj||i|_yr.)rrrrr/r0s r)rz4GkerecommenderV1.ModelServerVersionsService.__init__Zs#  7 7GO dr*cL|jd}|j|||S)a{Fetches available model server versions. Open-source servers use their own versioning schemas (e.g., `vllm` uses semver like `v1.0.0`). Some model servers have different versioning schemas depending on the accelerator. For example, `vllm` uses semver on GPUs, but returns nightly build tags on TPUs. All available versions will be returned when different schemas are present. Args: request: (GkerecommenderModelServerVersionsFetchRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (FetchModelServerVersionsResponse) The response message. r3r4r6r9s r)r3z1GkerecommenderV1.ModelServerVersionsService.Fetch_r<r*c Dtjddgggdddddd S) NGETz(gkerecommender.modelServerVersions.fetch)r modelServerpageSize pageTokenzv1/modelServerVersions:fetch-GkerecommenderModelServerVersionsFetchRequest FetchModelServerVersionsResponseFrBrMrOr*r)rPz4GkerecommenderV1.ModelServerVersionsService.ls-("8"8<F4I= #r*r.rQrYs@r)rr[Us"= !E 8 Er*rc>eZdZdZdZfdZddZde_xZS)$GkerecommenderV1.ModelServersServicez,Service class for the modelServers resource.rcNttj||i|_yr.)rrrrr/r0s r)rz-GkerecommenderV1.ModelServersService.__init__~s#  0 0$@H dr*cL|jd}|j|||S)a`Fetches available model servers. Open-source model servers use simplified, lowercase names (e.g., `vllm`). Args: request: (GkerecommenderModelServersFetchRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (FetchModelServersResponse) The response message. r3r4r6r9s r)r3z*GkerecommenderV1.ModelServersService.Fetchr<r*c Dtjddgggdddddd S) Nr_z!gkerecommender.modelServers.fetch)rrarbzv1/modelServers:fetchrc&GkerecommenderModelServersFetchRequestFetchModelServersResponseFrBrMrOr*r)rPz-GkerecommenderV1.ModelServersService.s-("8"857-B6 #r*r.rQrYs@r)rrgys"6 E 8 Er*rc>eZdZdZdZfdZddZde_xZS)GkerecommenderV1.ModelsServicez&Service class for the models resource.r!cNttj||i|_yr.)rrr rr/r0s r)rz'GkerecommenderV1.ModelsService.__init__s#  * *D:6B dr*cL|jd}|j|||S)aLFetches available models. Open-source models follow the Huggingface Hub `owner/model_name` format. Args: request: (GkerecommenderModelsFetchRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (FetchModelsResponse) The response message. r3r4r6r9s r)r3z$GkerecommenderV1.ModelsService.Fetchr<r*c Dtjddggddgddddd S) Nr_zgkerecommender.models.fetchrarbzv1/models:fetchrc GkerecommenderModelsFetchRequestFetchModelsResponseFrBrMrOr*r)rPz'GkerecommenderV1.ModelsService.s2("8"8/ +.'<0 #r*r.rQrYs@r)r rns"0 E 8 Er*r c>eZdZdZdZfdZddZde_xZS))GkerecommenderV1.OptimizedManifestServicez1Service class for the optimizedManifest resource.r#cNttj||i|_yr.)rrr"rr/r0s r)rz2GkerecommenderV1.OptimizedManifestService.__init__s#  5 5tEfM dr*cL|jd}|j|||S)a[Generates an optimized deployment manifest for a given model and model server, based on the specified accelerator, performance targets, and configurations. See [Run best practice inference with GKE Inference Quickstart recipes](https://cloud.google.com/kubernetes-engine/docs/how-to/machine-learning/inference/inference-quickstart) for deployment details. Args: request: (GenerateOptimizedManifestRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (GenerateOptimizedManifestResponse) The response message. Generater4r6r9s r)rxz2GkerecommenderV1.OptimizedManifestService.Generates1##J/f __ '88r*c @tjddgggddddd S) Nr>z)gkerecommender.optimizedManifest.generatezv1/optimizedManifest:generater? GenerateOptimizedManifestRequest!GenerateOptimizedManifestResponseFrBrMrOr*r)rPz2GkerecommenderV1.OptimizedManifestService.s-X%;%;=5!<> &r*r.) rRrSrTrUrVrrxrWrXrYs@r)r"rus"; E 8 Hr*r"c>eZdZdZdZfdZddZde_xZS) GkerecommenderV1.ProfilesServicez(Service class for the profiles resource.r%cNttj||i|_yr.)rrr$rr/r0s r)rz)GkerecommenderV1.ProfilesService.__init__s#  , ,dzgkerecommender.profiles.fetchzv1/profiles:fetchr?FetchProfilesRequestFetchProfilesResponseFrBrMrOr*r)rPz)GkerecommenderV1.ProfilesService.s-("8"81)!02 #r*r.rQrYs@r)r$r}s"2 E 8 Er*r$) rcNTNNFFNNNN)rRrSrTrUmessagesMESSAGES_MODULEr MTLS_BASE_URL_PACKAGE_SCOPES_VERSION _CLIENT_ID_CLIENT_SECRET _USER_AGENT_CLIENT_CLASS_NAME _URL_VERSION_API_KEYrrBaseApiServicerrrr r"r$rXrYs@r)rr sG/ 5(?- ( = >' (*"."+), ()-6:/4<@?C /," 7 7"H"8#:#:"H"H33"H"h--"H"!8!8"H"//""r*rN) rU __future__rapitools.base.pyr7googlecloudsdk.generated_clients.apis.gkerecommender.v1rr BaseApiClientrrOr*r)rs'='%j}x--}r*