From 095e816347b6e123ef24d630d9b8aa71f36be412 Mon Sep 17 00:00:00 2001 From: Kotesh Kumar Yelamati Date: Sun, 12 Jul 2026 10:11:02 -0400 Subject: [PATCH] Fix OverflowError in offsets_for_times() when timestamp is float infinity Add error handling for invalid timestamps in Kafka consumer. --- kafka/consumer/group.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/kafka/consumer/group.py b/kafka/consumer/group.py index ed4530797..8e071b3a2 100644 --- a/kafka/consumer/group.py +++ b/kafka/consumer/group.py @@ -1228,7 +1228,12 @@ def offsets_for_times(self, timestamps, timeout_ms=None): """ timeout_ms = self.config['request_timeout_ms'] if timeout_ms is None else timeout_ms for tp, ts in timestamps.items(): - timestamps[tp] = int(ts) + try: + timestamps[tp] = int(ts) + except (ValueError, TypeError, OverflowError): + raise ValueError( + "The target timestamp for partition {} is {}. Timestamps " + "must be a valid integer (milliseconds since epoch).".format(tp, ts)) if ts < 0: raise ValueError( "The target time for partition {} is {}. The target time "