Skip to content

Commit 4d64231

Browse files
authored
Fix import TranscriptFilteringConfig in batch sdk (#91)
1 parent 528958a commit 4d64231

2 files changed

Lines changed: 11 additions & 2 deletions

File tree

sdk/batch/speechmatics/batch/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
from ._models import SummarizationConfig
3030
from ._models import TopicDetectionConfig
3131
from ._models import Transcript
32+
from ._models import TranscriptFilteringConfig
3233
from ._models import TranscriptionConfig
3334
from ._models import TranslationConfig
3435

@@ -54,6 +55,7 @@
5455
"NotificationContents",
5556
"NotificationMethod",
5657
"OperatingPoint",
58+
"OutputConfig",
5759
"SentimentAnalysisConfig",
5860
"StaticKeyAuth",
5961
"SummarizationConfig",
@@ -62,6 +64,6 @@
6264
"Transcript",
6365
"TranscriptionConfig",
6466
"TranslationConfig",
65-
"OutputConfig",
6667
"TransportError",
68+
"TranscriptFilteringConfig",
6769
]

sdk/batch/speechmatics/batch/_models.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,7 @@ class TranscriptionConfig:
100100
transcript_filtering_config: Configuration for filtering transcription.
101101
defaults to None.
102102
"""
103+
103104
language: str = "en"
104105
operating_point: OperatingPoint = OperatingPoint.ENHANCED
105106
output_locale: Optional[str] = None
@@ -121,6 +122,7 @@ def to_dict(self) -> dict[str, Any]:
121122
result["transcript_filtering_config"] = self.transcript_filtering_config.to_dict()
122123
return result
123124

125+
124126
@dataclass
125127
class OutputConfig:
126128
"""Configuration for output formatting."""
@@ -132,6 +134,7 @@ def to_dict(self) -> dict[str, Any]:
132134
"""Convert to dictionary, excluding None values."""
133135
return {k: v for k, v in asdict(self).items() if v is not None}
134136

137+
135138
@dataclass
136139
class AlignmentConfig:
137140
"""Configuration for alignment jobs."""
@@ -268,6 +271,7 @@ def to_dict(self) -> dict[str, Any]:
268271
"""Convert to dictionary, excluding None values."""
269272
return {k: v for k, v in asdict(self).items() if v is not None}
270273

274+
271275
@dataclass
272276
class TranscriptFilteringConfig:
273277
"""Configuration for transcript filtering."""
@@ -279,6 +283,7 @@ def to_dict(self) -> dict[str, Any]:
279283
"""Convert to dictionary, excluding None values."""
280284
return {k: v for k, v in asdict(self).items() if v is not None}
281285

286+
282287
@dataclass
283288
class JobConfig:
284289
"""
@@ -360,7 +365,9 @@ def from_dict(cls, data: dict[str, Any]) -> JobConfig:
360365
if "transcription_config" in data:
361366
tc_data = data["transcription_config"].copy()
362367
if "transcript_filtering_config" in tc_data and isinstance(tc_data["transcript_filtering_config"], dict):
363-
tc_data["transcript_filtering_config"] = TranscriptFilteringConfig(**tc_data["transcript_filtering_config"])
368+
tc_data["transcript_filtering_config"] = TranscriptFilteringConfig(
369+
**tc_data["transcript_filtering_config"]
370+
)
364371
transcription_config = TranscriptionConfig(**tc_data)
365372

366373
alignment_config = None

0 commit comments

Comments
 (0)