From bd16244d936c29903c137ea75955a7dc88262aa1 Mon Sep 17 00:00:00 2001
From: Simo Lin <linsimo.mark@gmail.com>
Date: Fri, 19 Dec 2025 11:02:57 -1000
Subject: [PATCH] [model-gateway] Improve logging in policies module (#15496)

---
 sgl-model-gateway/src/policies/bucket.rs |  4 ++--
 sgl-model-gateway/src/policies/tree.rs   | 10 +++++-----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/sgl-model-gateway/src/policies/bucket.rs b/sgl-model-gateway/src/policies/bucket.rs
index 56d8401df..c86b2fbc7 100644
--- a/sgl-model-gateway/src/policies/bucket.rs
+++ b/sgl-model-gateway/src/policies/bucket.rs
@@ -58,7 +58,7 @@ impl BucketPolicy {
                             bucket_guard.adjust_boundary();
                         }
                         Err(e) => {
-                            eprintln!(
+                            error!(
                                 "Failed to acquire write lock for bucket {}: {}",
                                 model_id, e
                             );
@@ -112,7 +112,7 @@ impl BucketPolicy {
             if let Ok(mut bucket_guard) = lock_result {
                 bucket_guard.init_prefill_worker_urls(worker_urls);
             } else {
-                eprintln!("Failed to acquire write lock for bucket initialization");
+                error!("Failed to acquire write lock for bucket initialization");
             }
         }
     }
diff --git a/sgl-model-gateway/src/policies/tree.rs b/sgl-model-gateway/src/policies/tree.rs
index 97df66cee..528ba9644 100644
--- a/sgl-model-gateway/src/policies/tree.rs
+++ b/sgl-model-gateway/src/policies/tree.rs
@@ -10,7 +10,7 @@ use std::{
 };
 
 use dashmap::{mapref::entry::Entry, DashMap};
-use tracing::info;
+use tracing::debug;
 
 type NodeRef = Arc<Node>;
 
@@ -674,9 +674,9 @@ impl Tree {
             }
         }
 
-        info!("Before eviction - Used size per tenant:");
+        debug!("Before eviction - Used size per tenant:");
         for entry in self.tenant_char_count.iter() {
-            info!("Tenant: {}, Size: {}", entry.key(), entry.value());
+            debug!("Tenant: {}, Size: {}", entry.key(), entry.value());
         }
 
         // Process eviction
@@ -728,9 +728,9 @@ impl Tree {
             };
         }
 
-        info!("After eviction - Used size per tenant:");
+        debug!("After eviction - Used size per tenant:");
         for entry in self.tenant_char_count.iter() {
-            info!("Tenant: {}, Size: {}", entry.key(), entry.value());
+            debug!("Tenant: {}, Size: {}", entry.key(), entry.value());
         }
     }