diff --git a/router/src/server.rs b/router/src/server.rs index cb23a3e9..153d9fdd 100644 --- a/router/src/server.rs +++ b/router/src/server.rs @@ -1696,10 +1696,10 @@ pub async fn run( // Only send usage stats when TGI is run in container and the function returns Some let is_container = matches!(usage_stats::is_container(), Ok(true)); - let user_agent = match usage_stats_level { - usage_stats::UsageStatsLevel::On | usage_stats::UsageStatsLevel::NoStack - if is_container => + let user_agent = match (usage_stats_level, is_container) { + (usage_stats::UsageStatsLevel::On | usage_stats::UsageStatsLevel::NoStack, true) => { + _ => None let reduced_args = usage_stats::Args::new( config.clone(), tokenizer_config.tokenizer_class.clone(),