IETM_FIXE\ietm6
6d00543a59
Fonctionnalité de comparaison PDF / campagne pour faire un mix et demander des conseils à l'IA
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
"""Retry avec backoff pour les appels LLM one-shot (imports).
|
|
|
|
Les imports enchaînent de nombreux appels en série ; un échec TRANSITOIRE sur un
|
|
seul morceau (503/502 surcharge serveur, 504/524 passerelle, timeout réseau) ne
|
|
doit pas faire échouer tout l'import. On réessaie quelques fois avec une attente
|
|
croissante. Après épuisement, on relaie l'erreur (problème durable : quota, panne).
|
|
|
|
Réservé aux appels `generate` (one-shot, bufferisé) : réessayer est propre, sans
|
|
risque de doublons. À NE PAS utiliser sur le streaming (re-jouerait des tokens).
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import logging
|
|
|
|
from app.domain.ports import LLMProvider, LLMProviderError
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
_ATTEMPTS = 3
|
|
_BASE_DELAY_SECONDS = 2.0
|
|
|
|
|
|
async def generate_with_retry(
|
|
llm: LLMProvider,
|
|
prompt: str,
|
|
*,
|
|
output_format: str | None = None,
|
|
temperature: float | None = None,
|
|
) -> str:
|
|
"""Comme `llm.generate`, mais réessaie les erreurs transitoires (backoff x2)."""
|
|
delay = _BASE_DELAY_SECONDS
|
|
last_error: LLMProviderError | None = None
|
|
for attempt in range(_ATTEMPTS):
|
|
try:
|
|
return await llm.generate(prompt, output_format=output_format, temperature=temperature)
|
|
except LLMProviderError as exc:
|
|
last_error = exc
|
|
if attempt < _ATTEMPTS - 1:
|
|
logger.warning(
|
|
"Appel LLM échoué (tentative %s/%s) : %s — nouvelle tentative dans %ss.",
|
|
attempt + 1, _ATTEMPTS, exc, delay,
|
|
)
|
|
await asyncio.sleep(delay)
|
|
delay *= 2
|
|
assert last_error is not None
|
|
raise last_error
|