#![allow(rustdoc::redundant_explicit_links)]
#![allow(rustdoc::broken_intra_doc_links)]
#![allow(rustdoc::invalid_html_tags)]
#[cfg(feature = "data-foundry-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "data-foundry-service")))]
#[derive(Clone, Debug)]
pub struct DataFoundryService {
inner: std::sync::Arc<dyn super::stub::dynamic::DataFoundryService>,
}
#[cfg(feature = "data-foundry-service")]
impl DataFoundryService {
pub fn builder() -> super::builder::data_foundry_service::ClientBuilder {
crate::new_client_builder(super::builder::data_foundry_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::DataFoundryService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::DataFoundryService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::DataFoundryService> {
super::transport::DataFoundryService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::DataFoundryService> {
Self::build_transport(conf)
.await
.map(super::tracing::DataFoundryService::new)
}
pub fn generate_synthetic_data(
&self,
) -> super::builder::data_foundry_service::GenerateSyntheticData {
super::builder::data_foundry_service::GenerateSyntheticData::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::data_foundry_service::ListLocations {
super::builder::data_foundry_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::data_foundry_service::GetLocation {
super::builder::data_foundry_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::data_foundry_service::SetIamPolicy {
super::builder::data_foundry_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::data_foundry_service::GetIamPolicy {
super::builder::data_foundry_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::data_foundry_service::TestIamPermissions {
super::builder::data_foundry_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::data_foundry_service::ListOperations {
super::builder::data_foundry_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::data_foundry_service::GetOperation {
super::builder::data_foundry_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::data_foundry_service::DeleteOperation {
super::builder::data_foundry_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::data_foundry_service::CancelOperation {
super::builder::data_foundry_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::data_foundry_service::WaitOperation {
super::builder::data_foundry_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "dataset-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "dataset-service")))]
#[derive(Clone, Debug)]
pub struct DatasetService {
inner: std::sync::Arc<dyn super::stub::dynamic::DatasetService>,
}
#[cfg(feature = "dataset-service")]
impl DatasetService {
pub fn builder() -> super::builder::dataset_service::ClientBuilder {
crate::new_client_builder(super::builder::dataset_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::DatasetService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::DatasetService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::DatasetService> {
super::transport::DatasetService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::DatasetService> {
Self::build_transport(conf)
.await
.map(super::tracing::DatasetService::new)
}
pub fn create_dataset(&self) -> super::builder::dataset_service::CreateDataset {
super::builder::dataset_service::CreateDataset::new(self.inner.clone())
}
pub fn get_dataset(&self) -> super::builder::dataset_service::GetDataset {
super::builder::dataset_service::GetDataset::new(self.inner.clone())
}
pub fn update_dataset(&self) -> super::builder::dataset_service::UpdateDataset {
super::builder::dataset_service::UpdateDataset::new(self.inner.clone())
}
pub fn list_datasets(&self) -> super::builder::dataset_service::ListDatasets {
super::builder::dataset_service::ListDatasets::new(self.inner.clone())
}
pub fn delete_dataset(&self) -> super::builder::dataset_service::DeleteDataset {
super::builder::dataset_service::DeleteDataset::new(self.inner.clone())
}
pub fn import_data(&self) -> super::builder::dataset_service::ImportData {
super::builder::dataset_service::ImportData::new(self.inner.clone())
}
pub fn export_data(&self) -> super::builder::dataset_service::ExportData {
super::builder::dataset_service::ExportData::new(self.inner.clone())
}
pub fn create_dataset_version(&self) -> super::builder::dataset_service::CreateDatasetVersion {
super::builder::dataset_service::CreateDatasetVersion::new(self.inner.clone())
}
pub fn update_dataset_version(&self) -> super::builder::dataset_service::UpdateDatasetVersion {
super::builder::dataset_service::UpdateDatasetVersion::new(self.inner.clone())
}
pub fn delete_dataset_version(&self) -> super::builder::dataset_service::DeleteDatasetVersion {
super::builder::dataset_service::DeleteDatasetVersion::new(self.inner.clone())
}
pub fn get_dataset_version(&self) -> super::builder::dataset_service::GetDatasetVersion {
super::builder::dataset_service::GetDatasetVersion::new(self.inner.clone())
}
pub fn list_dataset_versions(&self) -> super::builder::dataset_service::ListDatasetVersions {
super::builder::dataset_service::ListDatasetVersions::new(self.inner.clone())
}
pub fn restore_dataset_version(
&self,
) -> super::builder::dataset_service::RestoreDatasetVersion {
super::builder::dataset_service::RestoreDatasetVersion::new(self.inner.clone())
}
pub fn list_data_items(&self) -> super::builder::dataset_service::ListDataItems {
super::builder::dataset_service::ListDataItems::new(self.inner.clone())
}
pub fn search_data_items(&self) -> super::builder::dataset_service::SearchDataItems {
super::builder::dataset_service::SearchDataItems::new(self.inner.clone())
}
pub fn list_saved_queries(&self) -> super::builder::dataset_service::ListSavedQueries {
super::builder::dataset_service::ListSavedQueries::new(self.inner.clone())
}
pub fn delete_saved_query(&self) -> super::builder::dataset_service::DeleteSavedQuery {
super::builder::dataset_service::DeleteSavedQuery::new(self.inner.clone())
}
pub fn get_annotation_spec(&self) -> super::builder::dataset_service::GetAnnotationSpec {
super::builder::dataset_service::GetAnnotationSpec::new(self.inner.clone())
}
pub fn list_annotations(&self) -> super::builder::dataset_service::ListAnnotations {
super::builder::dataset_service::ListAnnotations::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::dataset_service::ListLocations {
super::builder::dataset_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::dataset_service::GetLocation {
super::builder::dataset_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::dataset_service::SetIamPolicy {
super::builder::dataset_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::dataset_service::GetIamPolicy {
super::builder::dataset_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::dataset_service::TestIamPermissions {
super::builder::dataset_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::dataset_service::ListOperations {
super::builder::dataset_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::dataset_service::GetOperation {
super::builder::dataset_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::dataset_service::DeleteOperation {
super::builder::dataset_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::dataset_service::CancelOperation {
super::builder::dataset_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::dataset_service::WaitOperation {
super::builder::dataset_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "deployment-resource-pool-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "deployment-resource-pool-service")))]
#[derive(Clone, Debug)]
pub struct DeploymentResourcePoolService {
inner: std::sync::Arc<dyn super::stub::dynamic::DeploymentResourcePoolService>,
}
#[cfg(feature = "deployment-resource-pool-service")]
impl DeploymentResourcePoolService {
pub fn builder() -> super::builder::deployment_resource_pool_service::ClientBuilder {
crate::new_client_builder(super::builder::deployment_resource_pool_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::DeploymentResourcePoolService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<
std::sync::Arc<dyn super::stub::dynamic::DeploymentResourcePoolService>,
> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::DeploymentResourcePoolService> {
super::transport::DeploymentResourcePoolService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::DeploymentResourcePoolService> {
Self::build_transport(conf)
.await
.map(super::tracing::DeploymentResourcePoolService::new)
}
pub fn create_deployment_resource_pool(
&self,
) -> super::builder::deployment_resource_pool_service::CreateDeploymentResourcePool {
super::builder::deployment_resource_pool_service::CreateDeploymentResourcePool::new(
self.inner.clone(),
)
}
pub fn get_deployment_resource_pool(
&self,
) -> super::builder::deployment_resource_pool_service::GetDeploymentResourcePool {
super::builder::deployment_resource_pool_service::GetDeploymentResourcePool::new(
self.inner.clone(),
)
}
pub fn list_deployment_resource_pools(
&self,
) -> super::builder::deployment_resource_pool_service::ListDeploymentResourcePools {
super::builder::deployment_resource_pool_service::ListDeploymentResourcePools::new(
self.inner.clone(),
)
}
pub fn update_deployment_resource_pool(
&self,
) -> super::builder::deployment_resource_pool_service::UpdateDeploymentResourcePool {
super::builder::deployment_resource_pool_service::UpdateDeploymentResourcePool::new(
self.inner.clone(),
)
}
pub fn delete_deployment_resource_pool(
&self,
) -> super::builder::deployment_resource_pool_service::DeleteDeploymentResourcePool {
super::builder::deployment_resource_pool_service::DeleteDeploymentResourcePool::new(
self.inner.clone(),
)
}
pub fn query_deployed_models(
&self,
) -> super::builder::deployment_resource_pool_service::QueryDeployedModels {
super::builder::deployment_resource_pool_service::QueryDeployedModels::new(
self.inner.clone(),
)
}
pub fn list_locations(
&self,
) -> super::builder::deployment_resource_pool_service::ListLocations {
super::builder::deployment_resource_pool_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::deployment_resource_pool_service::GetLocation {
super::builder::deployment_resource_pool_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::deployment_resource_pool_service::SetIamPolicy {
super::builder::deployment_resource_pool_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::deployment_resource_pool_service::GetIamPolicy {
super::builder::deployment_resource_pool_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::deployment_resource_pool_service::TestIamPermissions {
super::builder::deployment_resource_pool_service::TestIamPermissions::new(
self.inner.clone(),
)
}
pub fn list_operations(
&self,
) -> super::builder::deployment_resource_pool_service::ListOperations {
super::builder::deployment_resource_pool_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::deployment_resource_pool_service::GetOperation {
super::builder::deployment_resource_pool_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(
&self,
) -> super::builder::deployment_resource_pool_service::DeleteOperation {
super::builder::deployment_resource_pool_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(
&self,
) -> super::builder::deployment_resource_pool_service::CancelOperation {
super::builder::deployment_resource_pool_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(
&self,
) -> super::builder::deployment_resource_pool_service::WaitOperation {
super::builder::deployment_resource_pool_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "endpoint-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "endpoint-service")))]
#[derive(Clone, Debug)]
pub struct EndpointService {
inner: std::sync::Arc<dyn super::stub::dynamic::EndpointService>,
}
#[cfg(feature = "endpoint-service")]
impl EndpointService {
pub fn builder() -> super::builder::endpoint_service::ClientBuilder {
crate::new_client_builder(super::builder::endpoint_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::EndpointService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::EndpointService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::EndpointService> {
super::transport::EndpointService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::EndpointService> {
Self::build_transport(conf)
.await
.map(super::tracing::EndpointService::new)
}
pub fn create_endpoint(&self) -> super::builder::endpoint_service::CreateEndpoint {
super::builder::endpoint_service::CreateEndpoint::new(self.inner.clone())
}
pub fn get_endpoint(&self) -> super::builder::endpoint_service::GetEndpoint {
super::builder::endpoint_service::GetEndpoint::new(self.inner.clone())
}
pub fn list_endpoints(&self) -> super::builder::endpoint_service::ListEndpoints {
super::builder::endpoint_service::ListEndpoints::new(self.inner.clone())
}
pub fn update_endpoint(&self) -> super::builder::endpoint_service::UpdateEndpoint {
super::builder::endpoint_service::UpdateEndpoint::new(self.inner.clone())
}
pub fn update_endpoint_long_running(
&self,
) -> super::builder::endpoint_service::UpdateEndpointLongRunning {
super::builder::endpoint_service::UpdateEndpointLongRunning::new(self.inner.clone())
}
pub fn delete_endpoint(&self) -> super::builder::endpoint_service::DeleteEndpoint {
super::builder::endpoint_service::DeleteEndpoint::new(self.inner.clone())
}
pub fn deploy_model(&self) -> super::builder::endpoint_service::DeployModel {
super::builder::endpoint_service::DeployModel::new(self.inner.clone())
}
pub fn undeploy_model(&self) -> super::builder::endpoint_service::UndeployModel {
super::builder::endpoint_service::UndeployModel::new(self.inner.clone())
}
pub fn mutate_deployed_model(&self) -> super::builder::endpoint_service::MutateDeployedModel {
super::builder::endpoint_service::MutateDeployedModel::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::endpoint_service::ListLocations {
super::builder::endpoint_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::endpoint_service::GetLocation {
super::builder::endpoint_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::endpoint_service::SetIamPolicy {
super::builder::endpoint_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::endpoint_service::GetIamPolicy {
super::builder::endpoint_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::endpoint_service::TestIamPermissions {
super::builder::endpoint_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::endpoint_service::ListOperations {
super::builder::endpoint_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::endpoint_service::GetOperation {
super::builder::endpoint_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::endpoint_service::DeleteOperation {
super::builder::endpoint_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::endpoint_service::CancelOperation {
super::builder::endpoint_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::endpoint_service::WaitOperation {
super::builder::endpoint_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "evaluation-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "evaluation-service")))]
#[derive(Clone, Debug)]
pub struct EvaluationService {
inner: std::sync::Arc<dyn super::stub::dynamic::EvaluationService>,
}
#[cfg(feature = "evaluation-service")]
impl EvaluationService {
pub fn builder() -> super::builder::evaluation_service::ClientBuilder {
crate::new_client_builder(super::builder::evaluation_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::EvaluationService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::EvaluationService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::EvaluationService> {
super::transport::EvaluationService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::EvaluationService> {
Self::build_transport(conf)
.await
.map(super::tracing::EvaluationService::new)
}
pub fn evaluate_instances(&self) -> super::builder::evaluation_service::EvaluateInstances {
super::builder::evaluation_service::EvaluateInstances::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::evaluation_service::ListLocations {
super::builder::evaluation_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::evaluation_service::GetLocation {
super::builder::evaluation_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::evaluation_service::SetIamPolicy {
super::builder::evaluation_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::evaluation_service::GetIamPolicy {
super::builder::evaluation_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::evaluation_service::TestIamPermissions {
super::builder::evaluation_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::evaluation_service::ListOperations {
super::builder::evaluation_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::evaluation_service::GetOperation {
super::builder::evaluation_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::evaluation_service::DeleteOperation {
super::builder::evaluation_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::evaluation_service::CancelOperation {
super::builder::evaluation_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::evaluation_service::WaitOperation {
super::builder::evaluation_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "feature-online-store-admin-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "feature-online-store-admin-service")))]
#[derive(Clone, Debug)]
pub struct FeatureOnlineStoreAdminService {
inner: std::sync::Arc<dyn super::stub::dynamic::FeatureOnlineStoreAdminService>,
}
#[cfg(feature = "feature-online-store-admin-service")]
impl FeatureOnlineStoreAdminService {
pub fn builder() -> super::builder::feature_online_store_admin_service::ClientBuilder {
crate::new_client_builder(
super::builder::feature_online_store_admin_service::client::Factory,
)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::FeatureOnlineStoreAdminService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<
std::sync::Arc<dyn super::stub::dynamic::FeatureOnlineStoreAdminService>,
> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::FeatureOnlineStoreAdminService> {
super::transport::FeatureOnlineStoreAdminService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::FeatureOnlineStoreAdminService> {
Self::build_transport(conf)
.await
.map(super::tracing::FeatureOnlineStoreAdminService::new)
}
pub fn create_feature_online_store(
&self,
) -> super::builder::feature_online_store_admin_service::CreateFeatureOnlineStore {
super::builder::feature_online_store_admin_service::CreateFeatureOnlineStore::new(
self.inner.clone(),
)
}
pub fn get_feature_online_store(
&self,
) -> super::builder::feature_online_store_admin_service::GetFeatureOnlineStore {
super::builder::feature_online_store_admin_service::GetFeatureOnlineStore::new(
self.inner.clone(),
)
}
pub fn list_feature_online_stores(
&self,
) -> super::builder::feature_online_store_admin_service::ListFeatureOnlineStores {
super::builder::feature_online_store_admin_service::ListFeatureOnlineStores::new(
self.inner.clone(),
)
}
pub fn update_feature_online_store(
&self,
) -> super::builder::feature_online_store_admin_service::UpdateFeatureOnlineStore {
super::builder::feature_online_store_admin_service::UpdateFeatureOnlineStore::new(
self.inner.clone(),
)
}
pub fn delete_feature_online_store(
&self,
) -> super::builder::feature_online_store_admin_service::DeleteFeatureOnlineStore {
super::builder::feature_online_store_admin_service::DeleteFeatureOnlineStore::new(
self.inner.clone(),
)
}
pub fn create_feature_view(
&self,
) -> super::builder::feature_online_store_admin_service::CreateFeatureView {
super::builder::feature_online_store_admin_service::CreateFeatureView::new(
self.inner.clone(),
)
}
pub fn get_feature_view(
&self,
) -> super::builder::feature_online_store_admin_service::GetFeatureView {
super::builder::feature_online_store_admin_service::GetFeatureView::new(self.inner.clone())
}
pub fn list_feature_views(
&self,
) -> super::builder::feature_online_store_admin_service::ListFeatureViews {
super::builder::feature_online_store_admin_service::ListFeatureViews::new(
self.inner.clone(),
)
}
pub fn update_feature_view(
&self,
) -> super::builder::feature_online_store_admin_service::UpdateFeatureView {
super::builder::feature_online_store_admin_service::UpdateFeatureView::new(
self.inner.clone(),
)
}
pub fn delete_feature_view(
&self,
) -> super::builder::feature_online_store_admin_service::DeleteFeatureView {
super::builder::feature_online_store_admin_service::DeleteFeatureView::new(
self.inner.clone(),
)
}
pub fn sync_feature_view(
&self,
) -> super::builder::feature_online_store_admin_service::SyncFeatureView {
super::builder::feature_online_store_admin_service::SyncFeatureView::new(self.inner.clone())
}
pub fn get_feature_view_sync(
&self,
) -> super::builder::feature_online_store_admin_service::GetFeatureViewSync {
super::builder::feature_online_store_admin_service::GetFeatureViewSync::new(
self.inner.clone(),
)
}
pub fn list_feature_view_syncs(
&self,
) -> super::builder::feature_online_store_admin_service::ListFeatureViewSyncs {
super::builder::feature_online_store_admin_service::ListFeatureViewSyncs::new(
self.inner.clone(),
)
}
pub fn list_locations(
&self,
) -> super::builder::feature_online_store_admin_service::ListLocations {
super::builder::feature_online_store_admin_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::feature_online_store_admin_service::GetLocation {
super::builder::feature_online_store_admin_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(
&self,
) -> super::builder::feature_online_store_admin_service::SetIamPolicy {
super::builder::feature_online_store_admin_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(
&self,
) -> super::builder::feature_online_store_admin_service::GetIamPolicy {
super::builder::feature_online_store_admin_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::feature_online_store_admin_service::TestIamPermissions {
super::builder::feature_online_store_admin_service::TestIamPermissions::new(
self.inner.clone(),
)
}
pub fn list_operations(
&self,
) -> super::builder::feature_online_store_admin_service::ListOperations {
super::builder::feature_online_store_admin_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(
&self,
) -> super::builder::feature_online_store_admin_service::GetOperation {
super::builder::feature_online_store_admin_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(
&self,
) -> super::builder::feature_online_store_admin_service::DeleteOperation {
super::builder::feature_online_store_admin_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(
&self,
) -> super::builder::feature_online_store_admin_service::CancelOperation {
super::builder::feature_online_store_admin_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(
&self,
) -> super::builder::feature_online_store_admin_service::WaitOperation {
super::builder::feature_online_store_admin_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "feature-online-store-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "feature-online-store-service")))]
#[derive(Clone, Debug)]
pub struct FeatureOnlineStoreService {
inner: std::sync::Arc<dyn super::stub::dynamic::FeatureOnlineStoreService>,
}
#[cfg(feature = "feature-online-store-service")]
impl FeatureOnlineStoreService {
pub fn builder() -> super::builder::feature_online_store_service::ClientBuilder {
crate::new_client_builder(super::builder::feature_online_store_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::FeatureOnlineStoreService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<
std::sync::Arc<dyn super::stub::dynamic::FeatureOnlineStoreService>,
> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::FeatureOnlineStoreService> {
super::transport::FeatureOnlineStoreService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::FeatureOnlineStoreService> {
Self::build_transport(conf)
.await
.map(super::tracing::FeatureOnlineStoreService::new)
}
pub fn fetch_feature_values(
&self,
) -> super::builder::feature_online_store_service::FetchFeatureValues {
super::builder::feature_online_store_service::FetchFeatureValues::new(self.inner.clone())
}
pub fn search_nearest_entities(
&self,
) -> super::builder::feature_online_store_service::SearchNearestEntities {
super::builder::feature_online_store_service::SearchNearestEntities::new(self.inner.clone())
}
pub fn generate_fetch_access_token(
&self,
) -> super::builder::feature_online_store_service::GenerateFetchAccessToken {
super::builder::feature_online_store_service::GenerateFetchAccessToken::new(
self.inner.clone(),
)
}
pub fn list_locations(&self) -> super::builder::feature_online_store_service::ListLocations {
super::builder::feature_online_store_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::feature_online_store_service::GetLocation {
super::builder::feature_online_store_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::feature_online_store_service::SetIamPolicy {
super::builder::feature_online_store_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::feature_online_store_service::GetIamPolicy {
super::builder::feature_online_store_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::feature_online_store_service::TestIamPermissions {
super::builder::feature_online_store_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::feature_online_store_service::ListOperations {
super::builder::feature_online_store_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::feature_online_store_service::GetOperation {
super::builder::feature_online_store_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(
&self,
) -> super::builder::feature_online_store_service::DeleteOperation {
super::builder::feature_online_store_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(
&self,
) -> super::builder::feature_online_store_service::CancelOperation {
super::builder::feature_online_store_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::feature_online_store_service::WaitOperation {
super::builder::feature_online_store_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "feature-registry-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "feature-registry-service")))]
#[derive(Clone, Debug)]
pub struct FeatureRegistryService {
inner: std::sync::Arc<dyn super::stub::dynamic::FeatureRegistryService>,
}
#[cfg(feature = "feature-registry-service")]
impl FeatureRegistryService {
pub fn builder() -> super::builder::feature_registry_service::ClientBuilder {
crate::new_client_builder(super::builder::feature_registry_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::FeatureRegistryService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::FeatureRegistryService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::FeatureRegistryService> {
super::transport::FeatureRegistryService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::FeatureRegistryService> {
Self::build_transport(conf)
.await
.map(super::tracing::FeatureRegistryService::new)
}
pub fn create_feature_group(
&self,
) -> super::builder::feature_registry_service::CreateFeatureGroup {
super::builder::feature_registry_service::CreateFeatureGroup::new(self.inner.clone())
}
pub fn get_feature_group(&self) -> super::builder::feature_registry_service::GetFeatureGroup {
super::builder::feature_registry_service::GetFeatureGroup::new(self.inner.clone())
}
pub fn list_feature_groups(
&self,
) -> super::builder::feature_registry_service::ListFeatureGroups {
super::builder::feature_registry_service::ListFeatureGroups::new(self.inner.clone())
}
pub fn update_feature_group(
&self,
) -> super::builder::feature_registry_service::UpdateFeatureGroup {
super::builder::feature_registry_service::UpdateFeatureGroup::new(self.inner.clone())
}
pub fn delete_feature_group(
&self,
) -> super::builder::feature_registry_service::DeleteFeatureGroup {
super::builder::feature_registry_service::DeleteFeatureGroup::new(self.inner.clone())
}
pub fn create_feature(&self) -> super::builder::feature_registry_service::CreateFeature {
super::builder::feature_registry_service::CreateFeature::new(self.inner.clone())
}
pub fn batch_create_features(
&self,
) -> super::builder::feature_registry_service::BatchCreateFeatures {
super::builder::feature_registry_service::BatchCreateFeatures::new(self.inner.clone())
}
pub fn get_feature(&self) -> super::builder::feature_registry_service::GetFeature {
super::builder::feature_registry_service::GetFeature::new(self.inner.clone())
}
pub fn list_features(&self) -> super::builder::feature_registry_service::ListFeatures {
super::builder::feature_registry_service::ListFeatures::new(self.inner.clone())
}
pub fn update_feature(&self) -> super::builder::feature_registry_service::UpdateFeature {
super::builder::feature_registry_service::UpdateFeature::new(self.inner.clone())
}
pub fn delete_feature(&self) -> super::builder::feature_registry_service::DeleteFeature {
super::builder::feature_registry_service::DeleteFeature::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::feature_registry_service::ListLocations {
super::builder::feature_registry_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::feature_registry_service::GetLocation {
super::builder::feature_registry_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::feature_registry_service::SetIamPolicy {
super::builder::feature_registry_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::feature_registry_service::GetIamPolicy {
super::builder::feature_registry_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::feature_registry_service::TestIamPermissions {
super::builder::feature_registry_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::feature_registry_service::ListOperations {
super::builder::feature_registry_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::feature_registry_service::GetOperation {
super::builder::feature_registry_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::feature_registry_service::DeleteOperation {
super::builder::feature_registry_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::feature_registry_service::CancelOperation {
super::builder::feature_registry_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::feature_registry_service::WaitOperation {
super::builder::feature_registry_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "featurestore-online-serving-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "featurestore-online-serving-service")))]
#[derive(Clone, Debug)]
pub struct FeaturestoreOnlineServingService {
inner: std::sync::Arc<dyn super::stub::dynamic::FeaturestoreOnlineServingService>,
}
#[cfg(feature = "featurestore-online-serving-service")]
impl FeaturestoreOnlineServingService {
pub fn builder() -> super::builder::featurestore_online_serving_service::ClientBuilder {
crate::new_client_builder(
super::builder::featurestore_online_serving_service::client::Factory,
)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::FeaturestoreOnlineServingService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<
std::sync::Arc<dyn super::stub::dynamic::FeaturestoreOnlineServingService>,
> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::FeaturestoreOnlineServingService> {
super::transport::FeaturestoreOnlineServingService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::FeaturestoreOnlineServingService> {
Self::build_transport(conf)
.await
.map(super::tracing::FeaturestoreOnlineServingService::new)
}
pub fn read_feature_values(
&self,
) -> super::builder::featurestore_online_serving_service::ReadFeatureValues {
super::builder::featurestore_online_serving_service::ReadFeatureValues::new(
self.inner.clone(),
)
}
pub fn write_feature_values(
&self,
) -> super::builder::featurestore_online_serving_service::WriteFeatureValues {
super::builder::featurestore_online_serving_service::WriteFeatureValues::new(
self.inner.clone(),
)
}
pub fn list_locations(
&self,
) -> super::builder::featurestore_online_serving_service::ListLocations {
super::builder::featurestore_online_serving_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::featurestore_online_serving_service::GetLocation {
super::builder::featurestore_online_serving_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(
&self,
) -> super::builder::featurestore_online_serving_service::SetIamPolicy {
super::builder::featurestore_online_serving_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(
&self,
) -> super::builder::featurestore_online_serving_service::GetIamPolicy {
super::builder::featurestore_online_serving_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::featurestore_online_serving_service::TestIamPermissions {
super::builder::featurestore_online_serving_service::TestIamPermissions::new(
self.inner.clone(),
)
}
pub fn list_operations(
&self,
) -> super::builder::featurestore_online_serving_service::ListOperations {
super::builder::featurestore_online_serving_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(
&self,
) -> super::builder::featurestore_online_serving_service::GetOperation {
super::builder::featurestore_online_serving_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(
&self,
) -> super::builder::featurestore_online_serving_service::DeleteOperation {
super::builder::featurestore_online_serving_service::DeleteOperation::new(
self.inner.clone(),
)
}
pub fn cancel_operation(
&self,
) -> super::builder::featurestore_online_serving_service::CancelOperation {
super::builder::featurestore_online_serving_service::CancelOperation::new(
self.inner.clone(),
)
}
pub fn wait_operation(
&self,
) -> super::builder::featurestore_online_serving_service::WaitOperation {
super::builder::featurestore_online_serving_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "featurestore-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "featurestore-service")))]
#[derive(Clone, Debug)]
pub struct FeaturestoreService {
inner: std::sync::Arc<dyn super::stub::dynamic::FeaturestoreService>,
}
#[cfg(feature = "featurestore-service")]
impl FeaturestoreService {
pub fn builder() -> super::builder::featurestore_service::ClientBuilder {
crate::new_client_builder(super::builder::featurestore_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::FeaturestoreService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::FeaturestoreService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::FeaturestoreService> {
super::transport::FeaturestoreService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::FeaturestoreService> {
Self::build_transport(conf)
.await
.map(super::tracing::FeaturestoreService::new)
}
pub fn create_featurestore(&self) -> super::builder::featurestore_service::CreateFeaturestore {
super::builder::featurestore_service::CreateFeaturestore::new(self.inner.clone())
}
pub fn get_featurestore(&self) -> super::builder::featurestore_service::GetFeaturestore {
super::builder::featurestore_service::GetFeaturestore::new(self.inner.clone())
}
pub fn list_featurestores(&self) -> super::builder::featurestore_service::ListFeaturestores {
super::builder::featurestore_service::ListFeaturestores::new(self.inner.clone())
}
pub fn update_featurestore(&self) -> super::builder::featurestore_service::UpdateFeaturestore {
super::builder::featurestore_service::UpdateFeaturestore::new(self.inner.clone())
}
pub fn delete_featurestore(&self) -> super::builder::featurestore_service::DeleteFeaturestore {
super::builder::featurestore_service::DeleteFeaturestore::new(self.inner.clone())
}
pub fn create_entity_type(&self) -> super::builder::featurestore_service::CreateEntityType {
super::builder::featurestore_service::CreateEntityType::new(self.inner.clone())
}
pub fn get_entity_type(&self) -> super::builder::featurestore_service::GetEntityType {
super::builder::featurestore_service::GetEntityType::new(self.inner.clone())
}
pub fn list_entity_types(&self) -> super::builder::featurestore_service::ListEntityTypes {
super::builder::featurestore_service::ListEntityTypes::new(self.inner.clone())
}
pub fn update_entity_type(&self) -> super::builder::featurestore_service::UpdateEntityType {
super::builder::featurestore_service::UpdateEntityType::new(self.inner.clone())
}
pub fn delete_entity_type(&self) -> super::builder::featurestore_service::DeleteEntityType {
super::builder::featurestore_service::DeleteEntityType::new(self.inner.clone())
}
pub fn create_feature(&self) -> super::builder::featurestore_service::CreateFeature {
super::builder::featurestore_service::CreateFeature::new(self.inner.clone())
}
pub fn batch_create_features(
&self,
) -> super::builder::featurestore_service::BatchCreateFeatures {
super::builder::featurestore_service::BatchCreateFeatures::new(self.inner.clone())
}
pub fn get_feature(&self) -> super::builder::featurestore_service::GetFeature {
super::builder::featurestore_service::GetFeature::new(self.inner.clone())
}
pub fn list_features(&self) -> super::builder::featurestore_service::ListFeatures {
super::builder::featurestore_service::ListFeatures::new(self.inner.clone())
}
pub fn update_feature(&self) -> super::builder::featurestore_service::UpdateFeature {
super::builder::featurestore_service::UpdateFeature::new(self.inner.clone())
}
pub fn delete_feature(&self) -> super::builder::featurestore_service::DeleteFeature {
super::builder::featurestore_service::DeleteFeature::new(self.inner.clone())
}
pub fn import_feature_values(
&self,
) -> super::builder::featurestore_service::ImportFeatureValues {
super::builder::featurestore_service::ImportFeatureValues::new(self.inner.clone())
}
pub fn batch_read_feature_values(
&self,
) -> super::builder::featurestore_service::BatchReadFeatureValues {
super::builder::featurestore_service::BatchReadFeatureValues::new(self.inner.clone())
}
pub fn export_feature_values(
&self,
) -> super::builder::featurestore_service::ExportFeatureValues {
super::builder::featurestore_service::ExportFeatureValues::new(self.inner.clone())
}
pub fn delete_feature_values(
&self,
) -> super::builder::featurestore_service::DeleteFeatureValues {
super::builder::featurestore_service::DeleteFeatureValues::new(self.inner.clone())
}
pub fn search_features(&self) -> super::builder::featurestore_service::SearchFeatures {
super::builder::featurestore_service::SearchFeatures::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::featurestore_service::ListLocations {
super::builder::featurestore_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::featurestore_service::GetLocation {
super::builder::featurestore_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::featurestore_service::SetIamPolicy {
super::builder::featurestore_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::featurestore_service::GetIamPolicy {
super::builder::featurestore_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::featurestore_service::TestIamPermissions {
super::builder::featurestore_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::featurestore_service::ListOperations {
super::builder::featurestore_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::featurestore_service::GetOperation {
super::builder::featurestore_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::featurestore_service::DeleteOperation {
super::builder::featurestore_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::featurestore_service::CancelOperation {
super::builder::featurestore_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::featurestore_service::WaitOperation {
super::builder::featurestore_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "gen-ai-cache-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "gen-ai-cache-service")))]
#[derive(Clone, Debug)]
pub struct GenAiCacheService {
inner: std::sync::Arc<dyn super::stub::dynamic::GenAiCacheService>,
}
#[cfg(feature = "gen-ai-cache-service")]
impl GenAiCacheService {
pub fn builder() -> super::builder::gen_ai_cache_service::ClientBuilder {
crate::new_client_builder(super::builder::gen_ai_cache_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::GenAiCacheService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::GenAiCacheService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::GenAiCacheService> {
super::transport::GenAiCacheService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::GenAiCacheService> {
Self::build_transport(conf)
.await
.map(super::tracing::GenAiCacheService::new)
}
pub fn create_cached_content(
&self,
) -> super::builder::gen_ai_cache_service::CreateCachedContent {
super::builder::gen_ai_cache_service::CreateCachedContent::new(self.inner.clone())
}
pub fn get_cached_content(&self) -> super::builder::gen_ai_cache_service::GetCachedContent {
super::builder::gen_ai_cache_service::GetCachedContent::new(self.inner.clone())
}
pub fn update_cached_content(
&self,
) -> super::builder::gen_ai_cache_service::UpdateCachedContent {
super::builder::gen_ai_cache_service::UpdateCachedContent::new(self.inner.clone())
}
pub fn delete_cached_content(
&self,
) -> super::builder::gen_ai_cache_service::DeleteCachedContent {
super::builder::gen_ai_cache_service::DeleteCachedContent::new(self.inner.clone())
}
pub fn list_cached_contents(&self) -> super::builder::gen_ai_cache_service::ListCachedContents {
super::builder::gen_ai_cache_service::ListCachedContents::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::gen_ai_cache_service::ListLocations {
super::builder::gen_ai_cache_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::gen_ai_cache_service::GetLocation {
super::builder::gen_ai_cache_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::gen_ai_cache_service::SetIamPolicy {
super::builder::gen_ai_cache_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::gen_ai_cache_service::GetIamPolicy {
super::builder::gen_ai_cache_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::gen_ai_cache_service::TestIamPermissions {
super::builder::gen_ai_cache_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::gen_ai_cache_service::ListOperations {
super::builder::gen_ai_cache_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::gen_ai_cache_service::GetOperation {
super::builder::gen_ai_cache_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::gen_ai_cache_service::DeleteOperation {
super::builder::gen_ai_cache_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::gen_ai_cache_service::CancelOperation {
super::builder::gen_ai_cache_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::gen_ai_cache_service::WaitOperation {
super::builder::gen_ai_cache_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "gen-ai-tuning-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "gen-ai-tuning-service")))]
#[derive(Clone, Debug)]
pub struct GenAiTuningService {
inner: std::sync::Arc<dyn super::stub::dynamic::GenAiTuningService>,
}
#[cfg(feature = "gen-ai-tuning-service")]
impl GenAiTuningService {
pub fn builder() -> super::builder::gen_ai_tuning_service::ClientBuilder {
crate::new_client_builder(super::builder::gen_ai_tuning_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::GenAiTuningService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::GenAiTuningService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::GenAiTuningService> {
super::transport::GenAiTuningService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::GenAiTuningService> {
Self::build_transport(conf)
.await
.map(super::tracing::GenAiTuningService::new)
}
pub fn create_tuning_job(&self) -> super::builder::gen_ai_tuning_service::CreateTuningJob {
super::builder::gen_ai_tuning_service::CreateTuningJob::new(self.inner.clone())
}
pub fn get_tuning_job(&self) -> super::builder::gen_ai_tuning_service::GetTuningJob {
super::builder::gen_ai_tuning_service::GetTuningJob::new(self.inner.clone())
}
pub fn list_tuning_jobs(&self) -> super::builder::gen_ai_tuning_service::ListTuningJobs {
super::builder::gen_ai_tuning_service::ListTuningJobs::new(self.inner.clone())
}
pub fn cancel_tuning_job(&self) -> super::builder::gen_ai_tuning_service::CancelTuningJob {
super::builder::gen_ai_tuning_service::CancelTuningJob::new(self.inner.clone())
}
pub fn rebase_tuned_model(&self) -> super::builder::gen_ai_tuning_service::RebaseTunedModel {
super::builder::gen_ai_tuning_service::RebaseTunedModel::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::gen_ai_tuning_service::ListLocations {
super::builder::gen_ai_tuning_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::gen_ai_tuning_service::GetLocation {
super::builder::gen_ai_tuning_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::gen_ai_tuning_service::SetIamPolicy {
super::builder::gen_ai_tuning_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::gen_ai_tuning_service::GetIamPolicy {
super::builder::gen_ai_tuning_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::gen_ai_tuning_service::TestIamPermissions {
super::builder::gen_ai_tuning_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::gen_ai_tuning_service::ListOperations {
super::builder::gen_ai_tuning_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::gen_ai_tuning_service::GetOperation {
super::builder::gen_ai_tuning_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::gen_ai_tuning_service::DeleteOperation {
super::builder::gen_ai_tuning_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::gen_ai_tuning_service::CancelOperation {
super::builder::gen_ai_tuning_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::gen_ai_tuning_service::WaitOperation {
super::builder::gen_ai_tuning_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "index-endpoint-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "index-endpoint-service")))]
#[derive(Clone, Debug)]
pub struct IndexEndpointService {
inner: std::sync::Arc<dyn super::stub::dynamic::IndexEndpointService>,
}
#[cfg(feature = "index-endpoint-service")]
impl IndexEndpointService {
pub fn builder() -> super::builder::index_endpoint_service::ClientBuilder {
crate::new_client_builder(super::builder::index_endpoint_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::IndexEndpointService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::IndexEndpointService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::IndexEndpointService> {
super::transport::IndexEndpointService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::IndexEndpointService> {
Self::build_transport(conf)
.await
.map(super::tracing::IndexEndpointService::new)
}
pub fn create_index_endpoint(
&self,
) -> super::builder::index_endpoint_service::CreateIndexEndpoint {
super::builder::index_endpoint_service::CreateIndexEndpoint::new(self.inner.clone())
}
pub fn get_index_endpoint(&self) -> super::builder::index_endpoint_service::GetIndexEndpoint {
super::builder::index_endpoint_service::GetIndexEndpoint::new(self.inner.clone())
}
pub fn list_index_endpoints(
&self,
) -> super::builder::index_endpoint_service::ListIndexEndpoints {
super::builder::index_endpoint_service::ListIndexEndpoints::new(self.inner.clone())
}
pub fn update_index_endpoint(
&self,
) -> super::builder::index_endpoint_service::UpdateIndexEndpoint {
super::builder::index_endpoint_service::UpdateIndexEndpoint::new(self.inner.clone())
}
pub fn delete_index_endpoint(
&self,
) -> super::builder::index_endpoint_service::DeleteIndexEndpoint {
super::builder::index_endpoint_service::DeleteIndexEndpoint::new(self.inner.clone())
}
pub fn deploy_index(&self) -> super::builder::index_endpoint_service::DeployIndex {
super::builder::index_endpoint_service::DeployIndex::new(self.inner.clone())
}
pub fn undeploy_index(&self) -> super::builder::index_endpoint_service::UndeployIndex {
super::builder::index_endpoint_service::UndeployIndex::new(self.inner.clone())
}
pub fn mutate_deployed_index(
&self,
) -> super::builder::index_endpoint_service::MutateDeployedIndex {
super::builder::index_endpoint_service::MutateDeployedIndex::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::index_endpoint_service::ListLocations {
super::builder::index_endpoint_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::index_endpoint_service::GetLocation {
super::builder::index_endpoint_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::index_endpoint_service::SetIamPolicy {
super::builder::index_endpoint_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::index_endpoint_service::GetIamPolicy {
super::builder::index_endpoint_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::index_endpoint_service::TestIamPermissions {
super::builder::index_endpoint_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::index_endpoint_service::ListOperations {
super::builder::index_endpoint_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::index_endpoint_service::GetOperation {
super::builder::index_endpoint_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::index_endpoint_service::DeleteOperation {
super::builder::index_endpoint_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::index_endpoint_service::CancelOperation {
super::builder::index_endpoint_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::index_endpoint_service::WaitOperation {
super::builder::index_endpoint_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "index-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "index-service")))]
#[derive(Clone, Debug)]
pub struct IndexService {
inner: std::sync::Arc<dyn super::stub::dynamic::IndexService>,
}
#[cfg(feature = "index-service")]
impl IndexService {
pub fn builder() -> super::builder::index_service::ClientBuilder {
crate::new_client_builder(super::builder::index_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::IndexService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::IndexService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::IndexService> {
super::transport::IndexService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::IndexService> {
Self::build_transport(conf)
.await
.map(super::tracing::IndexService::new)
}
pub fn create_index(&self) -> super::builder::index_service::CreateIndex {
super::builder::index_service::CreateIndex::new(self.inner.clone())
}
pub fn get_index(&self) -> super::builder::index_service::GetIndex {
super::builder::index_service::GetIndex::new(self.inner.clone())
}
pub fn list_indexes(&self) -> super::builder::index_service::ListIndexes {
super::builder::index_service::ListIndexes::new(self.inner.clone())
}
pub fn update_index(&self) -> super::builder::index_service::UpdateIndex {
super::builder::index_service::UpdateIndex::new(self.inner.clone())
}
pub fn delete_index(&self) -> super::builder::index_service::DeleteIndex {
super::builder::index_service::DeleteIndex::new(self.inner.clone())
}
pub fn upsert_datapoints(&self) -> super::builder::index_service::UpsertDatapoints {
super::builder::index_service::UpsertDatapoints::new(self.inner.clone())
}
pub fn remove_datapoints(&self) -> super::builder::index_service::RemoveDatapoints {
super::builder::index_service::RemoveDatapoints::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::index_service::ListLocations {
super::builder::index_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::index_service::GetLocation {
super::builder::index_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::index_service::SetIamPolicy {
super::builder::index_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::index_service::GetIamPolicy {
super::builder::index_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::index_service::TestIamPermissions {
super::builder::index_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::index_service::ListOperations {
super::builder::index_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::index_service::GetOperation {
super::builder::index_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::index_service::DeleteOperation {
super::builder::index_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::index_service::CancelOperation {
super::builder::index_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::index_service::WaitOperation {
super::builder::index_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "job-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "job-service")))]
#[derive(Clone, Debug)]
pub struct JobService {
inner: std::sync::Arc<dyn super::stub::dynamic::JobService>,
}
#[cfg(feature = "job-service")]
impl JobService {
pub fn builder() -> super::builder::job_service::ClientBuilder {
crate::new_client_builder(super::builder::job_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::JobService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::JobService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::JobService> {
super::transport::JobService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::JobService> {
Self::build_transport(conf)
.await
.map(super::tracing::JobService::new)
}
pub fn create_custom_job(&self) -> super::builder::job_service::CreateCustomJob {
super::builder::job_service::CreateCustomJob::new(self.inner.clone())
}
pub fn get_custom_job(&self) -> super::builder::job_service::GetCustomJob {
super::builder::job_service::GetCustomJob::new(self.inner.clone())
}
pub fn list_custom_jobs(&self) -> super::builder::job_service::ListCustomJobs {
super::builder::job_service::ListCustomJobs::new(self.inner.clone())
}
pub fn delete_custom_job(&self) -> super::builder::job_service::DeleteCustomJob {
super::builder::job_service::DeleteCustomJob::new(self.inner.clone())
}
pub fn cancel_custom_job(&self) -> super::builder::job_service::CancelCustomJob {
super::builder::job_service::CancelCustomJob::new(self.inner.clone())
}
pub fn create_data_labeling_job(&self) -> super::builder::job_service::CreateDataLabelingJob {
super::builder::job_service::CreateDataLabelingJob::new(self.inner.clone())
}
pub fn get_data_labeling_job(&self) -> super::builder::job_service::GetDataLabelingJob {
super::builder::job_service::GetDataLabelingJob::new(self.inner.clone())
}
pub fn list_data_labeling_jobs(&self) -> super::builder::job_service::ListDataLabelingJobs {
super::builder::job_service::ListDataLabelingJobs::new(self.inner.clone())
}
pub fn delete_data_labeling_job(&self) -> super::builder::job_service::DeleteDataLabelingJob {
super::builder::job_service::DeleteDataLabelingJob::new(self.inner.clone())
}
pub fn cancel_data_labeling_job(&self) -> super::builder::job_service::CancelDataLabelingJob {
super::builder::job_service::CancelDataLabelingJob::new(self.inner.clone())
}
pub fn create_hyperparameter_tuning_job(
&self,
) -> super::builder::job_service::CreateHyperparameterTuningJob {
super::builder::job_service::CreateHyperparameterTuningJob::new(self.inner.clone())
}
pub fn get_hyperparameter_tuning_job(
&self,
) -> super::builder::job_service::GetHyperparameterTuningJob {
super::builder::job_service::GetHyperparameterTuningJob::new(self.inner.clone())
}
pub fn list_hyperparameter_tuning_jobs(
&self,
) -> super::builder::job_service::ListHyperparameterTuningJobs {
super::builder::job_service::ListHyperparameterTuningJobs::new(self.inner.clone())
}
pub fn delete_hyperparameter_tuning_job(
&self,
) -> super::builder::job_service::DeleteHyperparameterTuningJob {
super::builder::job_service::DeleteHyperparameterTuningJob::new(self.inner.clone())
}
pub fn cancel_hyperparameter_tuning_job(
&self,
) -> super::builder::job_service::CancelHyperparameterTuningJob {
super::builder::job_service::CancelHyperparameterTuningJob::new(self.inner.clone())
}
pub fn create_nas_job(&self) -> super::builder::job_service::CreateNasJob {
super::builder::job_service::CreateNasJob::new(self.inner.clone())
}
pub fn get_nas_job(&self) -> super::builder::job_service::GetNasJob {
super::builder::job_service::GetNasJob::new(self.inner.clone())
}
pub fn list_nas_jobs(&self) -> super::builder::job_service::ListNasJobs {
super::builder::job_service::ListNasJobs::new(self.inner.clone())
}
pub fn delete_nas_job(&self) -> super::builder::job_service::DeleteNasJob {
super::builder::job_service::DeleteNasJob::new(self.inner.clone())
}
pub fn cancel_nas_job(&self) -> super::builder::job_service::CancelNasJob {
super::builder::job_service::CancelNasJob::new(self.inner.clone())
}
pub fn get_nas_trial_detail(&self) -> super::builder::job_service::GetNasTrialDetail {
super::builder::job_service::GetNasTrialDetail::new(self.inner.clone())
}
pub fn list_nas_trial_details(&self) -> super::builder::job_service::ListNasTrialDetails {
super::builder::job_service::ListNasTrialDetails::new(self.inner.clone())
}
pub fn create_batch_prediction_job(
&self,
) -> super::builder::job_service::CreateBatchPredictionJob {
super::builder::job_service::CreateBatchPredictionJob::new(self.inner.clone())
}
pub fn get_batch_prediction_job(&self) -> super::builder::job_service::GetBatchPredictionJob {
super::builder::job_service::GetBatchPredictionJob::new(self.inner.clone())
}
pub fn list_batch_prediction_jobs(
&self,
) -> super::builder::job_service::ListBatchPredictionJobs {
super::builder::job_service::ListBatchPredictionJobs::new(self.inner.clone())
}
pub fn delete_batch_prediction_job(
&self,
) -> super::builder::job_service::DeleteBatchPredictionJob {
super::builder::job_service::DeleteBatchPredictionJob::new(self.inner.clone())
}
pub fn cancel_batch_prediction_job(
&self,
) -> super::builder::job_service::CancelBatchPredictionJob {
super::builder::job_service::CancelBatchPredictionJob::new(self.inner.clone())
}
pub fn create_model_deployment_monitoring_job(
&self,
) -> super::builder::job_service::CreateModelDeploymentMonitoringJob {
super::builder::job_service::CreateModelDeploymentMonitoringJob::new(self.inner.clone())
}
pub fn search_model_deployment_monitoring_stats_anomalies(
&self,
) -> super::builder::job_service::SearchModelDeploymentMonitoringStatsAnomalies {
super::builder::job_service::SearchModelDeploymentMonitoringStatsAnomalies::new(
self.inner.clone(),
)
}
pub fn get_model_deployment_monitoring_job(
&self,
) -> super::builder::job_service::GetModelDeploymentMonitoringJob {
super::builder::job_service::GetModelDeploymentMonitoringJob::new(self.inner.clone())
}
pub fn list_model_deployment_monitoring_jobs(
&self,
) -> super::builder::job_service::ListModelDeploymentMonitoringJobs {
super::builder::job_service::ListModelDeploymentMonitoringJobs::new(self.inner.clone())
}
pub fn update_model_deployment_monitoring_job(
&self,
) -> super::builder::job_service::UpdateModelDeploymentMonitoringJob {
super::builder::job_service::UpdateModelDeploymentMonitoringJob::new(self.inner.clone())
}
pub fn delete_model_deployment_monitoring_job(
&self,
) -> super::builder::job_service::DeleteModelDeploymentMonitoringJob {
super::builder::job_service::DeleteModelDeploymentMonitoringJob::new(self.inner.clone())
}
pub fn pause_model_deployment_monitoring_job(
&self,
) -> super::builder::job_service::PauseModelDeploymentMonitoringJob {
super::builder::job_service::PauseModelDeploymentMonitoringJob::new(self.inner.clone())
}
pub fn resume_model_deployment_monitoring_job(
&self,
) -> super::builder::job_service::ResumeModelDeploymentMonitoringJob {
super::builder::job_service::ResumeModelDeploymentMonitoringJob::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::job_service::ListLocations {
super::builder::job_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::job_service::GetLocation {
super::builder::job_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::job_service::SetIamPolicy {
super::builder::job_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::job_service::GetIamPolicy {
super::builder::job_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::job_service::TestIamPermissions {
super::builder::job_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::job_service::ListOperations {
super::builder::job_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::job_service::GetOperation {
super::builder::job_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::job_service::DeleteOperation {
super::builder::job_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::job_service::CancelOperation {
super::builder::job_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::job_service::WaitOperation {
super::builder::job_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "llm-utility-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "llm-utility-service")))]
#[derive(Clone, Debug)]
pub struct LlmUtilityService {
inner: std::sync::Arc<dyn super::stub::dynamic::LlmUtilityService>,
}
#[cfg(feature = "llm-utility-service")]
impl LlmUtilityService {
pub fn builder() -> super::builder::llm_utility_service::ClientBuilder {
crate::new_client_builder(super::builder::llm_utility_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::LlmUtilityService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::LlmUtilityService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::LlmUtilityService> {
super::transport::LlmUtilityService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::LlmUtilityService> {
Self::build_transport(conf)
.await
.map(super::tracing::LlmUtilityService::new)
}
pub fn count_tokens(&self) -> super::builder::llm_utility_service::CountTokens {
super::builder::llm_utility_service::CountTokens::new(self.inner.clone())
}
pub fn compute_tokens(&self) -> super::builder::llm_utility_service::ComputeTokens {
super::builder::llm_utility_service::ComputeTokens::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::llm_utility_service::ListLocations {
super::builder::llm_utility_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::llm_utility_service::GetLocation {
super::builder::llm_utility_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::llm_utility_service::SetIamPolicy {
super::builder::llm_utility_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::llm_utility_service::GetIamPolicy {
super::builder::llm_utility_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::llm_utility_service::TestIamPermissions {
super::builder::llm_utility_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::llm_utility_service::ListOperations {
super::builder::llm_utility_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::llm_utility_service::GetOperation {
super::builder::llm_utility_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::llm_utility_service::DeleteOperation {
super::builder::llm_utility_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::llm_utility_service::CancelOperation {
super::builder::llm_utility_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::llm_utility_service::WaitOperation {
super::builder::llm_utility_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "match-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "match-service")))]
#[derive(Clone, Debug)]
pub struct MatchService {
inner: std::sync::Arc<dyn super::stub::dynamic::MatchService>,
}
#[cfg(feature = "match-service")]
impl MatchService {
pub fn builder() -> super::builder::match_service::ClientBuilder {
crate::new_client_builder(super::builder::match_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::MatchService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::MatchService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::MatchService> {
super::transport::MatchService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::MatchService> {
Self::build_transport(conf)
.await
.map(super::tracing::MatchService::new)
}
pub fn find_neighbors(&self) -> super::builder::match_service::FindNeighbors {
super::builder::match_service::FindNeighbors::new(self.inner.clone())
}
pub fn read_index_datapoints(&self) -> super::builder::match_service::ReadIndexDatapoints {
super::builder::match_service::ReadIndexDatapoints::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::match_service::ListLocations {
super::builder::match_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::match_service::GetLocation {
super::builder::match_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::match_service::SetIamPolicy {
super::builder::match_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::match_service::GetIamPolicy {
super::builder::match_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::match_service::TestIamPermissions {
super::builder::match_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::match_service::ListOperations {
super::builder::match_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::match_service::GetOperation {
super::builder::match_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::match_service::DeleteOperation {
super::builder::match_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::match_service::CancelOperation {
super::builder::match_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::match_service::WaitOperation {
super::builder::match_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "metadata-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "metadata-service")))]
#[derive(Clone, Debug)]
pub struct MetadataService {
inner: std::sync::Arc<dyn super::stub::dynamic::MetadataService>,
}
#[cfg(feature = "metadata-service")]
impl MetadataService {
pub fn builder() -> super::builder::metadata_service::ClientBuilder {
crate::new_client_builder(super::builder::metadata_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::MetadataService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::MetadataService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::MetadataService> {
super::transport::MetadataService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::MetadataService> {
Self::build_transport(conf)
.await
.map(super::tracing::MetadataService::new)
}
pub fn create_metadata_store(&self) -> super::builder::metadata_service::CreateMetadataStore {
super::builder::metadata_service::CreateMetadataStore::new(self.inner.clone())
}
pub fn get_metadata_store(&self) -> super::builder::metadata_service::GetMetadataStore {
super::builder::metadata_service::GetMetadataStore::new(self.inner.clone())
}
pub fn list_metadata_stores(&self) -> super::builder::metadata_service::ListMetadataStores {
super::builder::metadata_service::ListMetadataStores::new(self.inner.clone())
}
pub fn delete_metadata_store(&self) -> super::builder::metadata_service::DeleteMetadataStore {
super::builder::metadata_service::DeleteMetadataStore::new(self.inner.clone())
}
pub fn create_artifact(&self) -> super::builder::metadata_service::CreateArtifact {
super::builder::metadata_service::CreateArtifact::new(self.inner.clone())
}
pub fn get_artifact(&self) -> super::builder::metadata_service::GetArtifact {
super::builder::metadata_service::GetArtifact::new(self.inner.clone())
}
pub fn list_artifacts(&self) -> super::builder::metadata_service::ListArtifacts {
super::builder::metadata_service::ListArtifacts::new(self.inner.clone())
}
pub fn update_artifact(&self) -> super::builder::metadata_service::UpdateArtifact {
super::builder::metadata_service::UpdateArtifact::new(self.inner.clone())
}
pub fn delete_artifact(&self) -> super::builder::metadata_service::DeleteArtifact {
super::builder::metadata_service::DeleteArtifact::new(self.inner.clone())
}
pub fn purge_artifacts(&self) -> super::builder::metadata_service::PurgeArtifacts {
super::builder::metadata_service::PurgeArtifacts::new(self.inner.clone())
}
pub fn create_context(&self) -> super::builder::metadata_service::CreateContext {
super::builder::metadata_service::CreateContext::new(self.inner.clone())
}
pub fn get_context(&self) -> super::builder::metadata_service::GetContext {
super::builder::metadata_service::GetContext::new(self.inner.clone())
}
pub fn list_contexts(&self) -> super::builder::metadata_service::ListContexts {
super::builder::metadata_service::ListContexts::new(self.inner.clone())
}
pub fn update_context(&self) -> super::builder::metadata_service::UpdateContext {
super::builder::metadata_service::UpdateContext::new(self.inner.clone())
}
pub fn delete_context(&self) -> super::builder::metadata_service::DeleteContext {
super::builder::metadata_service::DeleteContext::new(self.inner.clone())
}
pub fn purge_contexts(&self) -> super::builder::metadata_service::PurgeContexts {
super::builder::metadata_service::PurgeContexts::new(self.inner.clone())
}
pub fn add_context_artifacts_and_executions(
&self,
) -> super::builder::metadata_service::AddContextArtifactsAndExecutions {
super::builder::metadata_service::AddContextArtifactsAndExecutions::new(self.inner.clone())
}
pub fn add_context_children(&self) -> super::builder::metadata_service::AddContextChildren {
super::builder::metadata_service::AddContextChildren::new(self.inner.clone())
}
pub fn remove_context_children(
&self,
) -> super::builder::metadata_service::RemoveContextChildren {
super::builder::metadata_service::RemoveContextChildren::new(self.inner.clone())
}
pub fn query_context_lineage_subgraph(
&self,
) -> super::builder::metadata_service::QueryContextLineageSubgraph {
super::builder::metadata_service::QueryContextLineageSubgraph::new(self.inner.clone())
}
pub fn create_execution(&self) -> super::builder::metadata_service::CreateExecution {
super::builder::metadata_service::CreateExecution::new(self.inner.clone())
}
pub fn get_execution(&self) -> super::builder::metadata_service::GetExecution {
super::builder::metadata_service::GetExecution::new(self.inner.clone())
}
pub fn list_executions(&self) -> super::builder::metadata_service::ListExecutions {
super::builder::metadata_service::ListExecutions::new(self.inner.clone())
}
pub fn update_execution(&self) -> super::builder::metadata_service::UpdateExecution {
super::builder::metadata_service::UpdateExecution::new(self.inner.clone())
}
pub fn delete_execution(&self) -> super::builder::metadata_service::DeleteExecution {
super::builder::metadata_service::DeleteExecution::new(self.inner.clone())
}
pub fn purge_executions(&self) -> super::builder::metadata_service::PurgeExecutions {
super::builder::metadata_service::PurgeExecutions::new(self.inner.clone())
}
pub fn add_execution_events(&self) -> super::builder::metadata_service::AddExecutionEvents {
super::builder::metadata_service::AddExecutionEvents::new(self.inner.clone())
}
pub fn query_execution_inputs_and_outputs(
&self,
) -> super::builder::metadata_service::QueryExecutionInputsAndOutputs {
super::builder::metadata_service::QueryExecutionInputsAndOutputs::new(self.inner.clone())
}
pub fn create_metadata_schema(&self) -> super::builder::metadata_service::CreateMetadataSchema {
super::builder::metadata_service::CreateMetadataSchema::new(self.inner.clone())
}
pub fn get_metadata_schema(&self) -> super::builder::metadata_service::GetMetadataSchema {
super::builder::metadata_service::GetMetadataSchema::new(self.inner.clone())
}
pub fn list_metadata_schemas(&self) -> super::builder::metadata_service::ListMetadataSchemas {
super::builder::metadata_service::ListMetadataSchemas::new(self.inner.clone())
}
pub fn query_artifact_lineage_subgraph(
&self,
) -> super::builder::metadata_service::QueryArtifactLineageSubgraph {
super::builder::metadata_service::QueryArtifactLineageSubgraph::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::metadata_service::ListLocations {
super::builder::metadata_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::metadata_service::GetLocation {
super::builder::metadata_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::metadata_service::SetIamPolicy {
super::builder::metadata_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::metadata_service::GetIamPolicy {
super::builder::metadata_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::metadata_service::TestIamPermissions {
super::builder::metadata_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::metadata_service::ListOperations {
super::builder::metadata_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::metadata_service::GetOperation {
super::builder::metadata_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::metadata_service::DeleteOperation {
super::builder::metadata_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::metadata_service::CancelOperation {
super::builder::metadata_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::metadata_service::WaitOperation {
super::builder::metadata_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "migration-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "migration-service")))]
#[derive(Clone, Debug)]
pub struct MigrationService {
inner: std::sync::Arc<dyn super::stub::dynamic::MigrationService>,
}
#[cfg(feature = "migration-service")]
impl MigrationService {
pub fn builder() -> super::builder::migration_service::ClientBuilder {
crate::new_client_builder(super::builder::migration_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::MigrationService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::MigrationService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::MigrationService> {
super::transport::MigrationService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::MigrationService> {
Self::build_transport(conf)
.await
.map(super::tracing::MigrationService::new)
}
pub fn search_migratable_resources(
&self,
) -> super::builder::migration_service::SearchMigratableResources {
super::builder::migration_service::SearchMigratableResources::new(self.inner.clone())
}
pub fn batch_migrate_resources(
&self,
) -> super::builder::migration_service::BatchMigrateResources {
super::builder::migration_service::BatchMigrateResources::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::migration_service::ListLocations {
super::builder::migration_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::migration_service::GetLocation {
super::builder::migration_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::migration_service::SetIamPolicy {
super::builder::migration_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::migration_service::GetIamPolicy {
super::builder::migration_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::migration_service::TestIamPermissions {
super::builder::migration_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::migration_service::ListOperations {
super::builder::migration_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::migration_service::GetOperation {
super::builder::migration_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::migration_service::DeleteOperation {
super::builder::migration_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::migration_service::CancelOperation {
super::builder::migration_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::migration_service::WaitOperation {
super::builder::migration_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "model-garden-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "model-garden-service")))]
#[derive(Clone, Debug)]
pub struct ModelGardenService {
inner: std::sync::Arc<dyn super::stub::dynamic::ModelGardenService>,
}
#[cfg(feature = "model-garden-service")]
impl ModelGardenService {
pub fn builder() -> super::builder::model_garden_service::ClientBuilder {
crate::new_client_builder(super::builder::model_garden_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::ModelGardenService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::ModelGardenService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ModelGardenService> {
super::transport::ModelGardenService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ModelGardenService> {
Self::build_transport(conf)
.await
.map(super::tracing::ModelGardenService::new)
}
pub fn get_publisher_model(&self) -> super::builder::model_garden_service::GetPublisherModel {
super::builder::model_garden_service::GetPublisherModel::new(self.inner.clone())
}
pub fn deploy(&self) -> super::builder::model_garden_service::Deploy {
super::builder::model_garden_service::Deploy::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::model_garden_service::ListLocations {
super::builder::model_garden_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::model_garden_service::GetLocation {
super::builder::model_garden_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::model_garden_service::SetIamPolicy {
super::builder::model_garden_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::model_garden_service::GetIamPolicy {
super::builder::model_garden_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::model_garden_service::TestIamPermissions {
super::builder::model_garden_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::model_garden_service::ListOperations {
super::builder::model_garden_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::model_garden_service::GetOperation {
super::builder::model_garden_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::model_garden_service::DeleteOperation {
super::builder::model_garden_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::model_garden_service::CancelOperation {
super::builder::model_garden_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::model_garden_service::WaitOperation {
super::builder::model_garden_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "model-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "model-service")))]
#[derive(Clone, Debug)]
pub struct ModelService {
inner: std::sync::Arc<dyn super::stub::dynamic::ModelService>,
}
#[cfg(feature = "model-service")]
impl ModelService {
pub fn builder() -> super::builder::model_service::ClientBuilder {
crate::new_client_builder(super::builder::model_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::ModelService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::ModelService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ModelService> {
super::transport::ModelService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ModelService> {
Self::build_transport(conf)
.await
.map(super::tracing::ModelService::new)
}
pub fn upload_model(&self) -> super::builder::model_service::UploadModel {
super::builder::model_service::UploadModel::new(self.inner.clone())
}
pub fn get_model(&self) -> super::builder::model_service::GetModel {
super::builder::model_service::GetModel::new(self.inner.clone())
}
pub fn list_models(&self) -> super::builder::model_service::ListModels {
super::builder::model_service::ListModels::new(self.inner.clone())
}
pub fn list_model_versions(&self) -> super::builder::model_service::ListModelVersions {
super::builder::model_service::ListModelVersions::new(self.inner.clone())
}
pub fn list_model_version_checkpoints(
&self,
) -> super::builder::model_service::ListModelVersionCheckpoints {
super::builder::model_service::ListModelVersionCheckpoints::new(self.inner.clone())
}
pub fn update_model(&self) -> super::builder::model_service::UpdateModel {
super::builder::model_service::UpdateModel::new(self.inner.clone())
}
pub fn update_explanation_dataset(
&self,
) -> super::builder::model_service::UpdateExplanationDataset {
super::builder::model_service::UpdateExplanationDataset::new(self.inner.clone())
}
pub fn delete_model(&self) -> super::builder::model_service::DeleteModel {
super::builder::model_service::DeleteModel::new(self.inner.clone())
}
pub fn delete_model_version(&self) -> super::builder::model_service::DeleteModelVersion {
super::builder::model_service::DeleteModelVersion::new(self.inner.clone())
}
pub fn merge_version_aliases(&self) -> super::builder::model_service::MergeVersionAliases {
super::builder::model_service::MergeVersionAliases::new(self.inner.clone())
}
pub fn export_model(&self) -> super::builder::model_service::ExportModel {
super::builder::model_service::ExportModel::new(self.inner.clone())
}
pub fn copy_model(&self) -> super::builder::model_service::CopyModel {
super::builder::model_service::CopyModel::new(self.inner.clone())
}
pub fn import_model_evaluation(&self) -> super::builder::model_service::ImportModelEvaluation {
super::builder::model_service::ImportModelEvaluation::new(self.inner.clone())
}
pub fn batch_import_model_evaluation_slices(
&self,
) -> super::builder::model_service::BatchImportModelEvaluationSlices {
super::builder::model_service::BatchImportModelEvaluationSlices::new(self.inner.clone())
}
pub fn batch_import_evaluated_annotations(
&self,
) -> super::builder::model_service::BatchImportEvaluatedAnnotations {
super::builder::model_service::BatchImportEvaluatedAnnotations::new(self.inner.clone())
}
pub fn get_model_evaluation(&self) -> super::builder::model_service::GetModelEvaluation {
super::builder::model_service::GetModelEvaluation::new(self.inner.clone())
}
pub fn list_model_evaluations(&self) -> super::builder::model_service::ListModelEvaluations {
super::builder::model_service::ListModelEvaluations::new(self.inner.clone())
}
pub fn get_model_evaluation_slice(
&self,
) -> super::builder::model_service::GetModelEvaluationSlice {
super::builder::model_service::GetModelEvaluationSlice::new(self.inner.clone())
}
pub fn list_model_evaluation_slices(
&self,
) -> super::builder::model_service::ListModelEvaluationSlices {
super::builder::model_service::ListModelEvaluationSlices::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::model_service::ListLocations {
super::builder::model_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::model_service::GetLocation {
super::builder::model_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::model_service::SetIamPolicy {
super::builder::model_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::model_service::GetIamPolicy {
super::builder::model_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::model_service::TestIamPermissions {
super::builder::model_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::model_service::ListOperations {
super::builder::model_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::model_service::GetOperation {
super::builder::model_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::model_service::DeleteOperation {
super::builder::model_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::model_service::CancelOperation {
super::builder::model_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::model_service::WaitOperation {
super::builder::model_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "notebook-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "notebook-service")))]
#[derive(Clone, Debug)]
pub struct NotebookService {
inner: std::sync::Arc<dyn super::stub::dynamic::NotebookService>,
}
#[cfg(feature = "notebook-service")]
impl NotebookService {
pub fn builder() -> super::builder::notebook_service::ClientBuilder {
crate::new_client_builder(super::builder::notebook_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::NotebookService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::NotebookService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::NotebookService> {
super::transport::NotebookService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::NotebookService> {
Self::build_transport(conf)
.await
.map(super::tracing::NotebookService::new)
}
pub fn create_notebook_runtime_template(
&self,
) -> super::builder::notebook_service::CreateNotebookRuntimeTemplate {
super::builder::notebook_service::CreateNotebookRuntimeTemplate::new(self.inner.clone())
}
pub fn get_notebook_runtime_template(
&self,
) -> super::builder::notebook_service::GetNotebookRuntimeTemplate {
super::builder::notebook_service::GetNotebookRuntimeTemplate::new(self.inner.clone())
}
pub fn list_notebook_runtime_templates(
&self,
) -> super::builder::notebook_service::ListNotebookRuntimeTemplates {
super::builder::notebook_service::ListNotebookRuntimeTemplates::new(self.inner.clone())
}
pub fn delete_notebook_runtime_template(
&self,
) -> super::builder::notebook_service::DeleteNotebookRuntimeTemplate {
super::builder::notebook_service::DeleteNotebookRuntimeTemplate::new(self.inner.clone())
}
pub fn update_notebook_runtime_template(
&self,
) -> super::builder::notebook_service::UpdateNotebookRuntimeTemplate {
super::builder::notebook_service::UpdateNotebookRuntimeTemplate::new(self.inner.clone())
}
pub fn assign_notebook_runtime(
&self,
) -> super::builder::notebook_service::AssignNotebookRuntime {
super::builder::notebook_service::AssignNotebookRuntime::new(self.inner.clone())
}
pub fn get_notebook_runtime(&self) -> super::builder::notebook_service::GetNotebookRuntime {
super::builder::notebook_service::GetNotebookRuntime::new(self.inner.clone())
}
pub fn list_notebook_runtimes(&self) -> super::builder::notebook_service::ListNotebookRuntimes {
super::builder::notebook_service::ListNotebookRuntimes::new(self.inner.clone())
}
pub fn delete_notebook_runtime(
&self,
) -> super::builder::notebook_service::DeleteNotebookRuntime {
super::builder::notebook_service::DeleteNotebookRuntime::new(self.inner.clone())
}
pub fn upgrade_notebook_runtime(
&self,
) -> super::builder::notebook_service::UpgradeNotebookRuntime {
super::builder::notebook_service::UpgradeNotebookRuntime::new(self.inner.clone())
}
pub fn start_notebook_runtime(&self) -> super::builder::notebook_service::StartNotebookRuntime {
super::builder::notebook_service::StartNotebookRuntime::new(self.inner.clone())
}
pub fn stop_notebook_runtime(&self) -> super::builder::notebook_service::StopNotebookRuntime {
super::builder::notebook_service::StopNotebookRuntime::new(self.inner.clone())
}
pub fn create_notebook_execution_job(
&self,
) -> super::builder::notebook_service::CreateNotebookExecutionJob {
super::builder::notebook_service::CreateNotebookExecutionJob::new(self.inner.clone())
}
pub fn get_notebook_execution_job(
&self,
) -> super::builder::notebook_service::GetNotebookExecutionJob {
super::builder::notebook_service::GetNotebookExecutionJob::new(self.inner.clone())
}
pub fn list_notebook_execution_jobs(
&self,
) -> super::builder::notebook_service::ListNotebookExecutionJobs {
super::builder::notebook_service::ListNotebookExecutionJobs::new(self.inner.clone())
}
pub fn delete_notebook_execution_job(
&self,
) -> super::builder::notebook_service::DeleteNotebookExecutionJob {
super::builder::notebook_service::DeleteNotebookExecutionJob::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::notebook_service::ListLocations {
super::builder::notebook_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::notebook_service::GetLocation {
super::builder::notebook_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::notebook_service::SetIamPolicy {
super::builder::notebook_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::notebook_service::GetIamPolicy {
super::builder::notebook_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::notebook_service::TestIamPermissions {
super::builder::notebook_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::notebook_service::ListOperations {
super::builder::notebook_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::notebook_service::GetOperation {
super::builder::notebook_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::notebook_service::DeleteOperation {
super::builder::notebook_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::notebook_service::CancelOperation {
super::builder::notebook_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::notebook_service::WaitOperation {
super::builder::notebook_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "persistent-resource-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "persistent-resource-service")))]
#[derive(Clone, Debug)]
pub struct PersistentResourceService {
inner: std::sync::Arc<dyn super::stub::dynamic::PersistentResourceService>,
}
#[cfg(feature = "persistent-resource-service")]
impl PersistentResourceService {
pub fn builder() -> super::builder::persistent_resource_service::ClientBuilder {
crate::new_client_builder(super::builder::persistent_resource_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::PersistentResourceService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<
std::sync::Arc<dyn super::stub::dynamic::PersistentResourceService>,
> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::PersistentResourceService> {
super::transport::PersistentResourceService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::PersistentResourceService> {
Self::build_transport(conf)
.await
.map(super::tracing::PersistentResourceService::new)
}
pub fn create_persistent_resource(
&self,
) -> super::builder::persistent_resource_service::CreatePersistentResource {
super::builder::persistent_resource_service::CreatePersistentResource::new(
self.inner.clone(),
)
}
pub fn get_persistent_resource(
&self,
) -> super::builder::persistent_resource_service::GetPersistentResource {
super::builder::persistent_resource_service::GetPersistentResource::new(self.inner.clone())
}
pub fn list_persistent_resources(
&self,
) -> super::builder::persistent_resource_service::ListPersistentResources {
super::builder::persistent_resource_service::ListPersistentResources::new(
self.inner.clone(),
)
}
pub fn delete_persistent_resource(
&self,
) -> super::builder::persistent_resource_service::DeletePersistentResource {
super::builder::persistent_resource_service::DeletePersistentResource::new(
self.inner.clone(),
)
}
pub fn update_persistent_resource(
&self,
) -> super::builder::persistent_resource_service::UpdatePersistentResource {
super::builder::persistent_resource_service::UpdatePersistentResource::new(
self.inner.clone(),
)
}
pub fn reboot_persistent_resource(
&self,
) -> super::builder::persistent_resource_service::RebootPersistentResource {
super::builder::persistent_resource_service::RebootPersistentResource::new(
self.inner.clone(),
)
}
pub fn list_locations(&self) -> super::builder::persistent_resource_service::ListLocations {
super::builder::persistent_resource_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::persistent_resource_service::GetLocation {
super::builder::persistent_resource_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::persistent_resource_service::SetIamPolicy {
super::builder::persistent_resource_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::persistent_resource_service::GetIamPolicy {
super::builder::persistent_resource_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::persistent_resource_service::TestIamPermissions {
super::builder::persistent_resource_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::persistent_resource_service::ListOperations {
super::builder::persistent_resource_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::persistent_resource_service::GetOperation {
super::builder::persistent_resource_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::persistent_resource_service::DeleteOperation {
super::builder::persistent_resource_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::persistent_resource_service::CancelOperation {
super::builder::persistent_resource_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::persistent_resource_service::WaitOperation {
super::builder::persistent_resource_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "pipeline-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "pipeline-service")))]
#[derive(Clone, Debug)]
pub struct PipelineService {
inner: std::sync::Arc<dyn super::stub::dynamic::PipelineService>,
}
#[cfg(feature = "pipeline-service")]
impl PipelineService {
pub fn builder() -> super::builder::pipeline_service::ClientBuilder {
crate::new_client_builder(super::builder::pipeline_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::PipelineService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::PipelineService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::PipelineService> {
super::transport::PipelineService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::PipelineService> {
Self::build_transport(conf)
.await
.map(super::tracing::PipelineService::new)
}
pub fn create_training_pipeline(
&self,
) -> super::builder::pipeline_service::CreateTrainingPipeline {
super::builder::pipeline_service::CreateTrainingPipeline::new(self.inner.clone())
}
pub fn get_training_pipeline(&self) -> super::builder::pipeline_service::GetTrainingPipeline {
super::builder::pipeline_service::GetTrainingPipeline::new(self.inner.clone())
}
pub fn list_training_pipelines(
&self,
) -> super::builder::pipeline_service::ListTrainingPipelines {
super::builder::pipeline_service::ListTrainingPipelines::new(self.inner.clone())
}
pub fn delete_training_pipeline(
&self,
) -> super::builder::pipeline_service::DeleteTrainingPipeline {
super::builder::pipeline_service::DeleteTrainingPipeline::new(self.inner.clone())
}
pub fn cancel_training_pipeline(
&self,
) -> super::builder::pipeline_service::CancelTrainingPipeline {
super::builder::pipeline_service::CancelTrainingPipeline::new(self.inner.clone())
}
pub fn create_pipeline_job(&self) -> super::builder::pipeline_service::CreatePipelineJob {
super::builder::pipeline_service::CreatePipelineJob::new(self.inner.clone())
}
pub fn get_pipeline_job(&self) -> super::builder::pipeline_service::GetPipelineJob {
super::builder::pipeline_service::GetPipelineJob::new(self.inner.clone())
}
pub fn list_pipeline_jobs(&self) -> super::builder::pipeline_service::ListPipelineJobs {
super::builder::pipeline_service::ListPipelineJobs::new(self.inner.clone())
}
pub fn delete_pipeline_job(&self) -> super::builder::pipeline_service::DeletePipelineJob {
super::builder::pipeline_service::DeletePipelineJob::new(self.inner.clone())
}
pub fn batch_delete_pipeline_jobs(
&self,
) -> super::builder::pipeline_service::BatchDeletePipelineJobs {
super::builder::pipeline_service::BatchDeletePipelineJobs::new(self.inner.clone())
}
pub fn cancel_pipeline_job(&self) -> super::builder::pipeline_service::CancelPipelineJob {
super::builder::pipeline_service::CancelPipelineJob::new(self.inner.clone())
}
pub fn batch_cancel_pipeline_jobs(
&self,
) -> super::builder::pipeline_service::BatchCancelPipelineJobs {
super::builder::pipeline_service::BatchCancelPipelineJobs::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::pipeline_service::ListLocations {
super::builder::pipeline_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::pipeline_service::GetLocation {
super::builder::pipeline_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::pipeline_service::SetIamPolicy {
super::builder::pipeline_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::pipeline_service::GetIamPolicy {
super::builder::pipeline_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::pipeline_service::TestIamPermissions {
super::builder::pipeline_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::pipeline_service::ListOperations {
super::builder::pipeline_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::pipeline_service::GetOperation {
super::builder::pipeline_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::pipeline_service::DeleteOperation {
super::builder::pipeline_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::pipeline_service::CancelOperation {
super::builder::pipeline_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::pipeline_service::WaitOperation {
super::builder::pipeline_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "prediction-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "prediction-service")))]
#[derive(Clone, Debug)]
pub struct PredictionService {
inner: std::sync::Arc<dyn super::stub::dynamic::PredictionService>,
}
#[cfg(feature = "prediction-service")]
impl PredictionService {
pub fn builder() -> super::builder::prediction_service::ClientBuilder {
crate::new_client_builder(super::builder::prediction_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::PredictionService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::PredictionService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::PredictionService> {
super::transport::PredictionService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::PredictionService> {
Self::build_transport(conf)
.await
.map(super::tracing::PredictionService::new)
}
pub fn predict(&self) -> super::builder::prediction_service::Predict {
super::builder::prediction_service::Predict::new(self.inner.clone())
}
pub fn raw_predict(&self) -> super::builder::prediction_service::RawPredict {
super::builder::prediction_service::RawPredict::new(self.inner.clone())
}
pub fn direct_predict(&self) -> super::builder::prediction_service::DirectPredict {
super::builder::prediction_service::DirectPredict::new(self.inner.clone())
}
pub fn direct_raw_predict(&self) -> super::builder::prediction_service::DirectRawPredict {
super::builder::prediction_service::DirectRawPredict::new(self.inner.clone())
}
pub fn explain(&self) -> super::builder::prediction_service::Explain {
super::builder::prediction_service::Explain::new(self.inner.clone())
}
pub fn generate_content(&self) -> super::builder::prediction_service::GenerateContent {
super::builder::prediction_service::GenerateContent::new(self.inner.clone())
}
pub fn embed_content(&self) -> super::builder::prediction_service::EmbedContent {
super::builder::prediction_service::EmbedContent::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::prediction_service::ListLocations {
super::builder::prediction_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::prediction_service::GetLocation {
super::builder::prediction_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::prediction_service::SetIamPolicy {
super::builder::prediction_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::prediction_service::GetIamPolicy {
super::builder::prediction_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::prediction_service::TestIamPermissions {
super::builder::prediction_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::prediction_service::ListOperations {
super::builder::prediction_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::prediction_service::GetOperation {
super::builder::prediction_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::prediction_service::DeleteOperation {
super::builder::prediction_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::prediction_service::CancelOperation {
super::builder::prediction_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::prediction_service::WaitOperation {
super::builder::prediction_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "reasoning-engine-execution-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "reasoning-engine-execution-service")))]
#[derive(Clone, Debug)]
pub struct ReasoningEngineExecutionService {
inner: std::sync::Arc<dyn super::stub::dynamic::ReasoningEngineExecutionService>,
}
#[cfg(feature = "reasoning-engine-execution-service")]
impl ReasoningEngineExecutionService {
pub fn builder() -> super::builder::reasoning_engine_execution_service::ClientBuilder {
crate::new_client_builder(
super::builder::reasoning_engine_execution_service::client::Factory,
)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::ReasoningEngineExecutionService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<
std::sync::Arc<dyn super::stub::dynamic::ReasoningEngineExecutionService>,
> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ReasoningEngineExecutionService> {
super::transport::ReasoningEngineExecutionService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ReasoningEngineExecutionService> {
Self::build_transport(conf)
.await
.map(super::tracing::ReasoningEngineExecutionService::new)
}
pub fn query_reasoning_engine(
&self,
) -> super::builder::reasoning_engine_execution_service::QueryReasoningEngine {
super::builder::reasoning_engine_execution_service::QueryReasoningEngine::new(
self.inner.clone(),
)
}
pub fn list_locations(
&self,
) -> super::builder::reasoning_engine_execution_service::ListLocations {
super::builder::reasoning_engine_execution_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::reasoning_engine_execution_service::GetLocation {
super::builder::reasoning_engine_execution_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(
&self,
) -> super::builder::reasoning_engine_execution_service::SetIamPolicy {
super::builder::reasoning_engine_execution_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(
&self,
) -> super::builder::reasoning_engine_execution_service::GetIamPolicy {
super::builder::reasoning_engine_execution_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::reasoning_engine_execution_service::TestIamPermissions {
super::builder::reasoning_engine_execution_service::TestIamPermissions::new(
self.inner.clone(),
)
}
pub fn list_operations(
&self,
) -> super::builder::reasoning_engine_execution_service::ListOperations {
super::builder::reasoning_engine_execution_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(
&self,
) -> super::builder::reasoning_engine_execution_service::GetOperation {
super::builder::reasoning_engine_execution_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(
&self,
) -> super::builder::reasoning_engine_execution_service::DeleteOperation {
super::builder::reasoning_engine_execution_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(
&self,
) -> super::builder::reasoning_engine_execution_service::CancelOperation {
super::builder::reasoning_engine_execution_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(
&self,
) -> super::builder::reasoning_engine_execution_service::WaitOperation {
super::builder::reasoning_engine_execution_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "reasoning-engine-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "reasoning-engine-service")))]
#[derive(Clone, Debug)]
pub struct ReasoningEngineService {
inner: std::sync::Arc<dyn super::stub::dynamic::ReasoningEngineService>,
}
#[cfg(feature = "reasoning-engine-service")]
impl ReasoningEngineService {
pub fn builder() -> super::builder::reasoning_engine_service::ClientBuilder {
crate::new_client_builder(super::builder::reasoning_engine_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::ReasoningEngineService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::ReasoningEngineService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ReasoningEngineService> {
super::transport::ReasoningEngineService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ReasoningEngineService> {
Self::build_transport(conf)
.await
.map(super::tracing::ReasoningEngineService::new)
}
pub fn create_reasoning_engine(
&self,
) -> super::builder::reasoning_engine_service::CreateReasoningEngine {
super::builder::reasoning_engine_service::CreateReasoningEngine::new(self.inner.clone())
}
pub fn get_reasoning_engine(
&self,
) -> super::builder::reasoning_engine_service::GetReasoningEngine {
super::builder::reasoning_engine_service::GetReasoningEngine::new(self.inner.clone())
}
pub fn list_reasoning_engines(
&self,
) -> super::builder::reasoning_engine_service::ListReasoningEngines {
super::builder::reasoning_engine_service::ListReasoningEngines::new(self.inner.clone())
}
pub fn update_reasoning_engine(
&self,
) -> super::builder::reasoning_engine_service::UpdateReasoningEngine {
super::builder::reasoning_engine_service::UpdateReasoningEngine::new(self.inner.clone())
}
pub fn delete_reasoning_engine(
&self,
) -> super::builder::reasoning_engine_service::DeleteReasoningEngine {
super::builder::reasoning_engine_service::DeleteReasoningEngine::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::reasoning_engine_service::ListLocations {
super::builder::reasoning_engine_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::reasoning_engine_service::GetLocation {
super::builder::reasoning_engine_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::reasoning_engine_service::SetIamPolicy {
super::builder::reasoning_engine_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::reasoning_engine_service::GetIamPolicy {
super::builder::reasoning_engine_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::reasoning_engine_service::TestIamPermissions {
super::builder::reasoning_engine_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::reasoning_engine_service::ListOperations {
super::builder::reasoning_engine_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::reasoning_engine_service::GetOperation {
super::builder::reasoning_engine_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::reasoning_engine_service::DeleteOperation {
super::builder::reasoning_engine_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::reasoning_engine_service::CancelOperation {
super::builder::reasoning_engine_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::reasoning_engine_service::WaitOperation {
super::builder::reasoning_engine_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "schedule-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "schedule-service")))]
#[derive(Clone, Debug)]
pub struct ScheduleService {
inner: std::sync::Arc<dyn super::stub::dynamic::ScheduleService>,
}
#[cfg(feature = "schedule-service")]
impl ScheduleService {
pub fn builder() -> super::builder::schedule_service::ClientBuilder {
crate::new_client_builder(super::builder::schedule_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::ScheduleService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::ScheduleService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ScheduleService> {
super::transport::ScheduleService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ScheduleService> {
Self::build_transport(conf)
.await
.map(super::tracing::ScheduleService::new)
}
pub fn create_schedule(&self) -> super::builder::schedule_service::CreateSchedule {
super::builder::schedule_service::CreateSchedule::new(self.inner.clone())
}
pub fn delete_schedule(&self) -> super::builder::schedule_service::DeleteSchedule {
super::builder::schedule_service::DeleteSchedule::new(self.inner.clone())
}
pub fn get_schedule(&self) -> super::builder::schedule_service::GetSchedule {
super::builder::schedule_service::GetSchedule::new(self.inner.clone())
}
pub fn list_schedules(&self) -> super::builder::schedule_service::ListSchedules {
super::builder::schedule_service::ListSchedules::new(self.inner.clone())
}
pub fn pause_schedule(&self) -> super::builder::schedule_service::PauseSchedule {
super::builder::schedule_service::PauseSchedule::new(self.inner.clone())
}
pub fn resume_schedule(&self) -> super::builder::schedule_service::ResumeSchedule {
super::builder::schedule_service::ResumeSchedule::new(self.inner.clone())
}
pub fn update_schedule(&self) -> super::builder::schedule_service::UpdateSchedule {
super::builder::schedule_service::UpdateSchedule::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::schedule_service::ListLocations {
super::builder::schedule_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::schedule_service::GetLocation {
super::builder::schedule_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::schedule_service::SetIamPolicy {
super::builder::schedule_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::schedule_service::GetIamPolicy {
super::builder::schedule_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::schedule_service::TestIamPermissions {
super::builder::schedule_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::schedule_service::ListOperations {
super::builder::schedule_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::schedule_service::GetOperation {
super::builder::schedule_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::schedule_service::DeleteOperation {
super::builder::schedule_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::schedule_service::CancelOperation {
super::builder::schedule_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::schedule_service::WaitOperation {
super::builder::schedule_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "session-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "session-service")))]
#[derive(Clone, Debug)]
pub struct SessionService {
inner: std::sync::Arc<dyn super::stub::dynamic::SessionService>,
}
#[cfg(feature = "session-service")]
impl SessionService {
pub fn builder() -> super::builder::session_service::ClientBuilder {
crate::new_client_builder(super::builder::session_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::SessionService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::SessionService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::SessionService> {
super::transport::SessionService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::SessionService> {
Self::build_transport(conf)
.await
.map(super::tracing::SessionService::new)
}
pub fn create_session(&self) -> super::builder::session_service::CreateSession {
super::builder::session_service::CreateSession::new(self.inner.clone())
}
pub fn get_session(&self) -> super::builder::session_service::GetSession {
super::builder::session_service::GetSession::new(self.inner.clone())
}
pub fn list_sessions(&self) -> super::builder::session_service::ListSessions {
super::builder::session_service::ListSessions::new(self.inner.clone())
}
pub fn update_session(&self) -> super::builder::session_service::UpdateSession {
super::builder::session_service::UpdateSession::new(self.inner.clone())
}
pub fn delete_session(&self) -> super::builder::session_service::DeleteSession {
super::builder::session_service::DeleteSession::new(self.inner.clone())
}
pub fn list_events(&self) -> super::builder::session_service::ListEvents {
super::builder::session_service::ListEvents::new(self.inner.clone())
}
pub fn append_event(&self) -> super::builder::session_service::AppendEvent {
super::builder::session_service::AppendEvent::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::session_service::ListLocations {
super::builder::session_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::session_service::GetLocation {
super::builder::session_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::session_service::SetIamPolicy {
super::builder::session_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::session_service::GetIamPolicy {
super::builder::session_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::session_service::TestIamPermissions {
super::builder::session_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::session_service::ListOperations {
super::builder::session_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::session_service::GetOperation {
super::builder::session_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::session_service::DeleteOperation {
super::builder::session_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::session_service::CancelOperation {
super::builder::session_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::session_service::WaitOperation {
super::builder::session_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "specialist-pool-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "specialist-pool-service")))]
#[derive(Clone, Debug)]
pub struct SpecialistPoolService {
inner: std::sync::Arc<dyn super::stub::dynamic::SpecialistPoolService>,
}
#[cfg(feature = "specialist-pool-service")]
impl SpecialistPoolService {
pub fn builder() -> super::builder::specialist_pool_service::ClientBuilder {
crate::new_client_builder(super::builder::specialist_pool_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::SpecialistPoolService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::SpecialistPoolService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::SpecialistPoolService> {
super::transport::SpecialistPoolService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::SpecialistPoolService> {
Self::build_transport(conf)
.await
.map(super::tracing::SpecialistPoolService::new)
}
pub fn create_specialist_pool(
&self,
) -> super::builder::specialist_pool_service::CreateSpecialistPool {
super::builder::specialist_pool_service::CreateSpecialistPool::new(self.inner.clone())
}
pub fn get_specialist_pool(
&self,
) -> super::builder::specialist_pool_service::GetSpecialistPool {
super::builder::specialist_pool_service::GetSpecialistPool::new(self.inner.clone())
}
pub fn list_specialist_pools(
&self,
) -> super::builder::specialist_pool_service::ListSpecialistPools {
super::builder::specialist_pool_service::ListSpecialistPools::new(self.inner.clone())
}
pub fn delete_specialist_pool(
&self,
) -> super::builder::specialist_pool_service::DeleteSpecialistPool {
super::builder::specialist_pool_service::DeleteSpecialistPool::new(self.inner.clone())
}
pub fn update_specialist_pool(
&self,
) -> super::builder::specialist_pool_service::UpdateSpecialistPool {
super::builder::specialist_pool_service::UpdateSpecialistPool::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::specialist_pool_service::ListLocations {
super::builder::specialist_pool_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::specialist_pool_service::GetLocation {
super::builder::specialist_pool_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::specialist_pool_service::SetIamPolicy {
super::builder::specialist_pool_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::specialist_pool_service::GetIamPolicy {
super::builder::specialist_pool_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::specialist_pool_service::TestIamPermissions {
super::builder::specialist_pool_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::specialist_pool_service::ListOperations {
super::builder::specialist_pool_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::specialist_pool_service::GetOperation {
super::builder::specialist_pool_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::specialist_pool_service::DeleteOperation {
super::builder::specialist_pool_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::specialist_pool_service::CancelOperation {
super::builder::specialist_pool_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::specialist_pool_service::WaitOperation {
super::builder::specialist_pool_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "tensorboard-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "tensorboard-service")))]
#[derive(Clone, Debug)]
pub struct TensorboardService {
inner: std::sync::Arc<dyn super::stub::dynamic::TensorboardService>,
}
#[cfg(feature = "tensorboard-service")]
impl TensorboardService {
pub fn builder() -> super::builder::tensorboard_service::ClientBuilder {
crate::new_client_builder(super::builder::tensorboard_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::TensorboardService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::TensorboardService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::TensorboardService> {
super::transport::TensorboardService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::TensorboardService> {
Self::build_transport(conf)
.await
.map(super::tracing::TensorboardService::new)
}
pub fn create_tensorboard(&self) -> super::builder::tensorboard_service::CreateTensorboard {
super::builder::tensorboard_service::CreateTensorboard::new(self.inner.clone())
}
pub fn get_tensorboard(&self) -> super::builder::tensorboard_service::GetTensorboard {
super::builder::tensorboard_service::GetTensorboard::new(self.inner.clone())
}
pub fn update_tensorboard(&self) -> super::builder::tensorboard_service::UpdateTensorboard {
super::builder::tensorboard_service::UpdateTensorboard::new(self.inner.clone())
}
pub fn list_tensorboards(&self) -> super::builder::tensorboard_service::ListTensorboards {
super::builder::tensorboard_service::ListTensorboards::new(self.inner.clone())
}
pub fn delete_tensorboard(&self) -> super::builder::tensorboard_service::DeleteTensorboard {
super::builder::tensorboard_service::DeleteTensorboard::new(self.inner.clone())
}
pub fn read_tensorboard_usage(
&self,
) -> super::builder::tensorboard_service::ReadTensorboardUsage {
super::builder::tensorboard_service::ReadTensorboardUsage::new(self.inner.clone())
}
pub fn read_tensorboard_size(
&self,
) -> super::builder::tensorboard_service::ReadTensorboardSize {
super::builder::tensorboard_service::ReadTensorboardSize::new(self.inner.clone())
}
pub fn create_tensorboard_experiment(
&self,
) -> super::builder::tensorboard_service::CreateTensorboardExperiment {
super::builder::tensorboard_service::CreateTensorboardExperiment::new(self.inner.clone())
}
pub fn get_tensorboard_experiment(
&self,
) -> super::builder::tensorboard_service::GetTensorboardExperiment {
super::builder::tensorboard_service::GetTensorboardExperiment::new(self.inner.clone())
}
pub fn update_tensorboard_experiment(
&self,
) -> super::builder::tensorboard_service::UpdateTensorboardExperiment {
super::builder::tensorboard_service::UpdateTensorboardExperiment::new(self.inner.clone())
}
pub fn list_tensorboard_experiments(
&self,
) -> super::builder::tensorboard_service::ListTensorboardExperiments {
super::builder::tensorboard_service::ListTensorboardExperiments::new(self.inner.clone())
}
pub fn delete_tensorboard_experiment(
&self,
) -> super::builder::tensorboard_service::DeleteTensorboardExperiment {
super::builder::tensorboard_service::DeleteTensorboardExperiment::new(self.inner.clone())
}
pub fn create_tensorboard_run(
&self,
) -> super::builder::tensorboard_service::CreateTensorboardRun {
super::builder::tensorboard_service::CreateTensorboardRun::new(self.inner.clone())
}
pub fn batch_create_tensorboard_runs(
&self,
) -> super::builder::tensorboard_service::BatchCreateTensorboardRuns {
super::builder::tensorboard_service::BatchCreateTensorboardRuns::new(self.inner.clone())
}
pub fn get_tensorboard_run(&self) -> super::builder::tensorboard_service::GetTensorboardRun {
super::builder::tensorboard_service::GetTensorboardRun::new(self.inner.clone())
}
pub fn update_tensorboard_run(
&self,
) -> super::builder::tensorboard_service::UpdateTensorboardRun {
super::builder::tensorboard_service::UpdateTensorboardRun::new(self.inner.clone())
}
pub fn list_tensorboard_runs(
&self,
) -> super::builder::tensorboard_service::ListTensorboardRuns {
super::builder::tensorboard_service::ListTensorboardRuns::new(self.inner.clone())
}
pub fn delete_tensorboard_run(
&self,
) -> super::builder::tensorboard_service::DeleteTensorboardRun {
super::builder::tensorboard_service::DeleteTensorboardRun::new(self.inner.clone())
}
pub fn batch_create_tensorboard_time_series(
&self,
) -> super::builder::tensorboard_service::BatchCreateTensorboardTimeSeries {
super::builder::tensorboard_service::BatchCreateTensorboardTimeSeries::new(
self.inner.clone(),
)
}
pub fn create_tensorboard_time_series(
&self,
) -> super::builder::tensorboard_service::CreateTensorboardTimeSeries {
super::builder::tensorboard_service::CreateTensorboardTimeSeries::new(self.inner.clone())
}
pub fn get_tensorboard_time_series(
&self,
) -> super::builder::tensorboard_service::GetTensorboardTimeSeries {
super::builder::tensorboard_service::GetTensorboardTimeSeries::new(self.inner.clone())
}
pub fn update_tensorboard_time_series(
&self,
) -> super::builder::tensorboard_service::UpdateTensorboardTimeSeries {
super::builder::tensorboard_service::UpdateTensorboardTimeSeries::new(self.inner.clone())
}
pub fn list_tensorboard_time_series(
&self,
) -> super::builder::tensorboard_service::ListTensorboardTimeSeries {
super::builder::tensorboard_service::ListTensorboardTimeSeries::new(self.inner.clone())
}
pub fn delete_tensorboard_time_series(
&self,
) -> super::builder::tensorboard_service::DeleteTensorboardTimeSeries {
super::builder::tensorboard_service::DeleteTensorboardTimeSeries::new(self.inner.clone())
}
pub fn batch_read_tensorboard_time_series_data(
&self,
) -> super::builder::tensorboard_service::BatchReadTensorboardTimeSeriesData {
super::builder::tensorboard_service::BatchReadTensorboardTimeSeriesData::new(
self.inner.clone(),
)
}
pub fn read_tensorboard_time_series_data(
&self,
) -> super::builder::tensorboard_service::ReadTensorboardTimeSeriesData {
super::builder::tensorboard_service::ReadTensorboardTimeSeriesData::new(self.inner.clone())
}
pub fn write_tensorboard_experiment_data(
&self,
) -> super::builder::tensorboard_service::WriteTensorboardExperimentData {
super::builder::tensorboard_service::WriteTensorboardExperimentData::new(self.inner.clone())
}
pub fn write_tensorboard_run_data(
&self,
) -> super::builder::tensorboard_service::WriteTensorboardRunData {
super::builder::tensorboard_service::WriteTensorboardRunData::new(self.inner.clone())
}
pub fn export_tensorboard_time_series_data(
&self,
) -> super::builder::tensorboard_service::ExportTensorboardTimeSeriesData {
super::builder::tensorboard_service::ExportTensorboardTimeSeriesData::new(
self.inner.clone(),
)
}
pub fn list_locations(&self) -> super::builder::tensorboard_service::ListLocations {
super::builder::tensorboard_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::tensorboard_service::GetLocation {
super::builder::tensorboard_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::tensorboard_service::SetIamPolicy {
super::builder::tensorboard_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::tensorboard_service::GetIamPolicy {
super::builder::tensorboard_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::tensorboard_service::TestIamPermissions {
super::builder::tensorboard_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::tensorboard_service::ListOperations {
super::builder::tensorboard_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::tensorboard_service::GetOperation {
super::builder::tensorboard_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::tensorboard_service::DeleteOperation {
super::builder::tensorboard_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::tensorboard_service::CancelOperation {
super::builder::tensorboard_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::tensorboard_service::WaitOperation {
super::builder::tensorboard_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "vertex-rag-data-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "vertex-rag-data-service")))]
#[derive(Clone, Debug)]
pub struct VertexRagDataService {
inner: std::sync::Arc<dyn super::stub::dynamic::VertexRagDataService>,
}
#[cfg(feature = "vertex-rag-data-service")]
impl VertexRagDataService {
pub fn builder() -> super::builder::vertex_rag_data_service::ClientBuilder {
crate::new_client_builder(super::builder::vertex_rag_data_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::VertexRagDataService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::VertexRagDataService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::VertexRagDataService> {
super::transport::VertexRagDataService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::VertexRagDataService> {
Self::build_transport(conf)
.await
.map(super::tracing::VertexRagDataService::new)
}
pub fn create_rag_corpus(&self) -> super::builder::vertex_rag_data_service::CreateRagCorpus {
super::builder::vertex_rag_data_service::CreateRagCorpus::new(self.inner.clone())
}
pub fn update_rag_corpus(&self) -> super::builder::vertex_rag_data_service::UpdateRagCorpus {
super::builder::vertex_rag_data_service::UpdateRagCorpus::new(self.inner.clone())
}
pub fn get_rag_corpus(&self) -> super::builder::vertex_rag_data_service::GetRagCorpus {
super::builder::vertex_rag_data_service::GetRagCorpus::new(self.inner.clone())
}
pub fn list_rag_corpora(&self) -> super::builder::vertex_rag_data_service::ListRagCorpora {
super::builder::vertex_rag_data_service::ListRagCorpora::new(self.inner.clone())
}
pub fn delete_rag_corpus(&self) -> super::builder::vertex_rag_data_service::DeleteRagCorpus {
super::builder::vertex_rag_data_service::DeleteRagCorpus::new(self.inner.clone())
}
pub fn upload_rag_file(&self) -> super::builder::vertex_rag_data_service::UploadRagFile {
super::builder::vertex_rag_data_service::UploadRagFile::new(self.inner.clone())
}
pub fn import_rag_files(&self) -> super::builder::vertex_rag_data_service::ImportRagFiles {
super::builder::vertex_rag_data_service::ImportRagFiles::new(self.inner.clone())
}
pub fn get_rag_file(&self) -> super::builder::vertex_rag_data_service::GetRagFile {
super::builder::vertex_rag_data_service::GetRagFile::new(self.inner.clone())
}
pub fn list_rag_files(&self) -> super::builder::vertex_rag_data_service::ListRagFiles {
super::builder::vertex_rag_data_service::ListRagFiles::new(self.inner.clone())
}
pub fn delete_rag_file(&self) -> super::builder::vertex_rag_data_service::DeleteRagFile {
super::builder::vertex_rag_data_service::DeleteRagFile::new(self.inner.clone())
}
pub fn update_rag_engine_config(
&self,
) -> super::builder::vertex_rag_data_service::UpdateRagEngineConfig {
super::builder::vertex_rag_data_service::UpdateRagEngineConfig::new(self.inner.clone())
}
pub fn get_rag_engine_config(
&self,
) -> super::builder::vertex_rag_data_service::GetRagEngineConfig {
super::builder::vertex_rag_data_service::GetRagEngineConfig::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::vertex_rag_data_service::ListLocations {
super::builder::vertex_rag_data_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::vertex_rag_data_service::GetLocation {
super::builder::vertex_rag_data_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::vertex_rag_data_service::SetIamPolicy {
super::builder::vertex_rag_data_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::vertex_rag_data_service::GetIamPolicy {
super::builder::vertex_rag_data_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(
&self,
) -> super::builder::vertex_rag_data_service::TestIamPermissions {
super::builder::vertex_rag_data_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::vertex_rag_data_service::ListOperations {
super::builder::vertex_rag_data_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::vertex_rag_data_service::GetOperation {
super::builder::vertex_rag_data_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::vertex_rag_data_service::DeleteOperation {
super::builder::vertex_rag_data_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::vertex_rag_data_service::CancelOperation {
super::builder::vertex_rag_data_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::vertex_rag_data_service::WaitOperation {
super::builder::vertex_rag_data_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "vertex-rag-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "vertex-rag-service")))]
#[derive(Clone, Debug)]
pub struct VertexRagService {
inner: std::sync::Arc<dyn super::stub::dynamic::VertexRagService>,
}
#[cfg(feature = "vertex-rag-service")]
impl VertexRagService {
pub fn builder() -> super::builder::vertex_rag_service::ClientBuilder {
crate::new_client_builder(super::builder::vertex_rag_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::VertexRagService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::VertexRagService>>
{
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::VertexRagService> {
super::transport::VertexRagService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::VertexRagService> {
Self::build_transport(conf)
.await
.map(super::tracing::VertexRagService::new)
}
pub fn retrieve_contexts(&self) -> super::builder::vertex_rag_service::RetrieveContexts {
super::builder::vertex_rag_service::RetrieveContexts::new(self.inner.clone())
}
pub fn augment_prompt(&self) -> super::builder::vertex_rag_service::AugmentPrompt {
super::builder::vertex_rag_service::AugmentPrompt::new(self.inner.clone())
}
pub fn corroborate_content(&self) -> super::builder::vertex_rag_service::CorroborateContent {
super::builder::vertex_rag_service::CorroborateContent::new(self.inner.clone())
}
pub fn ask_contexts(&self) -> super::builder::vertex_rag_service::AskContexts {
super::builder::vertex_rag_service::AskContexts::new(self.inner.clone())
}
pub fn async_retrieve_contexts(
&self,
) -> super::builder::vertex_rag_service::AsyncRetrieveContexts {
super::builder::vertex_rag_service::AsyncRetrieveContexts::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::vertex_rag_service::ListLocations {
super::builder::vertex_rag_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::vertex_rag_service::GetLocation {
super::builder::vertex_rag_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::vertex_rag_service::SetIamPolicy {
super::builder::vertex_rag_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::vertex_rag_service::GetIamPolicy {
super::builder::vertex_rag_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::vertex_rag_service::TestIamPermissions {
super::builder::vertex_rag_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::vertex_rag_service::ListOperations {
super::builder::vertex_rag_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::vertex_rag_service::GetOperation {
super::builder::vertex_rag_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::vertex_rag_service::DeleteOperation {
super::builder::vertex_rag_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::vertex_rag_service::CancelOperation {
super::builder::vertex_rag_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::vertex_rag_service::WaitOperation {
super::builder::vertex_rag_service::WaitOperation::new(self.inner.clone())
}
}
#[cfg(feature = "vizier-service")]
#[cfg_attr(docsrs, doc(cfg(feature = "vizier-service")))]
#[derive(Clone, Debug)]
pub struct VizierService {
inner: std::sync::Arc<dyn super::stub::dynamic::VizierService>,
}
#[cfg(feature = "vizier-service")]
impl VizierService {
pub fn builder() -> super::builder::vizier_service::ClientBuilder {
crate::new_client_builder(super::builder::vizier_service::client::Factory)
}
pub fn from_stub<T>(stub: T) -> Self
where
T: super::stub::VizierService + 'static,
{
Self {
inner: std::sync::Arc::new(stub),
}
}
pub(crate) async fn new(
config: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<Self> {
let inner = Self::build_inner(config).await?;
Ok(Self { inner })
}
async fn build_inner(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<std::sync::Arc<dyn super::stub::dynamic::VizierService>> {
if gaxi::options::tracing_enabled(&conf) {
return Ok(std::sync::Arc::new(Self::build_with_tracing(conf).await?));
}
Ok(std::sync::Arc::new(Self::build_transport(conf).await?))
}
async fn build_transport(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::VizierService> {
super::transport::VizierService::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::VizierService> {
Self::build_transport(conf)
.await
.map(super::tracing::VizierService::new)
}
pub fn create_study(&self) -> super::builder::vizier_service::CreateStudy {
super::builder::vizier_service::CreateStudy::new(self.inner.clone())
}
pub fn get_study(&self) -> super::builder::vizier_service::GetStudy {
super::builder::vizier_service::GetStudy::new(self.inner.clone())
}
pub fn list_studies(&self) -> super::builder::vizier_service::ListStudies {
super::builder::vizier_service::ListStudies::new(self.inner.clone())
}
pub fn delete_study(&self) -> super::builder::vizier_service::DeleteStudy {
super::builder::vizier_service::DeleteStudy::new(self.inner.clone())
}
pub fn lookup_study(&self) -> super::builder::vizier_service::LookupStudy {
super::builder::vizier_service::LookupStudy::new(self.inner.clone())
}
pub fn suggest_trials(&self) -> super::builder::vizier_service::SuggestTrials {
super::builder::vizier_service::SuggestTrials::new(self.inner.clone())
}
pub fn create_trial(&self) -> super::builder::vizier_service::CreateTrial {
super::builder::vizier_service::CreateTrial::new(self.inner.clone())
}
pub fn get_trial(&self) -> super::builder::vizier_service::GetTrial {
super::builder::vizier_service::GetTrial::new(self.inner.clone())
}
pub fn list_trials(&self) -> super::builder::vizier_service::ListTrials {
super::builder::vizier_service::ListTrials::new(self.inner.clone())
}
pub fn add_trial_measurement(&self) -> super::builder::vizier_service::AddTrialMeasurement {
super::builder::vizier_service::AddTrialMeasurement::new(self.inner.clone())
}
pub fn complete_trial(&self) -> super::builder::vizier_service::CompleteTrial {
super::builder::vizier_service::CompleteTrial::new(self.inner.clone())
}
pub fn delete_trial(&self) -> super::builder::vizier_service::DeleteTrial {
super::builder::vizier_service::DeleteTrial::new(self.inner.clone())
}
pub fn check_trial_early_stopping_state(
&self,
) -> super::builder::vizier_service::CheckTrialEarlyStoppingState {
super::builder::vizier_service::CheckTrialEarlyStoppingState::new(self.inner.clone())
}
pub fn stop_trial(&self) -> super::builder::vizier_service::StopTrial {
super::builder::vizier_service::StopTrial::new(self.inner.clone())
}
pub fn list_optimal_trials(&self) -> super::builder::vizier_service::ListOptimalTrials {
super::builder::vizier_service::ListOptimalTrials::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::vizier_service::ListLocations {
super::builder::vizier_service::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::vizier_service::GetLocation {
super::builder::vizier_service::GetLocation::new(self.inner.clone())
}
pub fn set_iam_policy(&self) -> super::builder::vizier_service::SetIamPolicy {
super::builder::vizier_service::SetIamPolicy::new(self.inner.clone())
}
pub fn get_iam_policy(&self) -> super::builder::vizier_service::GetIamPolicy {
super::builder::vizier_service::GetIamPolicy::new(self.inner.clone())
}
pub fn test_iam_permissions(&self) -> super::builder::vizier_service::TestIamPermissions {
super::builder::vizier_service::TestIamPermissions::new(self.inner.clone())
}
pub fn list_operations(&self) -> super::builder::vizier_service::ListOperations {
super::builder::vizier_service::ListOperations::new(self.inner.clone())
}
pub fn get_operation(&self) -> super::builder::vizier_service::GetOperation {
super::builder::vizier_service::GetOperation::new(self.inner.clone())
}
pub fn delete_operation(&self) -> super::builder::vizier_service::DeleteOperation {
super::builder::vizier_service::DeleteOperation::new(self.inner.clone())
}
pub fn cancel_operation(&self) -> super::builder::vizier_service::CancelOperation {
super::builder::vizier_service::CancelOperation::new(self.inner.clone())
}
pub fn wait_operation(&self) -> super::builder::vizier_service::WaitOperation {
super::builder::vizier_service::WaitOperation::new(self.inner.clone())
}
}