fix: fix some trtllm tests (#3084)

Author: richardhuo-nv

tests/serve/test_trtllm.py

@@ -59,7 +59,6 @@ class TRTLLMConfig(EngineConfig):
         request_payloads=[
             chat_payload_default(
                 expected_log=[
-                    r"ZMQ listener .* received batch with \d+ events \(seq=\d+\)",
                     r"Event processor for worker_id \d+ processing event: Stored\(",
                     r"Selected worker: \d+, logit: ",
                 ]
@@ -123,7 +122,7 @@ def test_chat_only_aggregated_with_test_logits_processor(
         script_name=base.script_name,  # agg.sh
         marks=[],  # not used by this direct test
         request_payloads=[
-            chat_payload_default(),
+            chat_payload_default(expected_response=["Hello world!"]),
         ],
         model="Qwen/Qwen3-0.6B",
         delayed_start=base.delayed_start,

tests/utils/engine_process.py

@@ -4,6 +4,7 @@
 import json
 import logging
 import os
+import time
 from dataclasses import dataclass, field
 from typing import Any, Dict, List, Optional
 
@@ -110,6 +111,9 @@ def check_response(
 
         # Optionally validate expected log patterns after response handling
         if payload.expected_log:
+            time.sleep(
+                0.5
+            )  # The kv event sometimes needs extra time to arrive and be reflected in the log.
             self.validate_expected_logs(payload.expected_log)
 
     def validate_expected_logs(self, patterns: Any) -> None: