Skip to content

feat(optimization): run-scoped usage, budget, and cancellation contracts (RFC #6357, PR 1/2) #6358

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
caohy1988 wants to merge 7 commits into
base: main
Choose a base branch
from
Draft

feat(optimization): run-scoped usage, budget, and cancellation contracts (RFC #6357, PR 1/2) #6358

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
e726c58
feat(optimization): add run-scoped usage, budget, and cancellation co…
caohy1988 Jul 10, 2026
f345eee
align contracts with revised RFC: call/run status split, indeterminat…
caohy1988 Jul 10, 2026
bd5b94d
harden contracts per adversarial review: terminal state machine, froz…
caohy1988 Jul 10, 2026
9ed15ac
atomic settlement, complete terminal observation, truthful failure ty…
caohy1988 Jul 11, 2026
815eded
round-3 contracts: success invariant boundary, complete field normali…
caohy1988 Jul 11, 2026
b12baac
expose terminal provenance on the snapshot (terminal_from_provider_call)
caohy1988 Jul 11, 2026
d0b10d5
round-4 contracts: boundary-hostile inputs cannot strand or corrupt t…
caohy1988 Jul 11, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
48 changes: 48 additions & 0 deletions src/google/adk/optimization/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,3 +11,51 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Agent optimization: run-scoped governance surface (experimental).

The run-context contract is exported here as the supported public surface;
optimizer and sampler classes remain importable from their submodules.
"""

from ._run_context import ContextAlreadyAttachedError
from ._run_context import ModelCallEvent
from ._run_context import ModelCallState
from ._run_context import OptimizationBudgetExceeded
from ._run_context import OptimizationBudgets
from ._run_context import OptimizationCancelledError
from ._run_context import OptimizationFailedError
from ._run_context import OptimizationProviderError
from ._run_context import OptimizationRunContext
from ._run_context import OptimizationRunContextError
from ._run_context import OptimizationRunFinalizedError
from ._run_context import OptimizationRunSnapshot
from ._run_context import OptimizerCapabilities
from ._run_context import RunStatus
from ._run_context import STAGE_CANDIDATE_GENERATION
from ._run_context import STAGE_REFLECTION
from ._run_context import TokenBudgetStatus
from ._run_context import UnsupportedOptimizationContextError
from ._run_context import UsageCoverage

__all__ = [
"ContextAlreadyAttachedError",
"ModelCallEvent",
"ModelCallState",
"OptimizationBudgetExceeded",
"OptimizationBudgets",
"OptimizationCancelledError",
"OptimizationFailedError",
"OptimizationProviderError",
"OptimizationRunContext",
"OptimizationRunContextError",
"OptimizationRunFinalizedError",
"OptimizationRunSnapshot",
"OptimizerCapabilities",
"RunStatus",
"STAGE_CANDIDATE_GENERATION",
"STAGE_REFLECTION",
"TokenBudgetStatus",
"UnsupportedOptimizationContextError",
"UsageCoverage",
]
Loading