gsadmin/meet
1
0
Fork 0
mirror of https://github.com/suitenumerique/meet.git synced 2026-08-10 18:57:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛(summary) relax whisperX payload format

Browse Source 
Sometimes whisperX response is partial, we don't
want to crash in such case.
This commit is contained in:
Florent Chehab
2026-04-02 23:07:00 +02:00
parent 45c5a443fb
commit 451be40bb7
4 changed files with 81 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/summary/tests/models/__init__.py
+1
View File
@@ -0,0 +1 @@
"""Tests for models."""
src/summary/tests/models/test_whisper_x_response.py
+60
View File
@@ -0,0 +1,60 @@
"""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": "?"},
],
}
)