From 00478579e339dbd4505443cf049d0f2869009e89 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Erik=20Kaunism=C3=A4ki?= Date: Wed, 31 Jul 2024 11:50:25 +0200 Subject: [PATCH] Update router/src/server.rs Co-authored-by: Nicolas Patry --- router/src/server.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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(),