From 384c7558b2b587b31298aa7ce2953b43fe749523 Mon Sep 17 00:00:00 2001 From: shrey kharbanda Date: Thu, 24 Sep 2026 00:37:19 +0000 Subject: [PATCH] refactor(images): keep format limits with the module constants --- litellm/images/dimensions.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/litellm/images/dimensions.py b/litellm/images/dimensions.py index 6ecb74707e1..f9cc2a27c93 100644 --- a/litellm/images/dimensions.py +++ b/litellm/images/dimensions.py @@ -11,6 +11,13 @@ from litellm.types.llms.openai import FileTypes _HEADER_READ_SIZE: Final = 32 +_JPEG_SOF_MARKERS: Final = frozenset(range(0xC0, 0xD0)) - {0xC4, 0xC8, 0xCC} +_JPEG_MAX_SEGMENTS: Final = 1024 +_JPEG_MAX_HEADER_OFFSET: Final = 16 * 1024 * 1024 +_JPEG_FIRST_SEGMENT_OFFSET: Final = 2 +_JPEG_SOF_PAYLOAD_SIZE: Final = 5 +_JPEG_FILL_CHUNK: Final = 64 * 1024 + @dataclass(frozen=True, slots=True) class ImageDimensions: @@ -117,14 +124,6 @@ def _webp_dimensions(head: bytes) -> ImageDimensions | None: return None -_JPEG_SOF_MARKERS: Final = frozenset(range(0xC0, 0xD0)) - {0xC4, 0xC8, 0xCC} -_JPEG_MAX_SEGMENTS: Final = 1024 -_JPEG_MAX_HEADER_OFFSET: Final = 16 * 1024 * 1024 -_JPEG_FIRST_SEGMENT_OFFSET: Final = 2 -_JPEG_SOF_PAYLOAD_SIZE: Final = 5 -_JPEG_FILL_CHUNK: Final = 64 * 1024 - - def _jpeg_fill_run(stream: IO[bytes], start: int, offset: int) -> int: stream.seek(start + offset) fill: Final = stream.read(_JPEG_FILL_CHUNK)