[CI][Router] Fix bench_one_batch_server for pd router test (#7731)

Signed-off-by: Shangming Cai <caishangming@linux.alibaba.com>

.github/workflows/pr-test-pd-router.yml

@@ -114,7 +114,7 @@ jobs:
       run: |
         echo "Installing SGLang with all extras..."
         python3 -m pip --no-cache-dir install -e "python[all]" --break-system-packages
-        python3 -m pip --no-cache-dir install mooncake-transfer-engine==0.3.4.post1
+        python3 -m pip --no-cache-dir install mooncake-transfer-engine==0.3.4.post2
 
     - name: Build and install sgl-router
       run: |

python/sglang/bench_one_batch_server.py

@@ -253,8 +253,11 @@ def run_benchmark(server_args: ServerArgs, bench_args: BenchArgs):
     else:
         proc, base_url = launch_server_process(server_args)
 
-    server_info = requests.get(base_url + "/get_server_info")
-    tokenizer_path = server_info.json()["tokenizer_path"]
+    server_info = requests.get(base_url + "/get_server_info").json()
+    if "tokenizer_path" in server_info:
+        tokenizer_path = server_info["tokenizer_path"]
+    elif "prefill" in server_info:
+        tokenizer_path = server_info["prefill"][0]["tokenizer_path"]
     tokenizer = get_tokenizer(tokenizer_path)
 
     # warmup