Generated banner for Apigee Agent Gateway Governance
Generated logo for Apigee Agent Gateway Governance

Apigee Agent Gateway Governance

GoogleOfficialPublished by Google

  • Skill
  • Workflow
  • Automation and integrations
  • Apache-2.0
  • Low risk
  • Has scripts

From the author

Stands up an Apigee API proxy that a Google Cloud Agent Gateway calls as an ext-proc (Service Extension) callout on an agent's LLM traffic, to meter token usage per model, extract caller identity, and apply Model Armor prompt screening or a semantic cache. Use when wiring Apigee in as an AI gateway behind an Agent Gateway, when an agent's model egress needs governing, or when a callout proxy returns no verdict and you need to find out why. The distinguishing question is whether Apigee forwards the traffic or only inspects it: this skill is inspect-only, and builds the proxy itself. Don't use it when Apigee is meant to receive a request and route it onward to a backend — including hosting an MCP server behind Apigee — which needs a proxy with a target instead. Don't use for governing the agent's calls to tools or MCP servers. Don't use for choosing or switching which model a request goes to. Don't use for creating the Agent Gateway itself, nor for Apigee hybrid.
— the description by Google

AutoGineer summary

Written by AutoGineer from the skill’s own files, not by the author

This skill automates the configuration of an Apigee API proxy to act as a service extension for a Google Cloud Agent Gateway. It enables inspection of LLM traffic to perform token metering, caller identity extraction, prompt screening via Model Armor, and semantic caching. The skill follows a structured deployment process using AFT templates and gcloud commands, ensuring the proxy is correctly wired to intercept and evaluate agent requests without acting as a backend router.

What it does

  • Stands up an inspect-only Apigee proxy for LLM traffic governance.
  • Configures token metering per model and enforces usage quotas.
  • Integrates Model Armor for prompt screening and semantic caching.
  • Manages gateway network connectivity and AuthzPolicy configuration.
  • Provides automated deployment scripts for consistent environment setup.
  • Supports both development and production modes with defined contract schemas.

When to use it

  • When you need to meter or govern LLM token usage for agents.
  • When you require prompt screening or semantic caching for agent egress.
  • When you need to extract caller identity from agent traffic.
  • When you are wiring Apigee as an AI gateway behind a Google Cloud Agent Gateway.

Advantages

  • Centralized governance for agent model egress.
  • Version-controlled proxy configuration via AFT templates.
  • Automated handling of complex network and service extension dependencies.
  • Non-intrusive inspection architecture that does not require backend routing.

Usage examples

“Set up Apigee governance for my existing Agent Gateway in dev mode.”
The agent will assess the current environment, apply default configurations for metering and identity, and generate a deployment plan for approval.
“Deploy the Apigee proxy using my production configuration file.”
The agent will validate the provided contract file, execute the deployment sequence, and verify the proxy is reporting as EXTENSIBLE.

Install

  1. Download the skill as a ZIP:

    https://mp-staging.autogineer.com/agent-skills/google/apigee-agent-gateway-governance/download
  2. Unzip it into the cross-client skills folder of your project:

    mkdir -p .agents/skills && unzip apigee-agent-gateway-governance.zip -d .agents/skills

Documentation by Google

Apigee as an Agent Gateway Service Extension

Wires an Apigee proxy in as an ext-proc callout on a GCP Agent Gateway. The proxy is a processor, not an API proxy: it has no backend, inspects the LLM request and response the gateway hands it, and returns a verdict.

Usage Attribution

MUST READ: Before executing any gcloud, REST API, or MCP command, you MUST read and follow the instructions in Usage Attribution.

You build the proxy; the user does not.

Two modes, one schema. Decide which before anything else, and say which. Input file present → prod mode: it is the contract, every value comes from it, and a missing or malformed field is a refusal, not a default. No input file and the user is describing an outcome → dev mode: apply the documented defaults, show a plan with durations and get approval, start with identity and metering only, write apigee-agent-gateway-governance.yaml at the end as the handoff. assets/apigee-agent-gateway-governance.example.yaml documents the schema both modes use.

Modes, defaults, durations, cost disclosure and the interlocks that bind both: references/modes.md. Read it before you start.

Colour the plan's ACTION column — green REUSE, cyan CREATE/RESERVE, yellow APPEND/MODIFY; full palette in references/modes.md. Yellow is the one that earns its place: it is the only signal that a resource the user already owns is being changed. Rendered plain, the one row that modifies something is indistinguishable from the dozen that do not — which is exactly the row the approval exists for.

Prerequisite: where the agent runs

Check this before building anything. It decides whether traffic can reach the gateway at all, and it is silent when wrong — every step below still succeeds while nothing arrives.

Use Agent Runtime, created with agent_gateway_config / agent_to_anywhere_config. It is the supported path and the orchestrator builds the client-side PSC, routing, DNS and cert trust for you. There is no gcloud surface for this and no console path — it is SDK-only, at agent creation time. See Route traffic through Agent Gateway for the binding, and references/networking-psc.md for a worked example.

Set GOOGLE_API_PREVENT_AGENT_TOKEN_SHARING_FOR_GCP_SERVICES: "False" in the agent's env_vars, or every invocation 401s on the agent's own sessions.create, before user code runs. Deploy-time invisible. Full symptom and isolation: references/gotchas.md, Agent Gateway.

Where the agent runs is otherwise settled — the premise is a working gateway and a working agent. Do not explain how to build one or list unsupported compute unprompted; references/networking-psc.md has the constraints if it turns out to matter.

Tool versions

Do not assume these are installed or current.

ToolMinimumWhy
gcloud588.0.0apigee apis import --from-template is on the GA track at this version; older SDKs need gcloud beta
google-cloud-aiplatform[agent_engines,adk]>=2.1,<3Agent Runtime client used to bind the agent to the gateway
google-adk>=2.9,<3only if you build the test agent with ADK

On gcloud >= 588.0.0: every feature must keep its uid, and the template must never declare endpoints or targets. All six features set uid today and the template declares neither, so the shipped bundle is correct — but both are load-bearing now, and breaking either one still compiles, still deploys, and reports nothing. Mechanism and symptoms: references/gotchas.md, under Authoring (AFT).

gcloud version | head -1          # confirm >= 588.0.0
gcloud components update          # if older
pip install "google-cloud-aiplatform[agent_engines,adk]>=2.1,<3" "google-adk>=2.9,<3"

scripts/compile_check.py needs PyYAML and names the interpreter if it is missing. pip install pyyaml, or run it with a Python 3 that has it; gcloud's bundled interpreter does not, and the SDK's vendored yaml is Python 2.

python3 scripts/compile_check.py assets/apigee-proxy-template.yaml

Execution model

Three layers, and the choice between them is not stylistic:

LayerUsed forWhy
AFT YAML templatethe proxy itselfversion-controlled, reviewable, diffable
gcloudPSC, DNS, AuthzExtension, AuthzPolicynative commands exist
REST (curl)env properties, envgroups, deployments, debug sessionsno gcloud equivalent today

The customer owns one input file (assets/apigee-agent-gateway-governance.example.yaml). It is the version-controlled artifact. Never hand-edit the generated proxy: regenerate it.

Run the Apigee-side sequence with the script, not by hand. Steps 1, 3 and 4 below are order-dependent and each fails in a way that reads as a different problem. scripts/deploy_proxy.sh does all three, preserves the properties map, polls the async operation, and exits non-zero if the deployment does not come up EXTENSIBLE:

scripts/deploy_proxy.sh --org "$ORG" --env "$ENV" --proxy "$PROXY" \
  --template assets/apigee-proxy-template.yaml [--service-account "$SA"] --yes \
  [--index-endpoint-id ID --index-id ID --sc-project P --sc-region R --embeddings-model M]

Run it without --yes first: it prints the plan and changes nothing. Show that plan, then re-run with --yes once the user agrees.

The vector-search flags are the one exception to "edit the default in every feature that declares it". Those three ids are substituted at deploy time and checked for leftovers; hand-editing them into feature-semantic-cache.yaml deploys fine, skips the check, and makes the file single-use.

scripts/ensure_gateway_connectivity.sh is the same idea for the gateway leg (step 0). It is the only part of this skill that modifies a resource the user already owns, so it applies nothing without --yes.

The steps are written out below so you can diagnose a failure, not so you can retype them.

Inputs you must resolve first

Environment prerequisites: references/setup.md — gcloud SDK (588+), project, billing, gcloud services enable, region, credentials, IAM. Nothing below creates any of it.

Prod: from the input file, then environment, then ask — never invent. Dev: apply the references/modes.md defaults, and ask only for what has none.

  • project, region

  • apigee.org, apigee.environment, apigee.envgroup, apigee.hostname, apigee.proxy

  • network.vpc, network.subnet, network.networkAttachment

  • gateway.name, gateway.matchPaths (a list, one entry per wire shape), gateway.manageNetworkConfig

matchPaths drives TWO artifacts; write both from the same list. The AuthzPolicy paths[].suffix gets one suffix per entry, verbatim with leading punctuation kept — it decides what the gateway routes to Apigee. The template's LLM_URI_MATCH gets one (request.uri Like "*<entry>*") clause per entry, OR-ed, leading : or / stripped — it decides what the proxy acts on once it arrives. The shipped two-entry default gives:

((request.uri Like "*generateContent*") or (request.uri Like "*chat/completions*"))

The AuthzPolicy is the outer gate and wins: a suffix list narrower than matchPaths means the missing shape never reaches Apigee at all. Derive both from the input file and never from the default when the file disagrees — the file is the reviewable contract and a mismatch is invisible once deployed.

A one-entry list is legitimate and governs one shape. Dropping an entry governs one fewer silently — deploys clean, zero transactions for the shape dropped — and narrows feature-normalize, Model Armor and the cache together. references/gotchas.md, Runtime.

semanticCache is a KNOWN ISSUE whenever it is on — warn, then build it. The key is the prompt embedding alone: no caller, no model, no shape. A hit therefore crosses callers — similar prompts share an answer — and one model behind one shape does not make it safe. With more shapes the first model to answer answers for the other too: empty HTTP 200 across shapes, wrong-model body within one. Warn whenever the cache is on, not only on multi-shape, then build. Wording: references/modes.md; mechanism: references/gotchas.md.

Each entry is a SUFFIX match. You write these, not the user — derive them from which endpoint their agent actually calls, and do not ask them to reason about suffixes. Vertex's OpenAI-compatible endpoint and api.openai.com need different entries, and getting it wrong governs nothing silently; the forms and the evidence are in references/gotchas.md, Runtime.

There is no apiShape field. Metering does not depend on the wire shape — the count is normalised in JS across both — so the field selected nothing and was removed. Adding a shape is an || branch in extract-model.js plus an entry in matchPaths.

Three places carry the callout hostname and all three must match — the Apigee envgroup, the AuthzExtension authority/service, and the private DNS A record. You set all three from one value: a build invariant to hold, not a question for the user. It becomes a topic only when a callout never lands, and then it is the first check. Build order matters alongside it — the AuthzExtension is inert until the host resolves and the proxy is deployed, so "the extension exists" is not evidence that leg works. references/gotchas.md, Runtime.

Workflow

0. Gateway: give it a route into the VPC

A gateway with no networkConfig has no route into the customer's VPC and cannot resolve the callout hostname. This skill builds it — network attachment, egress and DNS peering — so its absence is a build step, not a blocker, and should be reported that way. Do not open with an alarm about a resource you are about to create.

One thing here does need the user, and it is the reason this is step 0 rather than a silent prerequisite: attaching egress is one-way and affects traffic they already run. Settle it in the plan, before anything executes — see "Confirm everything up front" below. Capturing a known-good agent turn first is worth doing when they are nervous about a production gateway, because it separates "never worked" from "we broke it"; it is a precaution to offer, not a step to impose.

gcloud network-services agent-gateways describe "$GATEWAY" \
  --location="$REGION" --project="$PROJECT" --format='value(networkConfig)'
gcloud network-security authz-policies list \
  --location="$REGION" --project="$PROJECT"

A console-created gateway already has a CUSTOM AuthzPolicy — an IAP REQUEST_AUTHZ one, which coexists with ours and is not needed for the egress JWT. You never create the gateway, so assess it rather than assuming. references/gotchas.md, Agent Gateway.

Empty networkConfig output means missing, and it is easy to miss: the gateway still serves ordinary agent traffic perfectly, because that path goes to public Google endpoints and never enters the VPC. Only the Apigee callout fails.

If it is missing, behaviour depends on gateway.manageNetworkConfig:

  • true — run the script below. It creates the network attachment, then read-modify-writes the gateway to add the egress attachment and DNS peering. Agreement for this is taken in the plan, not here; --yes carries it.

  • false — the user has said they want to make this change themselves. Print the script invocation and let them run it. Do not modify the gateway.

scripts/ensure_gateway_connectivity.sh \
  --gateway "$GATEWAY" --project "$PROJECT" --region "$REGION" \
  --vpc "$VPC" --subnet "$SUBNET" --attachment "$ATTACHMENT" \
  --domain "$DNS_DOMAIN"            # add --yes only after the user agrees

Either way networkConfig has to exist before the Apigee work starts — not because its absence is an error, but because every later step then succeeds and governs nothing.

Use the script, not the raw commands: import is a whole-resource upsert and a hand-built YAML silently drops the gateway's registries and protocols. Why, and the etag and PROTOCOL_UNSPECIFIED handling: references/gotchas.md.

1. Apigee: enable the environment

The env must carry apigee-service-extension-enabled=true. No gcloud flag exists; PATCH the properties map.

curl -s -H "Authorization: Bearer $TOKEN" \
  "https://apigee.googleapis.com/v1/organizations/$ORG/environments/$ENV"

The PATCH replaces the whole map. Read it first and re-send every property you intend to keep, or you will silently drop them.

curl -X PATCH -H "Authorization: Bearer $TOKEN" -H "Content-Type: application/json" \
  "https://apigee.googleapis.com/v1/organizations/$ORG/environments/$ENV?updateMask=properties" \
  -d '{"properties":{"property":[{"name":"apigee-service-extension-enabled","value":"true"}]}}'

This does not disable Apigee Analytics — the two coexist; if analytics looks missing it is the separate per-environment add-on. references/gotchas.md.

Use a dedicated environment, attached to exactly one envgroup. An environment in two makes the gateway 404 every request with the extension never invoked. It also leaves the / base path free, which is what the proxy ships with.

scripts/ensure_apigee_environment.sh creates the environment, attaches it to the regional runtime instance, and creates and attaches the envgroup — so step 2 is already done when it succeeds. It creates billable resources and applies nothing without --yes; that agreement is taken in the plan, not here.

scripts/ensure_apigee_environment.sh \
  --org "$ORG" --env "$ENV" --region "$REGION" \
  --envgroup "$ENVGROUP" --hostname "$HOSTNAME"      # add --yes to apply

It is idempotent and resumable, and refuses with exit 2 when the environment is already in another envgroup or the envgroup exists on a different hostname. It also handles three things a runbook tends to get wrong — the omitted type on create, the four-minute instance attachment that locks the environment, and polling the operation rather than a GET that succeeds too early.

2. Apigee: envgroup on the callout hostname

Done already if step 1's script ran. Verify, or do it by hand for an envgroup that already exists. The hostname must be exactly the callout host, and the group must be attached to the env.

curl -s -H "Authorization: Bearer $TOKEN" \
  "https://apigee.googleapis.com/v1/organizations/$ORG/envgroups/$ENVGROUP/attachments"

How many envgroups is the environment in? More than one and the callout never lands. deploy_proxy.sh enforces this before touching anything and exits 2 naming them; assess_integration.sh reports it. Read the failure literally if it fires — the environment is wrong, not the script.

3. Build the proxy from the template
gcloud apigee apis import organizations/$ORG/apis/$PROXY \
  --from-template=assets/apigee-proxy-template.yaml

Each features: entry resolves relative to the template file's own directory, never the working directory, so a bare filename must sit beside the template (a subpath like features/feature-identity.yaml works too). assets/ is flat by convention, not compiler demand — but moving a file without updating features: fails the import with a bare FileNotFoundError.

If you get Permission denied on resource ... (or it may not exist) on an org you demonstrably have access to, it is the credential, not the access. With context-aware access enabled the default gcloud credential is rejected. Pass the ADC token instead:

CLOUDSDK_AUTH_ACCESS_TOKEN=$(gcloud auth application-default print-access-token) \
  gcloud apigee apis import ...

To change a parameter, edit the default in every FEATURE that declares it — not in the template. gcloud apigee apis import passes an empty caller-parameter map, so each feature resolves its own default and the template's declaration of the same name is never read. Editing the template changes nothing, silently, and the compile still passes. Measured: setting MA_CONTINUE_ON_ERROR to DRY_RUN in the template shipped ENFORCE on a fail-closed policy. LLM_URI_MATCH is declared in four features and all four must agree. scripts/compile_check.py fails on any disagreement. There is no --parameters flag.

Enable an optional recipe by adding its feature to the template's features list. See references/recipes.md.

tokenMetering.models in the input file is NOT generative. It documents which buckets exist; it does not create them. A bucket is real only when it has an LLMTokenQuota step in feature-token-metering.yaml and a matching *_TOKEN_LIMIT parameter. Three exist: gemini-pro, gemini-flash and other — the catch-all for any model whose name contains neither pro nor flash. Adding a fourth entry to the list meters nothing, and omitting other silently unmeters every non-Gemini model: the count is extracted and then discarded, with no ratelimit reference anywhere in the transaction. Adding a model is a feature-file edit, not a config change.

4. Deploy

Deploy to the service-extension env, then confirm the deployment reports proxyDeploymentType: EXTENSIBLE.

curl -s -H "Authorization: Bearer $TOKEN" \
  ".../environments/$ENV/deployments"

STANDARD means it will not govern the agent's off-Apigee tool egress.

If any policy calls a Google API — Model Armor, embeddings, Vector Search — deploy with a service account or the deploy is rejected with MISSING_SERVICE_ACCOUNT:

curl -X POST -H "Authorization: Bearer $TOKEN" \
  ".../apis/$PROXY/revisions/$REV/deployments?override=true&serviceAccount=$DEPLOY_SA"
5. Networking

This is the gateway → Apigee leg only. PSC endpoint to the Apigee instance service attachment, plus a private DNS A record. See references/networking-psc.md. The Apigee instance consumerAcceptList must include your project or the connection never reaches ACCEPTED.

The agent → gateway leg is the prerequisite above, and is not built here.

6. Bind the gateway

AuthzExtension then AuthzPolicy — in that order, and only once the host resolves and the proxy is deployed. The extension is inert until then.

Both document bodies ship as assets. Copy and substitute; do not invent them — assets/authz-extension.example.yaml and assets/authz-policy.example.yaml. They carry the fields that are load-bearing and have no obvious default: wireFormat: EXT_PROC_GRPC, failOpen, the authority/service pair, the metadata body flags, and the policy's customProvider / httpRules nesting.

gcloud service-extensions authz-extensions import $AUTHZ_EXT \
  --source=authz-extension.yaml --location=$REGION --project=$PROJECT
gcloud beta network-security authz-policies import $AUTHZ_POLICY \
  --source=authz-policy.yaml --location=$REGION --project=$PROJECT

The policy's resource paths take the project NUMBER, not the ID — both target.resources and customProvider.authzExtension.resources. Resolve it yourself; narrating two numbers at the user does not help. The trap, for when a callout never lands: the number in the gateway's serviceExtensionsServiceAccount is the Google-managed tenant project, not theirs, and it is the only one on screen, so it is the natural thing to copy. A wrong number imports cleanly and targets nothing. Copy the slots from assets/authz-policy.example.yaml.

7. Verify

You cannot test with curl — the host is internal and the proxy expects the gateway's IAP-egress JWT. Run scripts/read_debug_session.sh --arm, drive a turn, read it back with --session. Expect one transaction per turn per governed shape under a suffix list this narrow; zero alone does not mean the callout is dead — the script prints the probe that settles it.

Hard rules

Gate every step — how urgently depends on your AuthzPolicy. An ungated continueOnError="false" policy faults on any call that carries no prompt, and that breaks the agent's turn. Historically the most common failure here.

A narrow suffix list routes only the model call to Apigee, so the trap mostly does not fire. Broaden the policy and it fires at once: the ext-proc then sees every agent egress — session create/read, appendEvent, OpenTelemetry exports — measured at 12 calls for one single-question turn, of which one was the model call. Gate the steps regardless; a condition is cheap and it is what makes the policy safe to widen later. Full measurement: references/gotchas.md.

  • model call: (request.uri Like "*generateContent*")

Never fail-close on your own metering. Keep continueOnError="true" on the JavaScript and quota policies so a metering bug cannot break LLM traffic.

This skill meters by default and can enforce on request. Enforcement is assets/feature-token-enforce.yaml: request-flow EnforceOnly policies sharing each SharedName with the response-flow CountOnly half. Add it to the template's feature list. Verified live on ext-proc traffic — the over-budget caller gets a 429, model never called.

Never offer CountOnly=false as a substitute: in the response flow it cannot gate (the answer is already on the wire) and it freezes the counter, losing the metering too.

State two things whenever you offer it. It is one request late — the turn that crosses the limit completes, because a token cost is not known until the model answers. And it buckets on the requested model, which under an OpenAI alias can differ from the resolved model the counting side bills. The two traps that both present as "it never refuses": references/gotchas.md.

Normalize the model to a bucket. With an OpenAI-compatible endpoint the model is an alias in the request body and the resolved name is in the response body (google/gemini-2.5-pro). Exact-matching a raw name fails on the response path.

jsonPath needs its third argument true so it returns a scalar. Without it the quota policy receives a node it cannot parse.

References

  • references/modes.md — read first. How to answer "what can you do", dev vs prod, the defaults, durations, cost disclosure, the interlocks, and reading an assess report

  • scripts/assess_integration.sh — read-only. Setup Status Summary then a per-check table; --json carries both. Lead with the summary and read out the Enforcement row. Run it before planning: it decides CREATE vs REUSE. TODO is a build step, not a failure; only FAIL exits non-zero

  • references/networking-psc.md — both network legs, PSC, DNS, acceptance

  • references/recipes.md — Model Armor and semantic cache

  • references/verify.md — debug sessions and the checklist script

  • references/setup.md — SDK, project, billing, APIs, region, auth, IAM

  • references/gotchas.md — 63 failure modes and their symptoms

  • assets/apigee-agent-gateway-governance.example.yaml — the input contract

  • assets/apigee-proxy-template.yaml — the AFT template. Features merge by priority and all sit beside it: feature-identity.yaml (10), feature-normalize.yaml (15), feature-model-armor.yaml (20), feature-semantic-cache.yaml (30), feature-token-metering.yaml (50).

  • assets/authz-extension.example.yaml, assets/authz-policy.example.yaml — the step-6 bodies. Copy and substitute; paths take the project number

    Anything read out of the request body is normalised once, at priority 15, not inside whichever feature needs it first. LLM_URI_MATCH is shared, so widening it arms every feature gated on it simultaneously — including Model Armor, which fails closed. Model Armor and the cache both read agw.user_prompt; the quotas read agw.total_tokens, set on the response path by the metering feature's own JS for the same reason.

  • scripts/ensure_apigee_environment.sh — create the environment, attach it to the instance, create and attach the envgroup. Idempotent, --yes gated

  • scripts/ensure_gateway_connectivity.sh — route the gateway into the VPC. Read-modify-write; --yes gated

  • scripts/ensure_vector_search.sh — only when semanticCache is enabled: build index and endpoint, deploy one on the other, read back the domain. --yes gated, ~34 min, bills continuously

  • scripts/teardown_vector_search.sh — the only thing that STOPS that charge. Refuses until the revision drops SCL-lookup. --yes gated

  • scripts/deploy_proxy.sh — the Apigee-side sequence, idempotent, --dry-run

  • scripts/read_debug_session.sh — --arm a session (printing when it actually closes) and read it back as the verify.md checklist, instead of parsing the transaction JSON by hand

  • scripts/compile_check.py — compile the template offline, no org needed

Files

assets/feature-token-enforce.yaml

# Feature: per-model LLM token ENFORCEMENT.
#
# Optional and off by default, because it REFUSES traffic. Add it to the
# template's feature list to turn a budget into a limit.
#
# ---------------------------------------------------------------------------
# HOW IT WORKS
# ---------------------------------------------------------------------------
# LLMTokenQuota splits enforcement from counting across two policies that
# share a <SharedName>:
#
#   request flow   EnforceOnly=true   reads the counter, refuses if over
#   response flow  CountOnly=true     increments the counter, never refuses
#
# feature-token-metering.yaml owns the CountOnly half and runs unchanged.
# Exactly one of CountOnly/EnforceOnly may be set on a given policy.
#
# ENFORCEMENT IS ONE REQUEST LATE, BY CONSTRUCTION. The counter cannot include
# the current turn, because its cost does not exist until the model has
# answered. The turn that crosses the limit COMPLETES, and the caller's NEXT
# request is refused. Inherent, not a defect -- state it when offering this.
#
# ---------------------------------------------------------------------------
# VERIFIED LIVE
# ---------------------------------------------------------------------------
# Confirmed in the shape that matters: a no-route proxy invoked as
# EXT_PROC_GRPC by an Agent Gateway, on real Agent Runtime traffic, where the
# request and response phases arrive as separate gRPC messages.
#
# The over-budget turn was refused and the caller received:
#
#   429 Too Many Requests
#   faultstring: "Rate limit LLM Token quota violation. Quota limit exceeded.
#                 Identifier : principal://agents.global.org-..."
#
# From that transaction's trace, flash budget 500/hour:
#
#   ratelimit.enforce-LTQ-flash.used.count      = 1086   <- written by CountOnly
#   ratelimit.enforce-LTQ-flash.allowed.count   = 500
#   ratelimit.enforce-LTQ-flash.exceed.count    = 586
#   ratelimit.enforce-LTQ-flash.failed          = true
#   error.state                                 = PROXY_REQ_FLOW
#
# The request-path EnforceOnly policy read 1086 -- exactly what the
# response-path CountOnly policy had accumulated under the same SharedName --
# compared it against 500, and refused before the model was called. So the
# SharedName linkage does join the two halves in a no-route ext-proc proxy.
#
# <Identifier> ALSO SCOPES PER CALLER on the enforce half, confirmed in the
# same trace: ratelimit.*.identifier resolved to the calling agent's
# principal, not a shared bucket. It behaves the same as on the count half.
#
# An earlier report of "0 refusals at any budget" was a test harness pointed
# at a different project, so that traffic never reached the proxy under test.
# Not a platform defect.
#
# ---------------------------------------------------------------------------
# TWO TRAPS THAT BOTH PRODUCE "IT NEVER REFUSES"
# ---------------------------------------------------------------------------
# 1. BUCKETING ON A RAW URI SUBSTRING. The enforce half must bucket on the
#    REQUESTED model, derived from the request. Testing the whole URI for a
#    model name is unsafe, because
#      /v1beta1/projects/.../models/gemini-2.5-flash:generateContent
#    contains "pro" inside *pro*jects. A `request.uri Like "*pro*"` condition
#    therefore matches a FLASH request, routes it to the pro bucket, finds
#    headroom there and allows -- while the count half correctly bills flash.
#    That is indistinguishable from a broken platform and is a config error.
#
#    The JS below anchors the match after `models/` precisely for this, and
#    buckets on the extracted name rather than on the URI. Do not "simplify"
#    it into a substring test on request.uri.
#
# 2. OMITTING <LLMModelSource>, which is required -- see the note on the
#    policies below. That one fails loudly rather than silently.
#
# ---------------------------------------------------------------------------
# THE BUCKET DISAGREEMENT, WHICH IS REAL AND DELIBERATE
# ---------------------------------------------------------------------------
# The counting side buckets on the RESOLVED model, read from the response
# body. At request time that does not exist, so this side buckets on the
# REQUESTED model -- the alias in the request body, or the name in the URI.
#
# For Gemini-native traffic the two agree: the URI carries the real model.
# For an OpenAI-compatible endpoint the request carries an ALIAS, which the
# provider may resolve to a different model. A caller asking for "fast-model"
# that resolves to a pro model is ENFORCED against the flash bucket and
# COUNTED against the pro bucket. Everything is still counted and still
# checked; it is only unsafe if someone assumes a per-model limit is exact
# under aliasing.
#
# ---------------------------------------------------------------------------
# continueOnError IS false HERE, AND ONLY HERE
# ---------------------------------------------------------------------------
# The skill's standing rule is that metering must never fail closed: a bug in
# your own observability must not break LLM traffic. These policies are the
# deliberate exception, because refusing is their entire purpose -- a quota
# violation that is swallowed is not enforcement. The cost is that a quota
# policy ERROR also refuses, which is what the missing LLMModelSource did to
# every turn before it was found.

gateway: apigee
schemaVersion: 1.0.0
name: enforce
uid: enforce
# After identity (10), which sets agw.client_id -- the Identifier below needs
# it. Before normalize (15) and everything downstream: a refused request
# should not pay for prompt extraction, Model Armor or a cache lookup.
priority: 12
type: feature
description: >-
  Refuses a request when the caller has already exhausted its token budget
  for the model it is asking for. Reads the counters feature-token-metering
  writes; never increments them. One request late by construction.

parameters:
  # These MUST match the corresponding *_TOKEN_LIMIT values in
  # feature-token-metering.yaml. They are separate parameters because AFT does
  # not share parameters across features, and a mismatch is silent: the
  # enforce half would refuse at a different threshold than the count half
  # reports against. Keep them in step, and prefer changing both or neither.
  - name: PRO_TOKEN_LIMIT
    default: "10000"
  - name: FLASH_TOKEN_LIMIT
    default: "40000"
  - name: OTHER_TOKEN_LIMIT
    default: "40000"
  - name: QUOTA_INTERVAL
    default: "1"
  - name: QUOTA_TIME_UNIT
    default: hour
  # Byte-identical to the other features that declare it. They must match:
  # each feature resolves its OWN default (gcloud passes an empty caller map),
  # so a difference here is a silently different gate, not an override.
  - name: LLM_URI_MATCH
    default: '((request.uri Like "*generateContent*") or (request.uri Like "*chat/completions*"))'

resources:
  - name: extract-model-request.js
    type: jsc
    content: |
      // Request-phase model bucket, for the enforce-only quotas.
      //
      // Only request-available sources: the OpenAI-compatible alias in the
      // body, and the Gemini-native model name in the URI. The resolved name
      // lives in the response body and does not exist yet -- which is exactly
      // why enforcement buckets on the requested model and can disagree with
      // the bucket the counting side later uses. See the header.
      //
      // Bucketing is kept byte-identical to extract-model.js so the two sides
      // agree whenever the requested and resolved names agree. If you change
      // one, change the other.
      function readField(v, field) {
        try {
          var b = context.getVariable(v);
          if (b) { var j = JSON.parse(b); if (j && j[field]) return String(j[field]); }
        } catch (e) {}
        return "";
      }
      function readModelFromUri() {
        try {
          var u = context.getVariable("request.uri");
          if (u) { var m = String(u).match(/\/models\/([^:\/?]+)/); if (m) return m[1]; }
        } catch (e) {}
        return "";
      }
      var raw = readField("request.content", "model")
             || readModelFromUri()
             || "unknown";
      var ml = raw.toLowerCase(), bucket = "other";
      if (ml.indexOf("pro") >= 0) bucket = "gemini-pro";
      else if (ml.indexOf("flash") >= 0) bucket = "gemini-flash";
      context.setVariable("agw.req_model_bucket", bucket);
      context.setVariable("agw.req_model_raw", raw);

policies:
  - name: JS-extract-model-request
    type: Javascript
    content:
      Javascript:
        metadata:
          name: JS-extract-model-request
          # Stays true: a bucketing failure must not refuse the request. It
          # falls through to the "other" bucket, which is still enforced.
          continueOnError: true
          enabled: true
          timeLimit: "200"
        ResourceURL: jsc://extract-model-request.js

  # Three enforce policies, one per SharedName, because <SharedName> is a
  # static literal per policy -- it is the join key to the CountOnly half and
  # cannot be templated per request. One bucket therefore needs one policy,
  # each with its own Allow count. The step condition picks which one runs.
  # (Each still sets <LLMModelSource>; see the note on the first policy.)
  - name: LTQ-enforce-pro
    type: LLMTokenQuota
    content:
      LLMTokenQuota:
        metadata:
          name: LTQ-enforce-pro
          # false ON PURPOSE -- see the header. This is the only place in the
          # skill that fails closed on a quota.
          continueOnError: false
          enabled: true
        SharedName: llm-tokens-pro
        Allow:
          metadata:
            count: "{PRO_TOKEN_LIMIT}"
        Interval: "{QUOTA_INTERVAL}"
        TimeUnit: "{QUOTA_TIME_UNIT}"
        EnforceOnly: "true"
        Distributed: true
        Synchronous: true
        Identifier:
          metadata:
            ref: agw.client_id
        # REQUIRED. Omitting this does NOT mean "this policy needs no model".
        # The policy substitutes a DEFAULT of
        #   jsonPath('$.model', request.content, true)
        # and on a Gemini-native :generateContent call the model is in the URI,
        # not the body -- so it resolves to nothing, raises
        # FailedToResolveModelName, and because these policies are
        # continueOnError="false" the fault kills EVERY governed turn, not just
        # over-budget ones. Confirmed as a live outage (FINDINGS F12).
        # Feed it the bucket this feature's own JS computed one step earlier.
        LLMModelSource: "{agw.req_model_bucket}"

  - name: LTQ-enforce-flash
    type: LLMTokenQuota
    content:
      LLMTokenQuota:
        metadata:
          name: LTQ-enforce-flash
          continueOnError: false
          enabled: true
        SharedName: llm-tokens-flash
        Allow:
          metadata:
            count: "{FLASH_TOKEN_LIMIT}"
        Interval: "{QUOTA_INTERVAL}"
        TimeUnit: "{QUOTA_TIME_UNIT}"
        EnforceOnly: "true"
        Distributed: true
        Synchronous: true
        Identifier:
          metadata:
            ref: agw.client_id
        # REQUIRED. Omitting this does NOT mean "this policy needs no model".
        # The policy substitutes a DEFAULT of
        #   jsonPath('$.model', request.content, true)
        # and on a Gemini-native :generateContent call the model is in the URI,
        # not the body -- so it resolves to nothing, raises
        # FailedToResolveModelName, and because these policies are
        # continueOnError="false" the fault kills EVERY governed turn, not just
        # over-budget ones. Confirmed as a live outage (FINDINGS F12).
        # Feed it the bucket this feature's own JS computed one step earlier.
        LLMModelSource: "{agw.req_model_bucket}"

  - name: LTQ-enforce-other
    type: LLMTokenQuota
    content:
      LLMTokenQuota:
        metadata:
          name: LTQ-enforce-other
          continueOnError: false
          enabled: true
        SharedName: llm-tokens-other
        Allow:
          metadata:
            count: "{OTHER_TOKEN_LIMIT}"
        Interval: "{QUOTA_INTERVAL}"
        TimeUnit: "{QUOTA_TIME_UNIT}"
        EnforceOnly: "true"
        Distributed: true
        Synchronous: true
        Identifier:
          metadata:
            ref: agw.client_id
        # REQUIRED. Omitting this does NOT mean "this policy needs no model".
        # The policy substitutes a DEFAULT of
        #   jsonPath('$.model', request.content, true)
        # and on a Gemini-native :generateContent call the model is in the URI,
        # not the body -- so it resolves to nothing, raises
        # FailedToResolveModelName, and because these policies are
        # continueOnError="false" the fault kills EVERY governed turn, not just
        # over-budget ones. Confirmed as a live outage (FINDINGS F12).
        # Feed it the bucket this feature's own JS computed one step earlier.
        LLMModelSource: "{agw.req_model_bucket}"

  # Sanitises the refusal the caller sees. LLMTokenQuota's own faultstring
  # interpolates <Identifier>, and ours is a ref to agw.client_id -- the
  # gateway's assertion about the calling workload. Unmodified, an over-budget
  # caller is handed the full principal:
  #
  #   Identifier : principal://agents.global.org-<ORG_NUMBER>.system.id.goog
  #     /resources/aiplatform/projects/<PROJECT_NUMBER>/locations/<REGION>
  #     /reasoningEngines/<ENGINE_ID>
  #
  # That is the org number, the project number, the region and the agent's
  # resource id, in an error body that reaches whoever called the agent. Fine
  # in a demo you own end to end; an information-disclosure finding in a
  # customer deployment, where the caller may be an end user.
  #
  # The replacement keeps the SHAPE and the ERRORCODE so anything branching on
  # `policies.llmtokenquota.LLMTokenQuotaViolation` keeps working, and drops
  # only the `Identifier : ...` clause. The identifier is still in the proxy's
  # own analytics and debug trace, which is where it belongs.
  #
  # continueOnError is TRUE on purpose, and it is the opposite of the quota
  # policies above. This policy does not decide anything: if it fails, the
  # original fault still stands and the caller is still refused, just with the
  # unsanitised message. Failing closed here would turn a cosmetic problem
  # into a dropped refusal.
  - name: AM-quota-refusal
    type: AssignMessage
    content:
      AssignMessage:
        metadata:
          name: AM-quota-refusal
          continueOnError: true
          enabled: true
        IgnoreUnresolvedVariables: true
        Set:
          StatusCode: "429"
          ReasonPhrase: Too Many Requests
          Payload:
            metadata:
              contentType: application/json
            _text: >-
              {"fault":{"faultstring":"Rate limit LLM Token quota violation.
              Quota limit exceeded.","detail":{"errorcode":"policies.llmtokenquota.LLMTokenQuotaViolation"}}}
        AssignTo:
          metadata:
            createNew: "false"
            type: response

defaultEndpoint:
  name: default
  flows:
    - name: PreFlow
      mode: Request
      steps:
        # Gated on LLM_URI_MATCH like every other step in this template. The
        # ext-proc sees every agent egress -- session create/read,
        # appendEvent, OpenTelemetry exports -- and refusing one of those
        # because a model budget is exhausted would break the turn for a call
        # that consumed no tokens.
        - name: JS-extract-model-request
          condition: "{LLM_URI_MATCH}"
        - name: LTQ-enforce-pro
          condition: >-
            {LLM_URI_MATCH} and agw.req_model_bucket = "gemini-pro"
        - name: LTQ-enforce-flash
          condition: >-
            {LLM_URI_MATCH} and agw.req_model_bucket = "gemini-flash"
        - name: LTQ-enforce-other
          condition: >-
            {LLM_URI_MATCH} and agw.req_model_bucket = "other"

  # A fault rule, not a flow step: the quota policies run continueOnError
  # "false", so the moment one refuses, the request flow stops and no later
  # step runs. The only place left to touch the response is fault handling.
  #
  # A fault rule has no request/response phase -- do NOT give it a `mode`.
  # models.FaultRule inherits `mode` from Flow, so it parses, but the
  # converter logs "Ignoring mode ... a fault rule has no request or
  # response" and drops it.
  #
  # The fault still happens and the caller is still refused. This only
  # rewrites what they are shown.
  faultRules:
    - name: quota-refusal
      condition: (fault.name = "LLMTokenQuotaViolation")
      steps:
        - name: AM-quota-refusal

Requirements and compatibility

Works with: Claude Code, Claude apps, OpenAI Codex, GitHub Copilot, Cursor, Gemini CLI, Kiro, OpenCode, Goose, Amp, Roo Code, JetBrains Junie, Factory Droid, Devin (ex-Windsurf), Mistral Vibe, OpenHands, Letta

“Works with” means the tool reads the open Agent Skills format; a skill written for one tool may use features others lack.

Needs: Needs a command-line tool, Needs Python, Needs Node.js

Security

What our scanner noticed· Low risk

  • Instructions that use the network
  • Scripts that your agent may run

This skill includes scripts that your agent may run. Read them before you use it.

A static scan is not a guarantee. Skills are third-party content: review a skill before you let an agent use it.

Attribution and licence

Published by
Google
Publisher
Google (Official)
Source
google/skills · skills/cloud/apigee-agent-gateway-governance · commit 2194d3b
Licence
Apache License 2.0 (redistribution allowed)
  • LICENSE (added by AutoGineer from LICENSE)
  • ATTRIBUTION.md (written by AutoGineer)
Synced

AutoGineer is not affiliated with or endorsed by Google.

Download ZIP

The ZIP holds the skill’s files, its licence files and an ATTRIBUTION.md. Download apigee-agent-gateway-governance.zip