Make reset_cache optionally return None for unimplemented connectors

Signed-off-by: tovam <tovam@pliops.com>

Author: ptovam

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

@@ -432,11 +432,11 @@ def build_kv_connector_stats(
         """
         return None
 
-    def reset_cache(self) -> bool:
+    def reset_cache(self) -> bool | None:
         """
         Reset the connector's internal cache.
 
         Returns:
             bool: True if the cache was successfully reset, False otherwise.
         """
-        return False
+        return None

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

@@ -344,4 +344,4 @@ def get_kv_connector_stats(self) -> MultiKVConnectorStats | None:
         return stats_by_connector
 
     def reset_cache(self) -> bool:
-        return all(connector.reset_cache() for connector in self._connectors)
+        return all(c.reset_cache() is not False for c in self._connectors)

vllm/v1/core/sched/scheduler.py

@@ -1244,7 +1244,7 @@ def reset_connector_cache(self) -> bool:
         if self.connector is None:
             logger.warning("reset_connector called but no KV connector configured.")
             return False
-        return self.connector.reset_cache()
+        return self.connector.reset_cache() is not False
 
     def make_stats(
         self,