| 279 | |
| 280 | |
| 281 | async def predict_grpc( |
| 282 | service_name, |
| 283 | payload, |
| 284 | parameters=None, |
| 285 | version=constants.KSERVE_V1BETA1_VERSION, |
| 286 | model_name=None, |
| 287 | network_layer: str = "istio", |
| 288 | ) -> InferResponse: |
| 289 | kfs_client = KServeClient( |
| 290 | config_file=os.environ.get("KUBECONFIG", "~/.kube/config") |
| 291 | ) |
| 292 | isvc = kfs_client.get( |
| 293 | service_name, |
| 294 | namespace=KSERVE_TEST_NAMESPACE, |
| 295 | version=version, |
| 296 | ) |
| 297 | _, cluster_ip, host, _ = get_isvc_endpoint(isvc, network_layer) |
| 298 | |
| 299 | if model_name is None: |
| 300 | model_name = service_name |
| 301 | client = grpc_client(host, cluster_ip) |
| 302 | |
| 303 | response = await client.infer( |
| 304 | InferRequest.from_grpc( |
| 305 | pb.ModelInferRequest( |
| 306 | model_name=model_name, inputs=payload, parameters=parameters |
| 307 | ) |
| 308 | ) |
| 309 | ) |
| 310 | return response |
| 311 | |
| 312 | |
| 313 | async def predict_modelmesh( |