async def METHOD_NAME(
    self, request: pb.ModelInferRequest, context: ServicerContext
) -> pb.ModelInferResponse:
    headers = to_headers(context)
    infer_request = InferRequest.from_grpc(request)
    response_body, _ = await self._data_plane.infer(request=infer_request, headers=headers,
                                                    model_name=request.model_name)
    if isinstance(response_body, pb.ModelInferResponse):
        return response_body
    elif isinstance(response_body, InferResponse):
        return response_body.to_grpc()
    else:
        return pb.ModelInferResponse(id=response_body["id"],
                                     model_name=response_body["model_name"],
                                     outputs=response_body["outputs"])