Skip to content

Commit d1ddb4a

Browse files
kirillatrevHristo Stoychev
kirillatrev
and
Hristo Stoychev
authored
Add support for diarization_type, speakers_count, remove_atmospherics (revdotcom#113)
* Add support for diarization_type, speakers_count, remove_atmospherics Co-authored-by: Hristo Stoychev <[email protected]>
1 parent e46f5cf commit d1ddb4a

File tree

5 files changed

+98
-29
lines changed

5 files changed

+98
-29
lines changed

‎examples/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
1-
rev-ai==2.19.4
1+
rev-ai==2.19.5
22
pyaudio==0.2.11
33
six==1.12.0

‎src/rev_ai/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# -*- coding: utf-8 -*-
22
"""Top-level package for rev_ai"""
33

4-
__version__ = '2.19.4'
4+
__version__ = '2.19.5'
55

66
from .models import Job, JobStatus, Account, Transcript, Monologue, Element, MediaConfig, \
77
CaptionType, CustomVocabulary, TopicExtractionJob, TopicExtractionResult, Topic, Informant, \

‎src/rev_ai/apiclient.py

Lines changed: 55 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ def submit_job_url(
7171
skip_postprocessing=False,
7272
remove_atmospherics=False,
7373
speakers_count=None,
74+
diarization_type=None,
7475
summarization_config: SummarizationOptions = None,
7576
translation_config: TranslationOptions = None):
7677
"""Submit media given a URL for transcription.
@@ -122,21 +123,36 @@ def submit_job_url(
122123
:param remove_atmospherics: Atmospherics such as <laugh>, <affirmative>, etc. will not
123124
appear in the transcript.
124125
:param speakers_count: Use to specify the total number of unique speakers in the audio.
126+
:param diarization_type: Use to specify diarization type.
125127
:param summarization_config: Use to request transcript summary.
126128
:param translation_config: Use to request transcript translation.
127129
:returns: raw response data
128130
:raises: HTTPError
129131
"""
130-
payload = self._create_job_options_payload(media_url, metadata, callback_url,
131-
skip_diarization, skip_punctuation,
132-
speaker_channels_count,
133-
custom_vocabularies, filter_profanity,
134-
remove_disfluencies, delete_after_seconds,
135-
language, custom_vocabulary_id, transcriber,
136-
verbatim, rush, test_mode,
137-
segments_to_transcribe, speaker_names,
138-
source_config, notification_config,
139-
skip_postprocessing,
132+
payload = self._create_job_options_payload(media_url=media_url,
133+
metadata=metadata,
134+
callback_url=callback_url,
135+
skip_diarization=skip_diarization,
136+
skip_punctuation=skip_punctuation,
137+
speaker_channels_count=speaker_channels_count,
138+
custom_vocabularies=custom_vocabularies,
139+
filter_profanity=filter_profanity,
140+
remove_disfluencies=remove_disfluencies,
141+
delete_after_seconds=delete_after_seconds,
142+
language=language,
143+
custom_vocabulary_id=custom_vocabulary_id,
144+
transcriber=transcriber,
145+
verbatim=verbatim,
146+
rush=rush,
147+
test_mode=test_mode,
148+
segments_to_transcribe=segments_to_transcribe,
149+
speaker_names=speaker_names,
150+
source_config=source_config,
151+
notification_config=notification_config,
152+
skip_postprocessing=skip_postprocessing,
153+
remove_atmospherics=remove_atmospherics,
154+
speakers_count=speakers_count,
155+
diarization_type=diarization_type,
140156
summarization_config=summarization_config,
141157
translation_config=translation_config)
142158

@@ -172,6 +188,7 @@ def submit_job_local_file(
172188
skip_postprocessing=False,
173189
remove_atmospherics=False,
174190
speakers_count=None,
191+
diarization_type=None,
175192
summarization_config: SummarizationOptions = None,
176193
translation_config: TranslationOptions = None):
177194
"""Submit a local file for transcription.
@@ -220,6 +237,7 @@ def submit_job_local_file(
220237
:param remove_atmospherics: Atmospherics such as <laugh>, <affirmative>, etc. will not
221238
appear in the transcript.
222239
:param speakers_count: Use to specify the total number of unique speakers in the audio.
240+
:param diarization_type: Use to specify diarization type.
223241
:param summarization_config: Use to request transcript summary.
224242
:param translation_config: Use to request transcript translation.
225243
:returns: raw response data
@@ -228,15 +246,30 @@ def submit_job_local_file(
228246
if not filename:
229247
raise ValueError('filename must be provided')
230248

231-
payload = self._create_job_options_payload(None, metadata, callback_url,
232-
skip_diarization, skip_punctuation,
233-
speaker_channels_count,
234-
custom_vocabularies, filter_profanity,
235-
remove_disfluencies, delete_after_seconds,
236-
language, custom_vocabulary_id, transcriber,
237-
verbatim, rush, test_mode,
238-
segments_to_transcribe, speaker_names, None,
239-
notification_config, skip_postprocessing,
249+
payload = self._create_job_options_payload(media_url=None,
250+
metadata=metadata,
251+
callback_url=callback_url,
252+
skip_diarization=skip_diarization,
253+
skip_punctuation=skip_punctuation,
254+
speaker_channels_count=speaker_channels_count,
255+
custom_vocabularies=custom_vocabularies,
256+
filter_profanity=filter_profanity,
257+
remove_disfluencies=remove_disfluencies,
258+
delete_after_seconds=delete_after_seconds,
259+
language=language,
260+
custom_vocabulary_id=custom_vocabulary_id,
261+
transcriber=transcriber,
262+
verbatim=verbatim,
263+
rush=rush,
264+
test_mode=test_mode,
265+
segments_to_transcribe=segments_to_transcribe,
266+
speaker_names=speaker_names,
267+
source_config=None,
268+
notification_config=notification_config,
269+
skip_postprocessing=skip_postprocessing,
270+
remove_atmospherics=remove_atmospherics,
271+
speakers_count=speakers_count,
272+
diarization_type=diarization_type,
240273
summarization_config=summarization_config,
241274
translation_config=translation_config)
242275

@@ -714,6 +747,7 @@ def _create_job_options_payload(
714747
skip_postprocessing=False,
715748
remove_atmospherics=None,
716749
speakers_count=None,
750+
diarization_type=None,
717751
summarization_config: SummarizationOptions = None,
718752
translation_config: TranslationOptions = None):
719753
payload = {}
@@ -764,6 +798,8 @@ def _create_job_options_payload(
764798
payload['remove_atmospherics'] = remove_atmospherics
765799
if speakers_count:
766800
payload['speakers_count'] = speakers_count
801+
if diarization_type:
802+
payload['diarization_type'] = diarization_type
767803
if summarization_config:
768804
payload['summarization_config'] = summarization_config.to_dict()
769805
if translation_config:

‎src/rev_ai/models/asynchronous/job.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def __init__(
3030
segments_to_transcribe=None,
3131
remove_atmospherics=None,
3232
speakers_count=None,
33+
diarization_type=None,
3334
summarization: Summarization = None,
3435
translation: Translation = None):
3536
"""
@@ -62,6 +63,7 @@ def __init__(
6263
:param remove_atmospherics: Atmospherics such as <laugh>, <affirmative>, etc. will noT
6364
appear in the transcript.
6465
:param speakers_count: Use to specify the total number of unique speakers in the audio.
66+
:param diarization_type: Use to specify diarization type.
6567
"""
6668
self.id = id_
6769
self.created_on = created_on
@@ -88,6 +90,7 @@ def __init__(
8890
self.segments_to_transcribe = segments_to_transcribe
8991
self.remove_atmospherics = remove_atmospherics
9092
self.speakers_count = speakers_count
93+
self.diarization_type = diarization_type
9194
self.summarization = summarization
9295
self.translation = translation
9396

@@ -126,6 +129,7 @@ def from_json(cls, json):
126129
segments_to_transcribe=json.get('segments_to_transcribe'),
127130
remove_atmospherics=json.get('remove_atmospherics'),
128131
speakers_count=json.get('speakers_count'),
132+
diarization_type=json.get('diarization_type'),
129133
summarization=Summarization.from_json(json.get('summarization')),
130134
translation=Translation.from_json(json.get('translation'))
131135
)

‎tests/test_job.py

Lines changed: 37 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,10 @@ def test_submit_job_url_with_success(self, mock_session, make_mock_response):
120120
'delete_after_seconds': 0,
121121
'language': LANGUAGE,
122122
'transcriber': TRANSCRIBER,
123-
'skip_postprocessing': True
123+
'skip_postprocessing': True,
124+
'remove_atmospherics': True,
125+
'speakers_count': 123,
126+
'diarization_type': "premium"
124127
}
125128
response = make_mock_response(url=JOB_ID_URL, json_data=data)
126129
mock_session.request.return_value = response
@@ -130,7 +133,12 @@ def test_submit_job_url_with_success(self, mock_session, make_mock_response):
130133
NOTIFICATION_URL, True,
131134
True, 1, CUSTOM_VOCAB, True,
132135
True, 0, LANGUAGE, CUSTOM_VOCAB_ID,
133-
TRANSCRIBER, skip_postprocessing=True)
136+
TRANSCRIBER,
137+
skip_postprocessing=True,
138+
remove_atmospherics=True,
139+
speakers_count=123,
140+
diarization_type="premium"
141+
)
134142

135143
assert res == Job(JOB_ID,
136144
CREATED_ON,
@@ -143,7 +151,11 @@ def test_submit_job_url_with_success(self, mock_session, make_mock_response):
143151
remove_disfluencies=True,
144152
delete_after_seconds=0,
145153
language=LANGUAGE,
146-
transcriber=TRANSCRIBER)
154+
transcriber=TRANSCRIBER,
155+
remove_atmospherics=True,
156+
speakers_count=123,
157+
diarization_type="premium"
158+
)
147159
mock_session.request.assert_called_once_with(
148160
"POST",
149161
JOBS_URL,
@@ -161,7 +173,10 @@ def test_submit_job_url_with_success(self, mock_session, make_mock_response):
161173
'language': LANGUAGE,
162174
'custom_vocabulary_id': CUSTOM_VOCAB_ID,
163175
'transcriber': TRANSCRIBER,
164-
'skip_postprocessing': True
176+
'skip_postprocessing': True,
177+
'remove_atmospherics': True,
178+
'speakers_count': 123,
179+
'diarization_type': "premium"
165180
},
166181
headers=client.default_headers)
167182

@@ -283,7 +298,10 @@ def test_submit_job_local_file_with_success(self, mocker, mock_session, make_moc
283298
'delete_after_seconds': 0,
284299
'language': LANGUAGE,
285300
'transcriber': TRANSCRIBER,
286-
'skip_postprocessing': True
301+
'skip_postprocessing': True,
302+
'remove_atmospherics': True,
303+
'speakers_count': 123,
304+
'diarization_type': "premium"
287305
}
288306
response = make_mock_response(url=JOB_ID_URL, json_data=data)
289307
mock_session.request.return_value = response
@@ -294,7 +312,11 @@ def test_submit_job_local_file_with_success(self, mocker, mock_session, make_moc
294312
NOTIFICATION_URL, True,
295313
True, 1, CUSTOM_VOCAB, True,
296314
True, 0, LANGUAGE, CUSTOM_VOCAB_ID,
297-
TRANSCRIBER, skip_postprocessing=True)
315+
TRANSCRIBER, skip_postprocessing=True,
316+
remove_atmospherics=True,
317+
speakers_count=123,
318+
diarization_type="premium"
319+
)
298320

299321
assert res == Job(JOB_ID,
300322
CREATED_ON,
@@ -308,7 +330,11 @@ def test_submit_job_local_file_with_success(self, mocker, mock_session, make_moc
308330
remove_disfluencies=True,
309331
delete_after_seconds=0,
310332
language=LANGUAGE,
311-
transcriber=TRANSCRIBER)
333+
transcriber=TRANSCRIBER,
334+
remove_atmospherics=True,
335+
speakers_count=123,
336+
diarization_type="premium"
337+
)
312338
mock_session.request.assert_called_once_with(
313339
"POST",
314340
JOBS_URL,
@@ -329,7 +355,10 @@ def test_submit_job_local_file_with_success(self, mocker, mock_session, make_moc
329355
'language': LANGUAGE,
330356
'custom_vocabulary_id': CUSTOM_VOCAB_ID,
331357
'transcriber': TRANSCRIBER,
332-
'skip_postprocessing': True
358+
'skip_postprocessing': True,
359+
'remove_atmospherics': True,
360+
'speakers_count': 123,
361+
'diarization_type': "premium"
333362
}, sort_keys=True)
334363
)
335364
},

0 commit comments

Comments
 (0)