def get_model_metadata(self,
model_name,
model_version="",
headers=None,
as_json=False):
"""Contact the inference server and get the metadata for specified model.
Parameters
----------
model_name: str
The name of the model
model_version: str
The version of the model to get metadata. The default value
is an empty string which means then the server will choose
a version based on the model and internal policy.
headers: dict
Optional dictionary specifying additional HTTP
headers to include in the request.
as_json : bool
If True then returns model metadata as a json dict, otherwise
as a protobuf message. Default value is False.
Returns
-------
dict or protobuf message
The JSON dict or ModelMetadataResponse message holding
the metadata.
Raises
------
InferenceServerException
If unable to get model metadata.
"""
if headers is not None:
metadata = headers.items()
else:
metadata = ()
try:
request = grpc_service_v2_pb2.ModelMetadataRequest(
name=model_name, version=model_version)
response = self._client_stub.ModelMetadata(request=request,
metadata=metadata)
if as_json:
return json.loads(MessageToJson(response))
else:
return response
except grpc.RpcError as rpc_error:
raise_error_grpc(rpc_error)
request = grpc_service_v2_pb2.ServerReadyRequest()
response = grpc_stub.ServerReady(request)
print("server {}".format(response))
request = grpc_service_v2_pb2.ModelReadyRequest(name=model_name,
version=model_version)
response = grpc_stub.ModelReady(request)
print("model {}".format(response))
# Metadata
request = grpc_service_v2_pb2.ServerMetadataRequest()
response = grpc_stub.ServerMetadata(request)
print("server metadata:\n{}".format(response))
request = grpc_service_v2_pb2.ModelMetadataRequest(name=model_name,
version=model_version)
response = grpc_stub.ModelMetadata(request)
print("model metadata:\n{}".format(response))
# Configuration
request = grpc_service_v2_pb2.ModelConfigRequest(name=model_name,
version=model_version)
response = grpc_stub.ModelConfig(request)
print("model config:\n{}".format(response))
# Infer
request = grpc_service_v2_pb2.ModelInferRequest()
request.model_name = model_name
request.model_version = model_version
request.id = model_name + "-id-0"
request = grpc_service_v2_pb2.ServerReadyRequest()
response = grpc_stub.ServerReady(request)
print("server {}".format(response))
request = grpc_service_v2_pb2.ModelReadyRequest(
name="resnet_v1_50_graphdef", version=model_version)
response = grpc_stub.ModelReady(request)
print("model {}".format(response))
# Metadata
request = grpc_service_v2_pb2.ServerMetadataRequest()
response = grpc_stub.ServerMetadata(request)
print("server metadata:\n{}".format(response))
request = grpc_service_v2_pb2.ModelMetadataRequest(
name="resnet_v1_50_graphdef", version=model_version)
response = grpc_stub.ModelMetadata(request)
print("model metadata:\n{}".format(response))
# Configuration
request = grpc_service_v2_pb2.ModelConfigRequest(
name="resnet_v1_50_graphdef", version=model_version)
response = grpc_stub.ModelConfig(request)
print("model config:\n{}".format(response))
# Infer
request = grpc_service_v2_pb2.ModelInferRequest()
request.model_name = "resnet_v1_50_graphdef"
request.model_version = model_version
request.id = "my request id"