From 26e34aec2ddb8b6a0b09c8a4907cbcfd6ab708fb Mon Sep 17 00:00:00 2001 From: Mathieu Virbel Date: Wed, 9 Aug 2023 11:23:28 +0200 Subject: [PATCH] server: ensure transcript status model is updated + tests --- server/reflector/views/transcripts.py | 1 + server/tests/test_transcripts_rtc_ws.py | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/server/reflector/views/transcripts.py b/server/reflector/views/transcripts.py index 8f0cf832..dabd2f9d 100644 --- a/server/reflector/views/transcripts.py +++ b/server/reflector/views/transcripts.py @@ -282,6 +282,7 @@ async def handle_rtc_event(event: PipelineEvent, args, data): elif event == PipelineEvent.STATUS: resp = transcript.add_event(event=event, data=data) + transcript.status = data.value else: logger.warning(f"Unknown event: {event}") diff --git a/server/tests/test_transcripts_rtc_ws.py b/server/tests/test_transcripts_rtc_ws.py index 113e07cf..70ee209b 100644 --- a/server/tests/test_transcripts_rtc_ws.py +++ b/server/tests/test_transcripts_rtc_ws.py @@ -183,3 +183,8 @@ async def test_transcript_rtc_and_websocket(dummy_transcript, dummy_llm): # stop server # server.stop() + + # check that transcript status in model is updated + resp = await ac.get(f"/transcripts/{tid}") + assert resp.status_code == 200 + assert resp.json()["status"] == "ended"