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: add diarization logger information

Browse Source
This commit is contained in:
Mathieu Virbel
2023-11-02 16:53:57 +01:00
committed by Mathieu Virbel
parent bba5643237
commit 19b5ba2c4c
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
server/reflector/processors/audio_diarization_modal.py
View File

@@ -22,7 +22,9 @@ class AudioDiarizationModalProcessor(AudioDiarizationProcessor):
"audio_file_url": data.audio_url, "audio_file_url": data.audio_url,
"timestamp": 0, "timestamp": 0,
} }
self.logger.info("Diarization started", audio_file_url=data.audio_url)
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
try:
response = await client.post( response = await client.post(
self.diarization_url, self.diarization_url,
headers=self.headers, headers=self.headers,
@@ -30,7 +32,11 @@ class AudioDiarizationModalProcessor(AudioDiarizationProcessor):
timeout=None, timeout=None,
) )
response.raise_for_status() response.raise_for_status()
self.logger.info("Diarization finished")
return response.json()["text"] return response.json()["text"]
except Exception:
self.logger.exception("Diarization failed after retrying")
raise
AudioDiarizationAutoProcessor.register("modal", AudioDiarizationModalProcessor) AudioDiarizationAutoProcessor.register("modal", AudioDiarizationModalProcessor)