diff --git a/.github/workflows/cost-map-guard.yml b/.github/workflows/cost-map-guard.yml index 208a75434a5..3c6fae74f6a 100644 --- a/.github/workflows/cost-map-guard.yml +++ b/.github/workflows/cost-map-guard.yml @@ -39,5 +39,6 @@ jobs: MERGE_BASE: ${{ steps.revisions.outputs.merge_base }} HEAD_SHA: ${{ github.event.pull_request.head.sha }} HEAD_REF: ${{ github.event.pull_request.head.ref }} + LITELLM_LOCAL_MODEL_COST_MAP: "True" run: | uv run --frozen python ci_cd/cost_map_guard.py --base "$MERGE_BASE" --head "$HEAD_SHA" --head-ref "$HEAD_REF"