Support endpoint_id for AzureSTTService

This commit is contained in:
Abdulrahman Alrahma
2025-08-10 22:24:47 +01:00
parent 241ab19228
commit fc67cc3302
2 changed files with 9 additions and 0 deletions

View File

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
## Unreleased
### Added
- Added `endpoint_id` parameter to `AzureSTTService`. ([Custom EndpointId](https://docs.azure.cn/en-us/ai-services/speech-service/how-to-recognize-speech?pivots=programming-language-python#use-a-custom-endpoint))
### Changed
- Updated `pyproject.toml` to once again pin `numba` to `>=0.61.2` in order to

View File

@@ -60,6 +60,7 @@ class AzureSTTService(STTService):
region: str,
language: Language = Language.EN_US,
sample_rate: Optional[int] = None,
endpoint_id: Optional[str] = None,
**kwargs,
):
"""Initialize the Azure STT service.
@@ -69,6 +70,7 @@ class AzureSTTService(STTService):
region: Azure region for the Speech service (e.g., 'eastus').
language: Language for speech recognition. Defaults to English (US).
sample_rate: Audio sample rate in Hz. If None, uses service default.
endpoint_id: Custom model endpoint id.
**kwargs: Additional arguments passed to parent STTService.
"""
super().__init__(sample_rate=sample_rate, **kwargs)
@@ -79,6 +81,9 @@ class AzureSTTService(STTService):
speech_recognition_language=language_to_azure_language(language),
)
if endpoint_id:
self._speech_config.endpoint_id = endpoint_id
self._audio_stream = None
self._speech_recognizer = None
self._settings = {