server: ensure transcript status model is updated + tests

This commit is contained in:
Mathieu Virbel
2023-08-09 11:23:28 +02:00
parent a9e0c9aa03
commit 26e34aec2d
2 changed files with 6 additions and 0 deletions

View File

@@ -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}")

View File

@@ -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"