Update after review from gemini-code

Update comments:
- vllm-project#28309 (review)

Signed-off-by: Martin Hickey <martin.hickey@ie.ibm.com>

Author: hickeyma

vllm/distributed/kv_transfer/kv_connector/v1/lmcache_connector.py

@@ -244,7 +244,7 @@ def update_connector_output(self, connector_output: KVConnectorOutput):
             or not kv_events.events
         ):
             return
-        self._kv_events = kv_events.events
+        self._kv_events.extend(kv_events.events)
         return
 
     def request_finished(

vllm/v1/worker/kv_connector_model_runner_mixin.py

@@ -12,6 +12,7 @@
 )
 
 from vllm.config import VllmConfig
+from vllm.distributed.kv_events import KVEventBatch
 from vllm.distributed.kv_transfer import (
     ensure_kv_transfer_shutdown,
     get_kv_transfer_group,
@@ -147,7 +148,7 @@ def get_kv_connector_stats() -> KVConnectorStats | None:
         return None
 
     @staticmethod
-    def get_kv_connector_kv_cache_events() -> KVConnectorStats | None:
+    def get_kv_connector_kv_cache_events() -> KVEventBatch | None:
         if has_kv_transfer_group():
             return get_kv_transfer_group().get_kv_connector_kv_cache_events()
         return None