Add support for brotli compression

Author: joshuanapoli

poetry.lock

@@ -8,6 +8,7 @@ requires-python = ">=3.10"
 dependencies = [
     "pydantic>=2.12.0",
     "pyarrow>=22.0.0",
+    "brotli (>=1.2.0,<2.0.0)",
 ]
 license = "MIT"
 license-files = ['LICENSE']

pyproject.toml

@@ -4,6 +4,8 @@
 import os
 import time
 import zlib
+
+import brotli
 from datetime import datetime
 from typing import Any, Dict, List, Optional
 from urllib.error import HTTPError, URLError
@@ -113,7 +115,7 @@ def _get_headers(self) -> Dict[str, str]:
             "Authorization": f"Bearer {self._access_token}",
             "Content-Type": "application/json",
             "Accept": "application/json",
-            "Accept-Encoding": "gzip, deflate",
+            "Accept-Encoding": "br, gzip, deflate",
         }
 
     @staticmethod
@@ -125,7 +127,9 @@ def _read_response(response: Any) -> tuple[bytes, str]:
         """
         raw = response.read()
         encoding = response.headers.get("Content-Encoding", "")
-        if encoding == "gzip":
+        if encoding == "br":
+            raw = brotli.decompress(raw)
+        elif encoding == "gzip":
             raw = gzip.decompress(raw)
         elif encoding == "deflate":
             raw = zlib.decompress(raw)

src/cvec/cvec.py

@@ -1,11 +1,13 @@
-"""Tests for HTTP compression support (gzip/deflate)."""
+"""Tests for HTTP compression support (gzip/deflate/brotli)."""
 
 import gzip
 import json
 import zlib
 from typing import Any
 from unittest.mock import Mock, patch
 
+import brotli
+
 from cvec import CVec
 
 
@@ -63,7 +65,7 @@ def test_accept_encoding_header_sent(
         mock_fetch_key: Any,
         mock_login: Any,
     ) -> None:
-        """Verify Accept-Encoding: gzip, deflate is present in request headers."""
+        """Verify Accept-Encoding includes br, gzip, deflate."""
         client = _create_client()
 
         mock_response = _make_mock_response(b"[]")
@@ -72,7 +74,7 @@ def test_accept_encoding_header_sent(
         client.get_metrics()
 
         req = mock_urlopen.call_args[0][0]
-        assert req.get_header("Accept-encoding") == "gzip, deflate"
+        assert req.get_header("Accept-encoding") == "br, gzip, deflate"
 
     @patch.object(CVec, "_login_with_supabase", return_value=None)
     @patch.object(
@@ -196,3 +198,29 @@ def test_gzip_arrow_response_decompressed(
         result = client._make_request("GET", "/api/metrics/data/arrow")
         assert isinstance(result, bytes)
         assert result == arrow_bytes
+
+    @patch.object(CVec, "_login_with_supabase", return_value=None)
+    @patch.object(
+        CVec,
+        "_fetch_config",
+        autospec=True,
+        side_effect=mock_fetch_config_side_effect,
+    )
+    @patch("cvec.cvec.urlopen")
+    def test_brotli_response_decompressed(
+        self,
+        mock_urlopen: Any,
+        mock_fetch_key: Any,
+        mock_login: Any,
+    ) -> None:
+        """Mock a brotli-compressed JSON response, verify it is decompressed."""
+        client = _create_client()
+
+        data = [{"id": 4, "name": "metric4"}]
+        compressed = brotli.compress(json.dumps(data).encode("utf-8"))
+        mock_response = _make_mock_response(compressed, content_encoding="br")
+        mock_urlopen.return_value = mock_response
+
+        result = client.get_metrics()
+        assert len(result) == 1
+        assert result[0].name == "metric4"