From cb4cdb43a473b7c733540a4e0b9902cca6e1924e Mon Sep 17 00:00:00 2001
From: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>
Date: Mon, 8 Dec 2025 22:39:27 +0800
Subject: [PATCH] Fix dp-aware incompatible with service-discovery (#14629)

---
 sgl-model-gateway/src/config/validation.rs | 6 ------
 sgl-model-gateway/src/service_discovery.rs | 2 +-
 2 files changed, 1 insertion(+), 7 deletions(-)

diff --git a/sgl-model-gateway/src/config/validation.rs b/sgl-model-gateway/src/config/validation.rs
index eda1000c8..f8a564849 100644
--- a/sgl-model-gateway/src/config/validation.rs
+++ b/sgl-model-gateway/src/config/validation.rs
@@ -574,12 +574,6 @@ impl ConfigValidator {
             }
         }
 
-        if has_service_discovery && config.dp_aware {
-            return Err(ConfigError::IncompatibleConfig {
-                reason: "DP-aware routing is not compatible with service discovery".to_string(),
-            });
-        }
-
         Ok(())
     }
 
diff --git a/sgl-model-gateway/src/service_discovery.rs b/sgl-model-gateway/src/service_discovery.rs
index 7586d9ed1..aa8cd1d7a 100644
--- a/sgl-model-gateway/src/service_discovery.rs
+++ b/sgl-model-gateway/src/service_discovery.rs
@@ -393,7 +393,7 @@ async fn handle_pod_event(
                 health_failure_threshold: app_context.router_config.health_check.failure_threshold,
                 max_connection_attempts: app_context.router_config.health_check.success_threshold
                     * 20,
-                dp_aware: false,
+                dp_aware: app_context.router_config.dp_aware,
             };
 
             let job = Job::AddWorker {