feat: add support for STT orchestration

Author: hbredin

CHANGELOG.md

@@ -1,5 +1,9 @@
 # CHANGELOG
 
+## Version 0.4.0 (2026-01-14)
+
+- feat: add support for STT orchestration
+
 ## Version 0.3.0 (2025-10-10)
 
 - feat: add support for passing audio as a `{"audio": str | Path}` mapping

README.md

@@ -34,6 +34,21 @@ diarization = client.retrieve(job_id)
 
 Use `help(client.diarize)` to learn about options.
 
+## STT orchestration
+
+```python
+# submit a diarization job with STT orchestration
+job_id = client.diarize(
+  media_url, 
+  transcription=True,
+  transcription_config={"model": "parakeet-tdt-0.6b-v3"})
+
+# retrieve speaker-attributed transcription
+orchestration = client.retrieve(job_id)
+# orchestration['output']['turnLevelTranscription'] 
+# orchestration['output']['wordLevelTranscription']
+```
+
 ## Speaker identification
 
 ```python

src/pyannoteai/sdk/client.py

@@ -370,6 +370,8 @@ def diarize(
         turn_level_confidence: bool = False,
         exclusive: bool = False,
         model: str = "precision-2",
+        transcription: bool = False,
+        transcription_config: dict | None = None,
         **kwargs,
     ) -> str:
         """Initiate a diarization job on the pyannoteAI web API
@@ -394,6 +396,10 @@ def diarize(
             Enable exclusive speaker diarization.
         model : str, optional
             Defaults to "precision-2"
+        transcription : bool, optional
+            Enable STT orchestration.
+        transcription_config : dict, optional
+            STT configuration parameters, including model selection.
         **kwargs : optional
             Extra arguments to send in the body of the request.
 
@@ -416,7 +422,10 @@ def diarize(
             "confidence": confidence,
             "turnLevelConfidence": turn_level_confidence,
             "exclusive": exclusive,
+            "transcription": transcription,
+            "transcriptionConfig": transcription_config or dict(),
         }
+        
         # add extra arguments to the request body
         json.update(kwargs)