From e1c27feacc76c800e266ef665ea035e41d626192 Mon Sep 17 00:00:00 2001 From: khresth Date: Fri, 24 Oct 2025 10:18:33 +0100 Subject: [PATCH] replaced manual comparison with direct slice comparison --- src/mistralai/utils/eventstreaming.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/src/mistralai/utils/eventstreaming.py b/src/mistralai/utils/eventstreaming.py index 0969899..4605eff 100644 --- a/src/mistralai/utils/eventstreaming.py +++ b/src/mistralai/utils/eventstreaming.py @@ -238,11 +238,9 @@ def _parse_event( def _peek_sequence(position: int, buffer: bytearray, sequence: bytes): - if len(sequence) > (len(buffer) - position): + buffer_len = len(buffer) + seq_len = len(sequence) + if position + seq_len > buffer_len: return None - - for i, seq in enumerate(sequence): - if buffer[position + i] != seq: - return None - - return sequence + view = memoryview(buffer) + return sequence if view[position:position + seq_len] == sequence else None