diff --git a/crates/ml/src/deployment/endpoints.rs b/crates/ml/src/deployment/endpoints.rs index c5eba1615..340ef6dca 100644 --- a/crates/ml/src/deployment/endpoints.rs +++ b/crates/ml/src/deployment/endpoints.rs @@ -13,6 +13,7 @@ use prost::Message; use crate::types::{MLResult, MLError}; use crate::traits::MLModel; +use tracing::instrument; use super::{ ModelDeploymentRegistry, DeploymentConfig, DeploymentStrategy, versioning::ModelVersion, @@ -427,6 +428,7 @@ impl ModelDeploymentServiceImpl { #[tonic::async_trait] impl ModelDeploymentService for ModelDeploymentServiceImpl { + #[instrument(skip_all)] async fn deploy_model( &self, request: Request, @@ -469,6 +471,7 @@ impl ModelDeploymentService for ModelDeploymentServiceImpl { } } + #[instrument(skip_all)] async fn get_deployment( &self, request: Request, @@ -495,6 +498,7 @@ impl ModelDeploymentService for ModelDeploymentServiceImpl { } } + #[instrument(skip_all)] async fn list_deployments( &self, request: Request, @@ -539,6 +543,7 @@ impl ModelDeploymentService for ModelDeploymentServiceImpl { } } + #[instrument(skip_all)] async fn rollback_model( &self, request: Request, @@ -572,6 +577,7 @@ impl ModelDeploymentService for ModelDeploymentServiceImpl { } } + #[instrument(skip_all)] async fn undeploy_model( &self, request: Request, @@ -596,6 +602,7 @@ impl ModelDeploymentService for ModelDeploymentServiceImpl { } } + #[instrument(skip_all)] async fn get_model_metrics( &self, request: Request, @@ -630,6 +637,7 @@ impl ModelDeploymentService for ModelDeploymentServiceImpl { } } + #[instrument(skip_all)] async fn start_ab_test( &self, request: Request, @@ -682,6 +690,7 @@ impl ModelDeploymentService for ModelDeploymentServiceImpl { } } + #[instrument(skip_all)] async fn get_ab_test_status( &self, request: Request,