From 9336ed1985c529166f6752c6a25b80fa2f8cb083 Mon Sep 17 00:00:00 2001
From: Yaniv Israel <yaniv@deepkeep.ai>
Date: Mon, 15 Jun 2026 13:31:00 +0300
Subject: [PATCH] fix(makefile): fall back to upstream/litellm_internal_staging
 for strict-budget gate

origin/litellm_internal_staging exists on BerriAI's CI but not on forks
that use a different remote name (e.g. Azure DevOps as origin).  Fall
back to upstream/litellm_internal_staging when the origin ref is absent.
---
 Makefile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Makefile b/Makefile
index 41e3ab5f5d5..e89932ac896 100644
--- a/Makefile
+++ b/Makefile
@@ -126,7 +126,8 @@ lint-mypy: install-dev
 lint-black: format-check
 
 lint-strict-budget: install-dev
-	$(UV_RUN) python scripts/ruff_strict_gate.py
+	$(UV_RUN) python scripts/ruff_strict_gate.py \
+		--base $$(git rev-parse --verify origin/litellm_internal_staging 2>/dev/null && echo origin/litellm_internal_staging || echo upstream/litellm_internal_staging)
 
 lint-strict-budget-update: install-dev
 	$(UV_RUN) python scripts/ruff_strict_gate.py --update