From 19b5ba2c4c40c08f8cff57e71a7f62829436504b Mon Sep 17 00:00:00 2001 From: Mathieu Virbel Date: Thu, 2 Nov 2023 16:53:57 +0100 Subject: [PATCH] server: add diarization logger information --- .../processors/audio_diarization_modal.py | 22 ++++++++++++------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/server/reflector/processors/audio_diarization_modal.py b/server/reflector/processors/audio_diarization_modal.py index 52be7c5d..3c8d1b45 100644 --- a/server/reflector/processors/audio_diarization_modal.py +++ b/server/reflector/processors/audio_diarization_modal.py @@ -22,15 +22,21 @@ class AudioDiarizationModalProcessor(AudioDiarizationProcessor): "audio_file_url": data.audio_url, "timestamp": 0, } + self.logger.info("Diarization started", audio_file_url=data.audio_url) async with httpx.AsyncClient() as client: - response = await client.post( - self.diarization_url, - headers=self.headers, - params=params, - timeout=None, - ) - response.raise_for_status() - return response.json()["text"] + try: + response = await client.post( + self.diarization_url, + headers=self.headers, + params=params, + timeout=None, + ) + response.raise_for_status() + self.logger.info("Diarization finished") + return response.json()["text"] + except Exception: + self.logger.exception("Diarization failed after retrying") + raise AudioDiarizationAutoProcessor.register("modal", AudioDiarizationModalProcessor)