MCPcopy
hub / github.com/kserve/kserve / predict_grpc

Function predict_grpc

test/e2e/common/utils.py:281–310  ·  view source on GitHub ↗
(
    service_name,
    payload,
    parameters=None,
    version=constants.KSERVE_V1BETA1_VERSION,
    model_name=None,
    network_layer: str = "istio",
)

Source from the content-addressed store, hash-verified

279
280
281async 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
313async def predict_modelmesh(

Calls 6

getMethod · 0.95
KServeClientClass · 0.90
get_isvc_endpointFunction · 0.85
grpc_clientFunction · 0.85
inferMethod · 0.45
from_grpcMethod · 0.45