pratik-250620's picture
Upload folder using huggingface_hub
6835659 verified
from __future__ import annotations
from typing import Any, Dict, Optional
from src.coherence.retry_strategies import retry_msci, retry_si_a, retry_st_a
def route_retry(
classification: Dict[str, Any],
context: Dict[str, Any],
) -> Optional[Dict[str, Any]]:
"""
Route retry strategy based on coherence classification and weakest metric.
"""
label = classification.get("label")
metric = classification.get("weakest_metric")
if label == "GLOBAL_FAILURE":
return retry_msci(context)
if label != "MODALITY_FAILURE":
return None
if metric == "si_a":
return retry_si_a(context)
if metric == "st_a":
return retry_st_a(context)
return None