Enable runtime access to provider name (#5399)

This commit is contained in:
Will Pfleger
2025-10-28 14:21:24 -04:00
committed by GitHub
parent f1ad5490b6
commit 4d8c91efbd
25 changed files with 196 additions and 3 deletions
@@ -30,6 +30,8 @@ pub struct SageMakerTgiProvider {
sagemaker_client: SageMakerClient,
endpoint_name: String,
model: ModelConfig,
#[serde(skip)]
name: String,
}
impl SageMakerTgiProvider {
@@ -79,6 +81,7 @@ impl SageMakerTgiProvider {
sagemaker_client,
endpoint_name,
model,
name: Self::metadata().name,
})
}
@@ -272,6 +275,10 @@ impl Provider for SageMakerTgiProvider {
)
}
fn get_name(&self) -> &str {
&self.name
}
fn get_model_config(&self) -> ModelConfig {
self.model.clone()
}