fix: Router + SGLang DP testing (#6057)

components/src/dynamo/sglang/publisher.py

@@ -234,6 +234,7 @@ class DynamoSglangPublisher:
                     kv_block_size=self.server_args.page_size,
                     zmq_endpoint=zmq_ep,
                     enable_local_indexer=self.dynamo_args.enable_local_indexer,
+                    dp_rank=dp_rank,
                 )
                 logging.info(
                     f"Setting up ZMQ kv event subscriber for dp_rank={dp_rank} "

tests/router/test_router_e2e_with_sglang.py

@@ -25,7 +25,7 @@ from tests.utils.port_utils import allocate_ports, deallocate_ports
 
 logger = logging.getLogger(__name__)
 
-MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
+MODEL_NAME = "silence09/DeepSeek-R1-Small-2layers"
 
 pytestmark = [
     pytest.mark.e2e,
@@ -173,6 +173,9 @@ class SGLangProcess:
                     [
                         "--dp-size",
                         str(data_parallel_size),
+                        "--tp-size",
+                        str(data_parallel_size),
+                        "--enable-dp-attention",
                     ]
                 )