From 810a31b1f0d55e50e54b0ef1e202da47c7db8582 Mon Sep 17 00:00:00 2001 From: msramalho <19508417+msramalho@users.noreply.github.com> Date: Tue, 8 Aug 2023 18:06:48 +0100 Subject: [PATCH] fix: whisper handle error http code --- src/auto_archiver/enrichers/whisper_enricher.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/auto_archiver/enrichers/whisper_enricher.py b/src/auto_archiver/enrichers/whisper_enricher.py index 93d5e35..22dc6e7 100644 --- a/src/auto_archiver/enrichers/whisper_enricher.py +++ b/src/auto_archiver/enrichers/whisper_enricher.py @@ -57,7 +57,8 @@ class WhisperEnricher(Enricher): for i, m in enumerate(to_enrich.media): if m.is_video() or m.is_audio(): - job_id = to_enrich.media[i].get("whisper_model")["job_id"] + job_id = to_enrich.media[i].get("whisper_model", {}).get("job_id") + if not job_id: continue to_enrich.media[i].set("whisper_model", { "job_id": job_id, "job_status_check": f"{self.api_endpoint}/jobs/{job_id}",