#![allow(rustdoc::redundant_explicit_links)]
#![allow(rustdoc::broken_intra_doc_links)]
#[derive(Clone, Debug)]
pub struct ModelArmor {
inner: std::sync::Arc<dyn super::stub::dynamic::ModelArmor>,
}
impl ModelArmor {
pub fn builder() -> super::builder::model_armor::ClientBuilder {
crate::new_client_builder(super::builder::model_armor::client::Factory)
}
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,
) -> 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::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,
) -> crate::ClientBuilderResult<impl super::stub::ModelArmor> {
super::transport::ModelArmor::new(conf).await
}
async fn build_with_tracing(
conf: gaxi::options::ClientConfig,
) -> crate::ClientBuilderResult<impl super::stub::ModelArmor> {
Self::build_transport(conf)
.await
.map(super::tracing::ModelArmor::new)
}
pub fn list_templates(&self) -> super::builder::model_armor::ListTemplates {
super::builder::model_armor::ListTemplates::new(self.inner.clone())
}
pub fn get_template(&self) -> super::builder::model_armor::GetTemplate {
super::builder::model_armor::GetTemplate::new(self.inner.clone())
}
pub fn create_template(&self) -> super::builder::model_armor::CreateTemplate {
super::builder::model_armor::CreateTemplate::new(self.inner.clone())
}
pub fn update_template(&self) -> super::builder::model_armor::UpdateTemplate {
super::builder::model_armor::UpdateTemplate::new(self.inner.clone())
}
pub fn delete_template(&self) -> super::builder::model_armor::DeleteTemplate {
super::builder::model_armor::DeleteTemplate::new(self.inner.clone())
}
pub fn get_floor_setting(&self) -> super::builder::model_armor::GetFloorSetting {
super::builder::model_armor::GetFloorSetting::new(self.inner.clone())
}
pub fn update_floor_setting(&self) -> super::builder::model_armor::UpdateFloorSetting {
super::builder::model_armor::UpdateFloorSetting::new(self.inner.clone())
}
pub fn sanitize_user_prompt(&self) -> super::builder::model_armor::SanitizeUserPrompt {
super::builder::model_armor::SanitizeUserPrompt::new(self.inner.clone())
}
pub fn sanitize_model_response(&self) -> super::builder::model_armor::SanitizeModelResponse {
super::builder::model_armor::SanitizeModelResponse::new(self.inner.clone())
}
pub fn list_locations(&self) -> super::builder::model_armor::ListLocations {
super::builder::model_armor::ListLocations::new(self.inner.clone())
}
pub fn get_location(&self) -> super::builder::model_armor::GetLocation {
super::builder::model_armor::GetLocation::new(self.inner.clone())
}
}