#!/usr/bin/env python3
import sys
SCENARIOS={
 'stale-index':('retrieval_freshness','rebuild_or_refresh_index'),
 'tool-error':('tool_execution_failure','inspect_tool_dependency'),
 'approval':('human_approval_wait','wait_or_expire_approval'),
 'uncertain':('uncertain_side_effect','verify_before_retry'),
 'healthy':('healthy','none'),
}
def run(name):
 d,a=SCENARIOS[name];print(f'scenario={name}');print(f'diagnosis={d}');print(f'action={a}');return d,a
if __name__=='__main__':
 name=sys.argv[1] if len(sys.argv)>1 else 'healthy'
 if name=='--self-test':
  assert run('stale-index')[0]=='retrieval_freshness';assert run('uncertain')[1]=='verify_before_retry';print('PASS rag_agent_stage_localization')
 elif name in SCENARIOS:run(name)
 else:raise SystemExit('unknown scenario')