Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion sdk/batch/speechmatics/batch/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@
from ._models import SummarizationConfig
from ._models import TopicDetectionConfig
from ._models import Transcript
from ._models import TranscriptFilteringConfig
from ._models import TranscriptionConfig
from ._models import TranslationConfig

Expand All @@ -54,6 +55,7 @@
"NotificationContents",
"NotificationMethod",
"OperatingPoint",
"OutputConfig",
"SentimentAnalysisConfig",
"StaticKeyAuth",
"SummarizationConfig",
Expand All @@ -62,6 +64,6 @@
"Transcript",
"TranscriptionConfig",
"TranslationConfig",
"OutputConfig",
"TransportError",
"TranscriptFilteringConfig",
]
9 changes: 8 additions & 1 deletion sdk/batch/speechmatics/batch/_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,6 +100,7 @@ class TranscriptionConfig:
transcript_filtering_config: Configuration for filtering transcription.
defaults to None.
"""

language: str = "en"
operating_point: OperatingPoint = OperatingPoint.ENHANCED
output_locale: Optional[str] = None
Expand All @@ -121,6 +122,7 @@ def to_dict(self) -> dict[str, Any]:
result["transcript_filtering_config"] = self.transcript_filtering_config.to_dict()
return result


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


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


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


@dataclass
class JobConfig:
"""
Expand Down Expand Up @@ -360,7 +365,9 @@ def from_dict(cls, data: dict[str, Any]) -> JobConfig:
if "transcription_config" in data:
tc_data = data["transcription_config"].copy()
if "transcript_filtering_config" in tc_data and isinstance(tc_data["transcript_filtering_config"], dict):
tc_data["transcript_filtering_config"] = TranscriptFilteringConfig(**tc_data["transcript_filtering_config"])
tc_data["transcript_filtering_config"] = TranscriptFilteringConfig(
**tc_data["transcript_filtering_config"]
)
transcription_config = TranscriptionConfig(**tc_data)

alignment_config = None
Expand Down