selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-20 20:29:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

server: ensure transcript status model is updated + tests

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

1
server/reflector/views/transcripts.py
View File

@@ -282,6 +282,7 @@ async def handle_rtc_event(event: PipelineEvent, args, data):
elif event == PipelineEvent.STATUS: elif event == PipelineEvent.STATUS:
resp = transcript.add_event(event=event, data=data) resp = transcript.add_event(event=event, data=data)
transcript.status = data.value
else: else:
logger.warning(f"Unknown event: {event}") logger.warning(f"Unknown event: {event}")

5
server/tests/test_transcripts_rtc_ws.py
View File

@@ -183,3 +183,8 @@ async def test_transcript_rtc_and_websocket(dummy_transcript, dummy_llm):
# stop server # stop server
# server.stop() # 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"