Accept mapping header inputs in client configuration

Co-authored-by: Shri Sukhani <shrisukhani@users.noreply.github.com>

Author: cursoragent

hyperbrowser/config.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Dict, Optional
+from typing import Dict, Mapping, Optional
 import os
 
 from .exceptions import HyperbrowserError
@@ -18,8 +18,8 @@ def __post_init__(self) -> None:
             raise HyperbrowserError("api_key must be a string")
         if not isinstance(self.base_url, str):
             raise HyperbrowserError("base_url must be a string")
-        if self.headers is not None and not isinstance(self.headers, dict):
-            raise HyperbrowserError("headers must be a dictionary of string pairs")
+        if self.headers is not None and not isinstance(self.headers, Mapping):
+            raise HyperbrowserError("headers must be a mapping of string pairs")
         self.api_key = self.api_key.strip()
         if not self.api_key:
             raise HyperbrowserError("api_key must not be empty")
@@ -34,9 +34,7 @@ def __post_init__(self) -> None:
             normalized_headers: Dict[str, str] = {}
             for key, value in self.headers.items():
                 if not isinstance(key, str) or not isinstance(value, str):
-                    raise HyperbrowserError(
-                        "headers must be a dictionary of string pairs"
-                    )
+                    raise HyperbrowserError("headers must be a mapping of string pairs")
                 normalized_headers[key] = value
             self.headers = normalized_headers
 

tests/test_config.py

@@ -1,3 +1,5 @@
+from types import MappingProxyType
+
 import pytest
 
 from hyperbrowser.config import ClientConfig
@@ -61,7 +63,7 @@ def test_client_config_rejects_non_string_values():
     with pytest.raises(HyperbrowserError, match="base_url must be a string"):
         ClientConfig(api_key="test-key", base_url=None)  # type: ignore[arg-type]
 
-    with pytest.raises(HyperbrowserError, match="headers must be a dictionary"):
+    with pytest.raises(HyperbrowserError, match="headers must be a mapping"):
         ClientConfig(api_key="test-key", headers="x=1")  # type: ignore[arg-type]
 
     with pytest.raises(HyperbrowserError, match="api_key must not be empty"):
@@ -86,5 +88,12 @@ def test_client_config_normalizes_headers_to_internal_copy():
 
 
 def test_client_config_rejects_non_string_header_pairs():
-    with pytest.raises(HyperbrowserError, match="headers must be a dictionary"):
+    with pytest.raises(HyperbrowserError, match="headers must be a mapping"):
         ClientConfig(api_key="test-key", headers={"X-Correlation-Id": 123})  # type: ignore[dict-item]
+
+
+def test_client_config_accepts_mapping_header_inputs():
+    headers = MappingProxyType({"X-Correlation-Id": "abc123"})
+    config = ClientConfig(api_key="test-key", headers=headers)
+
+    assert config.headers == {"X-Correlation-Id": "abc123"}