google_cloud_modelarmor_v1/

client.rs

// Copyright 2025 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Code generated by sidekick. DO NOT EDIT.
#![allow(rustdoc::redundant_explicit_links)]
#![allow(rustdoc::broken_intra_doc_links)]

/// Implements a client for the Model Armor API.
///
/// # Example
/// ```
/// # tokio_test::block_on(async {
/// # use google_cloud_modelarmor_v1::client::ModelArmor;
/// let client = ModelArmor::builder().build().await?;
/// // use `client` to make requests to the Model Armor API.
/// # gax::client_builder::Result::<()>::Ok(()) });
/// ```
///
/// # Service Description
///
/// Service describing handlers for resources
///
/// # Configuration
///
/// To configure `ModelArmor` use the `with_*` methods in the type returned
/// by [builder()][ModelArmor::builder]. The default configuration should
/// work for most applications. Common configuration changes include
///
/// * [with_endpoint()]: by default this client uses the global default endpoint
///   (`https://modelarmor.googleapis.com`). Applications using regional
///   endpoints or running in restricted networks (e.g. a network configured
//    with [Private Google Access with VPC Service Controls]) may want to
///   override this default.
/// * [with_credentials()]: by default this client uses
///   [Application Default Credentials]. Applications using custom
///   authentication may need to override this default.
///
/// [with_endpoint()]: super::builder::model_armor::ClientBuilder::with_endpoint
/// [with_credentials()]: super::builder::model_armor::ClientBuilder::credentials
/// [Private Google Access with VPC Service Controls]: https://cloud.google.com/vpc-service-controls/docs/private-connectivity
/// [Application Default Credentials]: https://cloud.google.com/docs/authentication#adc
///
/// # Pooling and Cloning
///
/// `ModelArmor` holds a connection pool internally, it is advised to
/// create one and the reuse it.  You do not need to wrap `ModelArmor` in
/// an [Rc](std::rc::Rc) or [Arc](std::sync::Arc) to reuse it, because it
/// already uses an `Arc` internally.
#[derive(Clone, Debug)]
pub struct ModelArmor {
    inner: std::sync::Arc<dyn super::stub::dynamic::ModelArmor>,
}

impl ModelArmor {
    /// Returns a builder for [ModelArmor].
    ///
    /// ```
    /// # tokio_test::block_on(async {
    /// # use google_cloud_modelarmor_v1::client::ModelArmor;
    /// let client = ModelArmor::builder().build().await?;
    /// # gax::client_builder::Result::<()>::Ok(()) });
    /// ```
    pub fn builder() -> super::builder::model_armor::ClientBuilder {
        gax::client_builder::internal::new_builder(super::builder::model_armor::client::Factory)
    }

    /// Creates a new client from the provided stub.
    ///
    /// The most common case for calling this function is in tests mocking the
    /// client's behavior.
    pub fn from_stub<T>(stub: T) -> Self
    where
        T: super::stub::ModelArmor + 'static,
    {
        Self {
            inner: std::sync::Arc::new(stub),
        }
    }

    pub(crate) async fn new(
        config: gaxi::options::ClientConfig,
    ) -> gax::client_builder::Result<Self> {
        let inner = Self::build_inner(config).await?;
        Ok(Self { inner })
    }

    async fn build_inner(
        conf: gaxi::options::ClientConfig,
    ) -> gax::client_builder::Result<std::sync::Arc<dyn super::stub::dynamic::ModelArmor>> {
        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,
    ) -> gax::client_builder::Result<impl super::stub::ModelArmor> {
        super::transport::ModelArmor::new(conf).await
    }

    async fn build_with_tracing(
        conf: gaxi::options::ClientConfig,
    ) -> gax::client_builder::Result<impl super::stub::ModelArmor> {
        Self::build_transport(conf)
            .await
            .map(super::tracing::ModelArmor::new)
    }

    /// Lists Templates in a given project and location.
    pub fn list_templates(&self) -> super::builder::model_armor::ListTemplates {
        super::builder::model_armor::ListTemplates::new(self.inner.clone())
    }

    /// Gets details of a single Template.
    pub fn get_template(&self) -> super::builder::model_armor::GetTemplate {
        super::builder::model_armor::GetTemplate::new(self.inner.clone())
    }

    /// Creates a new Template in a given project and location.
    pub fn create_template(&self) -> super::builder::model_armor::CreateTemplate {
        super::builder::model_armor::CreateTemplate::new(self.inner.clone())
    }

    /// Updates the parameters of a single Template.
    pub fn update_template(&self) -> super::builder::model_armor::UpdateTemplate {
        super::builder::model_armor::UpdateTemplate::new(self.inner.clone())
    }

    /// Deletes a single Template.
    pub fn delete_template(&self) -> super::builder::model_armor::DeleteTemplate {
        super::builder::model_armor::DeleteTemplate::new(self.inner.clone())
    }

    /// Gets details of a single floor setting of a project
    pub fn get_floor_setting(&self) -> super::builder::model_armor::GetFloorSetting {
        super::builder::model_armor::GetFloorSetting::new(self.inner.clone())
    }

    /// Updates the parameters of a single floor setting of a project
    pub fn update_floor_setting(&self) -> super::builder::model_armor::UpdateFloorSetting {
        super::builder::model_armor::UpdateFloorSetting::new(self.inner.clone())
    }

    /// Sanitizes User Prompt.
    pub fn sanitize_user_prompt(&self) -> super::builder::model_armor::SanitizeUserPrompt {
        super::builder::model_armor::SanitizeUserPrompt::new(self.inner.clone())
    }

    /// Sanitizes Model Response.
    pub fn sanitize_model_response(&self) -> super::builder::model_armor::SanitizeModelResponse {
        super::builder::model_armor::SanitizeModelResponse::new(self.inner.clone())
    }

    /// Lists information about the supported locations for this service.
    pub fn list_locations(&self) -> super::builder::model_armor::ListLocations {
        super::builder::model_armor::ListLocations::new(self.inner.clone())
    }

    /// Gets information about a location.
    pub fn get_location(&self) -> super::builder::model_armor::GetLocation {
        super::builder::model_armor::GetLocation::new(self.inner.clone())
    }
}