feat(guardrails): implement Per-Session and Per-Org Cost Budgets #675

Merged

freemo merged 1 commit from feature/m6plus-per-session-per-org-cost-budgets into master 2026-03-10 18:24:32 +00:00

Conversation 0 Commits 1 Files changed 16 +2855 -10

freemo commented 2026-03-10 09:05:00 +00:00

Owner

Copy link

Summary

Implements Forgejo issue #584: Per-Session and Per-Org Cost Budgets.

Adds a three-tier budget hierarchy (per-plan → per-session → per-org) with the tightest limit winning.

Changes

Domain Models (cost_budget.py)

• BudgetLevel enum: PLAN, SESSION, ORG
• BudgetCheckResult: frozen Pydantic model with exceeded_level and warning fields
• SessionCostBudget: tracks per-session accumulated cost with utilization(), remaining(), is_exceeded(), would_exceed() methods
• OrgCostAccumulator: tracks per-org accumulated cost, requires org_id (min_length=1)
• ThreadSafeOrgCostAccumulator: thread-safe wrapper with snapshot() method

Application Services (cost_budget_service.py)

• CostBudgetService: manages budget state for sessions and orgs, enforces the three-tier hierarchy, emits BUDGET_WARNING (once per session) and BUDGET_EXCEEDED events via the event bus

AutonomyGuardrailService Integration

• associate_plan_with_session(): links a plan to a session for budget tracking
• check_budget_hierarchy(): checks plan → session → org hierarchy
• record_plan_cost_to_session(): records cost against session (and org if linked)

Configuration (settings.py)

• session_max_cost_usd: default session budget cap
• org_max_cost_usd: default org budget cap
• budget_warning_threshold: utilisation ratio for warning events (default 0.8)

Integration

• Session model: cost_budget field, as_cli_dict() includes budget data
• DI container: CostBudgetService registered as Singleton
• CLI session show: cost budget panel display

Tests

• 54 Behave scenarios (features/cost_budgets.feature)
• 11 Robot Framework integration tests (robot/cost_budgets.robot)
• ASV benchmarks (benchmarks/bench_budget_check.py)

Verification

• nox -s lint ✓
• nox -s typecheck ✓
• nox -s unit_tests ✓ (9809 scenarios pass)
• nox -s coverage_report ✓ (98% >= 97% threshold)
• nox -s integration_tests - all new cost budget tests pass; pre-existing failures unchanged

Closes #584

## Summary Implements Forgejo issue #584: Per-Session and Per-Org Cost Budgets. Adds a three-tier budget hierarchy (per-plan → per-session → per-org) with the tightest limit winning. ## Changes ### Domain Models (`cost_budget.py`) - `BudgetLevel` enum: PLAN, SESSION, ORG - `BudgetCheckResult`: frozen Pydantic model with `exceeded_level` and `warning` fields - `SessionCostBudget`: tracks per-session accumulated cost with `utilization()`, `remaining()`, `is_exceeded()`, `would_exceed()` methods - `OrgCostAccumulator`: tracks per-org accumulated cost, requires `org_id` (min_length=1) - `ThreadSafeOrgCostAccumulator`: thread-safe wrapper with `snapshot()` method ### Application Services (`cost_budget_service.py`) - `CostBudgetService`: manages budget state for sessions and orgs, enforces the three-tier hierarchy, emits `BUDGET_WARNING` (once per session) and `BUDGET_EXCEEDED` events via the event bus ### AutonomyGuardrailService Integration - `associate_plan_with_session()`: links a plan to a session for budget tracking - `check_budget_hierarchy()`: checks plan → session → org hierarchy - `record_plan_cost_to_session()`: records cost against session (and org if linked) ### Configuration (`settings.py`) - `session_max_cost_usd`: default session budget cap - `org_max_cost_usd`: default org budget cap - `budget_warning_threshold`: utilisation ratio for warning events (default 0.8) ### Integration - Session model: `cost_budget` field, `as_cli_dict()` includes budget data - DI container: `CostBudgetService` registered as Singleton - CLI `session show`: cost budget panel display ## Tests - 54 Behave scenarios (`features/cost_budgets.feature`) - 11 Robot Framework integration tests (`robot/cost_budgets.robot`) - ASV benchmarks (`benchmarks/bench_budget_check.py`) ## Verification - `nox -s lint` ✓ - `nox -s typecheck` ✓ - `nox -s unit_tests` ✓ (9809 scenarios pass) - `nox -s coverage_report` ✓ (98% >= 97% threshold) - `nox -s integration_tests` - all new cost budget tests pass; pre-existing failures unchanged Closes #584

freemo referenced this pull request 2026-03-10 09:05:39 +00:00

feat(guardrails): implement Per-Session and Per-Org Cost Budgets #584

freemo force-pushed feature/m6plus-per-session-per-org-cost-budgets from 882dde7930 to 4e435cbc80 2026-03-10 17:12:55 +00:00 Compare

freemo force-pushed feature/m6plus-per-session-per-org-cost-budgets from 4e435cbc80 to 5f938f50ea 2026-03-10 17:32:37 +00:00 Compare

freemo force-pushed feature/m6plus-per-session-per-org-cost-budgets from 5f938f50ea to 876217d0ca 2026-03-10 18:18:40 +00:00 Compare

freemo added this to the v3.6.0 milestone 2026-03-10 18:18:54 +00:00

freemo added the

Type

Feature

label 2026-03-10 18:19:12 +00:00

freemo self-assigned this 2026-03-10 18:19:20 +00:00

freemo scheduled this pull request to auto merge when all checks succeed 2026-03-10 18:19:37 +00:00

freemo merged commit 876217d0ca into master 2026-03-10 18:24:32 +00:00

freemo deleted branch feature/m6plus-per-session-per-org-cost-budgets 2026-03-10 18:24:32 +00:00

Sign in to join this conversation.

Reviewers

No reviewers

Labels

Clear labels   

auto/needs-reevaluation

Controller deferred this PR; awaiting Phase 6+ scope-evaluator or operator re-enablement.

  

controller-managed

Auto-agents controller manages this PR/issue (see tools/controller/deploy/RUNBOOK.md). Remove this label to abandon controller management.

  

auto/blocked-by-deps

PR blocked by an open issue dependency. Operator must close the dep (or remove the dependency link) before the merge driver can act. Auto-cleared by merge_drive when no open deps remain.

  

auto/ci-timeout

Most recent merge cycle hit CI timeout. Driver excludes this PR while last merge_cycle row is < 30 min old; label persists thereafter as visible history.

  

auto/claimed-implementer

Currently being processed by an implementer worker.

  

auto/claimed-merge

Currently being processed by the merge driver.

  

auto/claimed-reviewer

Currently being processed by a reviewer worker.

  

auto/driver-down

Merge driver heartbeat stale; pipeline halted. Closed automatically on next clean tick.

  

auto/invariant-violation

Detected master commit violating the strict merge invariant. Tracked as an issue (not a PR label); kept here for label completeness.

  

auto/last-attempt-tier-0

In-cycle escalation: most recent attempt ran at the Tier 0 slot (`tier-0`). Slot's model defined in .opencode/models/tiers.yaml.

  

auto/last-attempt-tier-1

In-cycle escalation: most recent attempt ran at the Tier 1 slot (`tier-1`). Slot's model defined in .opencode/models/tiers.yaml.

  

auto/last-attempt-tier-2

In-cycle escalation: most recent attempt ran at the Tier 2 slot (`tier-2`). Slot's model defined in .opencode/models/tiers.yaml. Gated behind IMPLEMENTER_ESCALATION_TIER2_ENABLED.

  

auto/last-attempt-tier-min

In-cycle escalation: most recent attempt ran at the Tier -1 slot (`tier-min`). Slot's model defined in .opencode/models/tiers.yaml. Suffix is ``-min`` (not ``--1``) so the Forgejo UI reads naturally.

  

Automation Tracking

Tracking issues used by the AI Automation system for agents to communicate and report.

  

auto/needs-conflict-resolution

Rebase conflict needs LLM conflict-resolver.

  

auto/needs-implementer

Failing CI needs implementer attention.

  

auto/postmortem

Documenting a driver incident or rollback.

  

auto/ready-to-merge

Reviewer has APPROVED this PR and no later REQUEST_CHANGES is outstanding. The merge driver requires this label to even consider a PR for merging. Set by the reviewer worker on APPROVE; cleared on REQUEST_CHANGES.

  

auto/restart-throttled

Train repeatedly lost master-tempo races. Driver excludes via merge_cycle until cooldown elapses; label persists as visible history.

  

auto/revert

Revert PR backing out an invariant violation. Fast-tracked through the merge driver.

  

auto/sentinel

Sentinel PR duplicated from upstream into a personal fork by tools/duplicate_prs_to_fork.py for pipeline testing. Lives only in the fork; the canonical pipeline never sees it.

  

auto/stale-inactivity

No implementer activity for N days. Flagged for human review. Auto-cleared on next push to head branch.

  

auto/unstable

Repeatedly fails on current master (>= 3 ci-fail-on-rebased-sha releases in 12 h). Excluded from driver until human triage.

  

Blocked

A ticket in a blocked state and unable to complete until some other task is completed first.

  

Bounty

$100

A bounty of $100 for any open-source contributor who provides a MR that solves this issue

  

Bounty

$1000

A bounty of $1000 for any open-source contributor who provides a MR that solves this issue

  

Bounty

$10000

A bounty of $10000 for any open-source contributor who provides a MR that solves this issue

  

Bounty

$20

A bounty of $20 for any open-source contributor who provides a MR that solves this issue

  

Bounty

$2000

A bounty of $2000 for any open-source contributor who provides a MR that solves this issue

  

Bounty

$250

A bounty of $250 for any open-source contributor who provides a MR that solves this issue

  

Bounty

$50

A bounty of $50 for any open-source contributor who provides a MR that solves this issue

  

Bounty

$500

A bounty of $500 for any open-source contributor who provides a MR that solves this issue

  

Bounty

$5000

A bounty of $5000 for any open-source contributor who provides a MR that solves this issue

  

Bounty

$750

A bounty of $750 for any open-source contributor who provides a MR that solves this issue

  

MoSCoW

Could have

Could have feature in order to satisfy the epic/legendary.

  

MoSCoW

Must have

Must have feature in order to satisfy the epic/legendary.

  

MoSCoW

Should have

Should have feature in order to satisfy the epic/legendary.

  

Needs Feedback

There are questions in the ticket that can not be completed until the project owner provides clarity.

  

Points

1

1 man-hours worth of work for an expert with no learning curve.

  

Points

13

13 man-hours worth of work for an expert with no learning curve.

  

Points

2

2 man-hours worth of work for an expert with no learning curve.

  

Points

21

21 man-hours worth of work for an expert with no learning curve.

  

Points

3

3 man-hours worth of work for an expert with no learning curve.

  

Points

34

34 man-hours worth of work for an expert with no learning curve.

  

Points

5

5 man-hours worth of work for an expert with no learning curve.

  

Points

55

55 man-hours worth of work for an expert with no learning curve.

  

Points

8

8 man-hours worth of work for an expert with no learning curve.

  

Points

88

88 man-hours worth of work for an expert with no learning curve.

  

Priority

Backlog

This ticket has backlogged priority and is not to be worked on yet

  

Priority

CI Blocker

Critical priority issue that blocks CI/CD pipeline and prevents PR merges

  

Priority

Critical

The priority is critical

  

Priority

High

The priority is high

  

Priority

Low

The priority is low

  

Priority

Medium

The priority is medium

  

Signed-off: Owner

When an epic or legendary is in review it must be signed off by owner, tech lead, and scrum master before being marked as completed.

  

Signed-off: Scrum Master

When an epic or legendary is in review it must be signed off by owner, tech lead, and scrum master before being marked as completed.

  

Signed-off: Tech Lead

When an epic or legendary is in review it must be signed off by owner, tech lead, and scrum master before being marked as completed.

  

Spike

A ticket for learning a tool or technology that is needed to be able to do future planning and design.

  

State

Completed

The ticket has been fully implemented, completed, and merged with the source code. This label should only be applied once a ticket is closed.

  

State

Duplicate

A ticket that represents the same content as an existing ticket.

  

State

In Progress

A ticket that is actively being developed.

  

State

In Review

A ticket that has had some code completed to implement but is waiting to pass peer review and is not yet merged in.

  

State

Paused

This ticket's work started but wasn't finished. It's on hold (likely in a feature branch) and will be resumed later, either due to a blocker or a delay.

  

State

Unverified

All new tickets start in this state. A developer may set it to show the ticket is unverified. This means we haven't agreed to work on it. It will either move to a verified state or be closed as wontdo.

  

State

Verified

The issue has been verified by a developer as legitimate. It will be worked on and verified tickets are now considered part of the backlog.

  

State

Wont Do

This ticket has been decided it wont be done. This may mean the bug has been determined to not be real (cant verify) or the feature is one we have decided we dont want to adopt.

  

Type

Automation

Any edits or discussion about the AI automated coding system.

  

Type

Bug

Something that doesnt work as intended.

  

Type

Discussion

Anytime a ticket represents a discussion about a subject and doesnt fall into one of the other categories.

  

Type

Documentation

An error or improvement needed in the documentation.

  

Type

Epic

Any first tier epic. That is, an epic which contains only issues as children and will not have sub-epics.

  

Type

Feature

Some new functionality not present.

  

Type

Legendary

A type of Epic which will contain other Epics.

  

Type

Refactor

A code change that restructures existing code without changing its external behavior.

  

Type

Support

Someone needs help using the project.

  

Type

Task

A generic task that doesnt fit into the other type categories.

  

Type

Testing

Work exclusively focusing on fixing or expanding testing.

No labels

auto/needs-reevaluation

controller-managed

auto/blocked-by-deps

auto/ci-timeout

auto/claimed-implementer

auto/claimed-merge

auto/claimed-reviewer

auto/driver-down

auto/invariant-violation

auto/last-attempt-tier-0

auto/last-attempt-tier-1

auto/last-attempt-tier-2

auto/last-attempt-tier-min

Automation Tracking

auto/needs-conflict-resolution

auto/needs-implementer

auto/postmortem

auto/ready-to-merge

auto/restart-throttled

auto/revert

auto/sentinel

auto/stale-inactivity

auto/unstable

Blocked

Bounty

$100

Bounty

$1000

Bounty

$10000

Bounty

$20

Bounty

$2000

Bounty

$250

Bounty

$50

Bounty

$500

Bounty

$5000

Bounty

$750

MoSCoW

Could have

MoSCoW

Must have

MoSCoW

Should have

Needs Feedback

Points

1

Points

13

Points

2

Points

21

Points

3

Points

34

Points

5

Points

55

Points

8

Points

88

Priority

Backlog

Priority

CI Blocker

Priority

Critical

Priority

High

Priority

Low

Priority

Medium

Signed-off: Owner

Signed-off: Scrum Master

Signed-off: Tech Lead

Spike

State

Completed

State

Duplicate

State

In Progress

State

In Review

State

Paused

State

Unverified

State

Verified

State

Wont Do

Type

Automation

Type

Bug

Type

Discussion

Type

Documentation

Type

Epic

Type

Feature

Type

Legendary

Type

Refactor

Type

Support

Type

Task

Type

Testing

Milestone

Clear milestone

No items

No milestone

v3.6.0

Projects

Clear projects

No items

No project

Assignees

Clear assignees

No assignees

freemo

1 participant

Notifications

Subscribe

Due date

The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference

cleveragents/cleveragents-core!675

No description provided.