make header name const

Author: Georgios Hadjiharalambous

sdk/batch/speechmatics/batch/__init__.py

@@ -32,9 +32,11 @@
 from ._models import TranscriptFilteringConfig
 from ._models import TranscriptionConfig
 from ._models import TranslationConfig
+from ._transport import PROCESSING_DATA_HEADER
 
 __all__ = [
     "AsyncClient",
+    "PROCESSING_DATA_HEADER",
     "AuthBase",
     "AuthenticationError",
     "AutoChaptersConfig",

sdk/batch/speechmatics/batch/_async_client.py

@@ -31,6 +31,7 @@
 from ._models import JobType
 from ._models import Transcript
 from ._models import TranscriptionConfig
+from ._transport import PROCESSING_DATA_HEADER
 from ._transport import Transport
 
 
@@ -537,7 +538,7 @@ async def _submit_and_create_job_details(
         """Submit job and create JobDetails response."""
         extra_headers: Optional[dict[str, dict[str, int]]] = None
         if requested_parallel is not None:
-            extra_headers = {"X-SM-Processing-Data": {"requested_parallel": requested_parallel}}
+            extra_headers = {PROCESSING_DATA_HEADER: {"requested_parallel": requested_parallel}}
         response = await self._transport.post("/jobs", multipart_data=multipart_data, extra_headers=extra_headers)
         job_id = response.get("id")
         if not job_id:

sdk/batch/speechmatics/batch/_transport.py

@@ -25,6 +25,8 @@
 from ._logging import get_logger
 from ._models import ConnectionConfig
 
+PROCESSING_DATA_HEADER = "X-SM-Processing-Data"
+
 
 class Transport:
     """

tests/batch/test_submit_job.py

@@ -13,6 +13,7 @@
 from speechmatics.batch import JobStatus
 from speechmatics.batch import JobType
 from speechmatics.batch import TranscriptionConfig
+from speechmatics.batch import PROCESSING_DATA_HEADER
 
 
 def _make_client(api_key: str = "test-key") -> AsyncClient:
@@ -53,8 +54,8 @@ async def test_header_sent_when_requested_parallel_provided(self):
 
         extra_headers = _captured_extra_headers(mock_post)
         assert extra_headers is not None
-        assert "X-SM-Processing-Data" in extra_headers
-        payload = extra_headers["X-SM-Processing-Data"]
+        assert PROCESSING_DATA_HEADER in extra_headers
+        payload = extra_headers[PROCESSING_DATA_HEADER]
         assert payload == {"requested_parallel": 4}
 
     @pytest.mark.asyncio
@@ -81,7 +82,7 @@ async def test_header_value_is_valid_json(self):
         extra_headers = _captured_extra_headers(mock_post)
         # Must be parseable JSON
         assert extra_headers is not None
-        parsed = extra_headers["X-SM-Processing-Data"]
+        parsed = extra_headers[PROCESSING_DATA_HEADER]
         assert parsed["requested_parallel"] == 8
 
     @pytest.mark.asyncio
@@ -95,7 +96,7 @@ async def test_requested_parallel_one(self):
 
         extra_headers = _captured_extra_headers(mock_post)
         assert extra_headers is not None
-        payload = extra_headers["X-SM-Processing-Data"]
+        payload = extra_headers[PROCESSING_DATA_HEADER]
         assert payload["requested_parallel"] == 1
 
     @pytest.mark.asyncio
@@ -120,7 +121,7 @@ async def test_header_sent_with_fetch_data_config(self):
 
         extra_headers = _captured_extra_headers(mock_post)
         assert extra_headers is not None
-        payload = extra_headers["X-SM-Processing-Data"]
+        payload = extra_headers[PROCESSING_DATA_HEADER]
         assert payload == {"requested_parallel": 2}