from agent_reliability_toolkit.tracing import trace_agent_run def run_and_trace(agent, prompt, context_len): trace = trace_agent_run(agent, prompt, context_len=context_len) if trace['failures']: print(f"FAIL at context {context_len}: {trace['failures']}") return trace