Skip to content

chore: sync sdk code with DeepLearning repo #115

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 8 commits into from
Apr 23, 2025
2 changes: 1 addition & 1 deletion assemblyai/__version__.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "0.39.0"
__version__ = "0.39.1"
31 changes: 31 additions & 0 deletions assemblyai/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -597,6 +597,13 @@ class RawTranscriptionConfig(BaseModel):
"""
The speech model to use for the transcription.
"""

prompt: Optional[str] = None
"The prompt used to generate the transcript with the Slam-1 speech model. Can't be used together with `keyterms_prompt`."

keyterms_prompt: Optional[List[str]] = None
"The list of key terms used to generate the transcript with the Slam-1 speech model. Can't be used together with `prompt`."

model_config = ConfigDict(extra="allow")


Expand Down Expand Up @@ -640,6 +647,8 @@ def __init__(
speech_threshold: Optional[float] = None,
raw_transcription_config: Optional[RawTranscriptionConfig] = None,
speech_model: Optional[SpeechModel] = None,
prompt: Optional[str] = None,
keyterms_prompt: Optional[List[str]] = None,
) -> None:
"""
Args:
Expand Down Expand Up @@ -728,6 +737,8 @@ def __init__(
self.language_confidence_threshold = language_confidence_threshold
self.speech_threshold = speech_threshold
self.speech_model = speech_model
self.prompt = prompt
self.keyterms_prompt = keyterms_prompt

@property
def raw(self) -> RawTranscriptionConfig:
Expand Down Expand Up @@ -756,6 +767,26 @@ def speech_model(self, speech_model: Optional[SpeechModel]) -> None:
"Sets the speech model to use for the transcription."
self._raw_transcription_config.speech_model = speech_model

@property
def prompt(self) -> Optional[str]:
"The prompt to use for the transcription."
return self._raw_transcription_config.prompt

@prompt.setter
def prompt(self, prompt: Optional[str]) -> None:
"Sets the prompt to use for the transcription."
self._raw_transcription_config.prompt = prompt

@property
def keyterms_prompt(self) -> Optional[List[str]]:
"The keyterms_prompt to use for the transcription."
return self._raw_transcription_config.keyterms_prompt

@keyterms_prompt.setter
def keyterms_prompt(self, keyterms_prompt: Optional[List[str]]) -> None:
"Sets the prompt to use for the transcription."
self._raw_transcription_config.keyterms_prompt = keyterms_prompt

@property
def punctuate(self) -> Optional[bool]:
"Returns the status of the Automatic Punctuation feature."
Expand Down