uprava na errors a try catch

This commit is contained in:
JiriUhlir
2026-07-13 08:54:45 +02:00
parent d55089b1c8
commit baba9ab54a
+41 -4
View File
@@ -6,7 +6,10 @@ co nejlepšího přepisu.
Audio přichází jako proměnná (upload `file` ve form-data). Všechny ostatní parametry
jsou v POST těle (form fields). API klíče jsou v X- hlavičkách (viz credentials.py).
Výstup je vždy JSON: {"gpt": ..., "deepgram": ..., "merge": ...}.
Výstup je vždy JSON: {"gpt": ..., "deepgram": ..., "merge": ..., "errors": [...]}.
Selhání jednoho z přepisů (Deepgram / OpenAI) nebo sloučení neshodí celý request —
jeho část zůstane prázdná a chyba se přidá do pole `errors`.
"""
import asyncio
@@ -41,6 +44,11 @@ class TranscribeResult(BaseModel):
...,
description="Sloučený, co nejpřesnější a úplný přepis se štítky mluvčích.",
)
errors: list[str] = Field(
default_factory=list,
description="Chyby dílčích kroků (Deepgram / OpenAI přepis / sloučení). "
"Prázdné, pokud vše proběhlo v pořádku. Selhavší část je pak prázdná.",
)
async def _read_upload(file: UploadFile) -> tuple[bytes, str, str]:
@@ -85,8 +93,10 @@ async def transcribe(
audio, filename, content_type = await _read_upload(file)
# Oba přepisy paralelně.
text_deepgram, text_gpt = await asyncio.gather(
errors: list[str] = []
# Oba přepisy paralelně; selhání jednoho neshodí druhý ani celý request.
result_deepgram, result_gpt = await asyncio.gather(
deepgram_client.transcribe(
api_key=deepgram_key,
audio=audio,
@@ -104,9 +114,28 @@ async def transcribe(
model=whisper_model,
language=language,
),
return_exceptions=True,
)
if isinstance(result_deepgram, BaseException):
log.exception("Deepgram přepis selhal", exc_info=result_deepgram)
errors.append(f"Deepgram přepis selhal: {result_deepgram}")
text_deepgram = ""
else:
text_deepgram = result_deepgram
if isinstance(result_gpt, BaseException):
log.exception("OpenAI (whisper) přepis selhal", exc_info=result_gpt)
errors.append(f"OpenAI (whisper) přepis selhal: {result_gpt}")
text_gpt = ""
else:
text_gpt = result_gpt
# Sloučení má smysl jen když je aspoň jeden přepis k dispozici.
merged = ""
if text_deepgram or text_gpt:
prompt = combine_prompt.strip() if combine_prompt and combine_prompt.strip() else DEFAULT_COMBINE_PROMPT
try:
merged = await openai_client.merge_transcripts(
api_key=openai_key,
model=chat_model,
@@ -114,5 +143,13 @@ async def transcribe(
text_deepgram=text_deepgram,
text_whisper=text_gpt,
)
except Exception as exc:
log.exception("Sloučení přepisů selhalo")
errors.append(f"Sloučení přepisů selhalo: {exc}")
return TranscribeResult(gpt=text_gpt, deepgram=text_deepgram, merge=merged)
return TranscribeResult(
gpt=text_gpt,
deepgram=text_deepgram,
merge=merged,
errors=errors,
)