from agent_reliability_toolkit.recovery import ( quarantine_agent, adaptive_retry, log_structured ) def safe_agent_handler(agent, *args, **kwargs): run_id = kwargs.get("run_id", "unknown") try: result = adaptive_retry(agent, *args, **kwargs) return result except Exception as exc: quarantine_agent(agent, run_id=run_id, reason=str(exc)) log_structured(run_id, agent, error=str(exc)) raise # Usage: # result = safe_agent_handler(my_agent, input, run_id="sess-1234")