From e5dd85409122c0ced1d3f27d6e4e2cbd3c8f33a7 Mon Sep 17 00:00:00 2001
From: Keesan <VAMAHALIN@GMAIL.COM>
Date: Fri, 22 May 2026 19:59:31 -0400
Subject: [PATCH] fix: ensure --budget alias properly overrides yaml config

---
 packages/cli/src/index.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/cli/src/index.ts b/packages/cli/src/index.ts
index 6f22a5d..cb1dcab 100644
--- a/packages/cli/src/index.ts
+++ b/packages/cli/src/index.ts
@@ -961,7 +961,7 @@ async function resolveGuardrails(
     maxTokens: config?.budget?.maxTokens ?? request.budget.maxTokens
   };
 
-  if (hasFlag(tokens, "--budget-usd")) {
+  if (hasFlag(tokens, "--budget-usd") || hasFlag(tokens, "--budget")) {
     budget.maxUsd = request.budget.maxUsd;
   }
   if (hasFlag(tokens, "--soft-limit-usd")) {