feat: set brotli compression quality

Author: anna-singleton-resolver

src/resolver_athena_client/client/athena_client.py

@@ -177,7 +177,7 @@ async def classify_single(
 
         # Apply compression if enabled
         if self.options.compress_images:
-            processed_image = compress_image(processed_image)
+            processed_image = compress_image(processed_image, self.options.compression_quality)
 
         request_encoding = (
             RequestEncoding.REQUEST_ENCODING_BROTLI
@@ -243,7 +243,7 @@ async def transform_image(image_data: ImageData) -> ClassificationInput:
 
             # Apply compression if enabled
             if self.options.compress_images:
-                compressed_image = compress_image(resized_image)
+                compressed_image = compress_image(resized_image, self.options.compression_quality)
             else:
                 compressed_image = resized_image
 

src/resolver_athena_client/client/athena_options.py

@@ -66,3 +66,4 @@ class AthenaOptions:
     correlation_provider: type[CorrelationProvider] = HashCorrelationProvider
     timeout: float | None = 120.0
     keepalive_interval: float | None = None
+    compression_quality: int = 11  # Brotli quality level (0-11)

src/resolver_athena_client/client/transformers/core.py

@@ -82,19 +82,21 @@ def process_image() -> tuple[bytes, bool]:
     return image_data
 
 
-def compress_image(image_data: ImageData) -> ImageData:
+def compress_image(image_data: ImageData, quality: int = 11) -> ImageData:
     """Compress image data using Brotli compression.
 
     Args:
     ----
         image_data: The ImageData object to compress
+        quality: Compression quality level (0-11), higher is better compression
+            but slower. Default is 11 for maximum compression.
 
     Returns:
     -------
         The same ImageData object with compressed data (modified in-place)
 
     """
-    compressed_bytes = brotli.compress(image_data.data)
+    compressed_bytes = brotli.compress(image_data.data, quality=quality)
     # Modify existing ImageData with compressed bytes but preserve hashes
     # since compression doesn't change image content
     image_data.data = compressed_bytes