diff --git a/agent.py b/agent.py index 0e2fb50..91f7e1f 100644 --- a/agent.py +++ b/agent.py @@ -878,7 +878,7 @@ def _propose(analysis: AnalyzeRevenueLeakageOutput, prioritized: PrioritizeAccou plans: list[RecoveryPlan] = [] for account in prioritized.prioritized_accounts[:max_plans]: findings = by_token.get(account.customer_token, []) - primary = max(findings, key=lambda f: f.recoverable_amount, default=None) + primary = max(findings, key=_recoverable_amount_key, default=None) action = RecoveryAction.suppress if account.policy_blockers else (primary.recommended_action if primary else RecoveryAction.data_quality_review) amount = account.recoverable_amount rationale = _rationale(action, account, primary)