mirror of
https://github.com/suitenumerique/meet.git
synced 2026-08-11 19:27:50 +00:00
Florent Chehab
451be40bb7
Sometimes whisperX response is partial, we don't want to crash in such case.
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
"""Tests for WhisperXResponse."""
|
|
|
|
from summary.core.shared_models import WhisperXResponse
|
|
|
|
|
|
def test_whisper_x_response_partial_response_is_valid():
|
|
"""A partial response from WhisperX is valid.
|
|
|
|
Sometimes WhisperX Api doesn't return all the data and timestamps,
|
|
we don't want to fail the whole process.
|
|
"""
|
|
WhisperXResponse.model_validate(
|
|
{
|
|
"segments": [
|
|
{
|
|
"start": 1.135,
|
|
"end": 7.3,
|
|
"text": " Test 1, 2, 3... Est-?",
|
|
"words": [
|
|
{
|
|
"word": "Test",
|
|
"start": 1.135,
|
|
"end": 2.216,
|
|
"score": 0.654,
|
|
"speaker": "SPEAKER_00",
|
|
},
|
|
{
|
|
"word": "1,",
|
|
"start": None,
|
|
"end": None,
|
|
"score": None,
|
|
"speaker": None,
|
|
},
|
|
{
|
|
"word": "2,",
|
|
"start": None,
|
|
"end": None,
|
|
"score": None,
|
|
"speaker": None,
|
|
},
|
|
{
|
|
"word": "3...",
|
|
},
|
|
],
|
|
"speaker": "SPEAKER_00",
|
|
}
|
|
],
|
|
"word_segments": [
|
|
{
|
|
"word": "Test",
|
|
"start": 1.135,
|
|
"end": 2.216,
|
|
"score": 0.654,
|
|
"speaker": "SPEAKER_00",
|
|
},
|
|
{"word": "1,"},
|
|
{"word": "?"},
|
|
],
|
|
}
|
|
)
|