The auth broker and auth gateway are two cooperating HTTP services that move OAuth refresh tokens and provider access tokens off developer laptops and into a single broker host.
omp auth-broker serveholds the canonical SQLite credential vault, performs OAuth refreshes, and exposes a small REST API (/v1/snapshot,/v1/snapshot/stream,/v1/credential/:id/refresh,/v1/credential/:id/disable,/v1/credential,/v1/usage,/v1/healthz).omp auth-gateway serveis a forward-proxy. It accepts OpenAI Chat Completions, Anthropic Messages, OpenAI Responses, and pi-native stream requests, resolves the broker-backed credential, and dispatches throughpi-aiprovider logic. Clients (containerised omp, llm-git, the macOS usage widget, …) never see the access token.
Transport security between operator, broker, and gateway is delegated to the operator (Tailscale / Wireguard / reverse proxy + TLS). Every endpoint except /v1/healthz (broker) and /healthz (gateway) requires a bearer token.
Source: packages/ai/src/auth-broker/, packages/ai/src/auth-gateway/, packages/coding-agent/src/cli/auth-broker-cli.ts, packages/coding-agent/src/cli/auth-gateway-cli.ts, packages/coding-agent/src/session/auth-broker-config.ts.
Data flow
graph TD %% Styling Definitions classDef client fill:#eceff1,stroke:#37474f,stroke-width:1px,color:#263238; classDef host fill:#f5f5f5,stroke:#9e9e9e,stroke-width:2px,stroke-dasharray: 5 5,color:#212121; classDef component fill:#e3f2fd,stroke:#1e88e5,stroke-width:2px,color:#0d47a1; classDef db fill:#efebe9,stroke:#6d4c41,stroke-width:2px,color:#3e2723; classDef provider fill:#ffe0b2,stroke:#f57c00,stroke-width:2px,color:#e65100; %% Nodes Declaration subgraph DevEnv ["Client / CI Environment"] Developer["Developer Laptop /<br/>CI / robomp"] end subgraph BrokerHost ["Broker Host"] AuthBroker["<b>omp auth-broker serve</b><br/>• Holds refresh tokens<br/>• Background refresher<br/>• /v1/{snapshot, refresh, ...}"] DB[("<b>SQLite agent.db</b><br/>(canonical writer)")] AuthGateway["<b>omp auth-gateway serve</b><br/>• /v1/{chat, messages, ...}<br/>• /v1/usage, /v1/models<br/>• /v1/credentials/check<br/><br/><i>RemoteAuthCredentialStore:</i><br/>Receives snapshot stream, refreshes<br/>credentials by id via the broker on expiry"] end GatewayClients["<b>Gateway Clients</b><br/>(llm-git, macOS widget, robomp containers, IDE plugins, ...)"] APIProviders["<b>API Providers</b><br/>(api.anthropic.com / api.openai.com / ...)"] %% Relationships and Flow Developer -->|Request| AuthBroker AuthBroker <-->|Read / Write| DB AuthBroker -->|bearer<br/>$CONFIG_DIR/auth-broker.token| AuthGateway AuthGateway -->|bearer<br/>$CONFIG_DIR/auth-gateway.token| GatewayClients GatewayClients -->|Provider request with<br/>broker-resolved credential| APIProviders %% Apply Classes class Developer client; class GatewayClients client; class AuthBroker,AuthGateway component; class DB db; class APIProviders provider; class BrokerHost host;
The broker is the only writer of OAuth refresh tokens. Clients (including the gateway itself) load a redacted snapshot in which every refresh field has been replaced with REMOTE_REFRESH_SENTINEL; when an access token expires the client calls POST /v1/credential/:id/refresh and the broker performs the refresh server-side. RemoteAuthCredentialStore rejects local replace/upsert/delete-by-provider mutations, with errors pointing at omp auth-broker login / omp auth-broker logout.
auth-broker
CLI
omp auth-broker serve [--bind=host:port] # boot the broker
omp auth-broker token [--regenerate] [--json] # print or rotate the bearer token
omp auth-broker login [<provider>] [--via=user@host] [--dry-run]
omp auth-broker logout [<provider>]
omp auth-broker list [--json]
omp auth-broker import <file|dir> [--provider=<id>] [--include-disabled] [--dry-run] [--json]
omp auth-broker migrate --from-local [--include-oauth] [--include-env] [--dry-run] [--json]
omp auth-broker status [--json]
serveopens the local SQLite store atgetAgentDbPath()and binds an HTTP listener (default127.0.0.1:8765). On startup a token is ensured at<config-dir>/auth-broker.token(mode0600,0700parent dir). The background refresher refreshes any OAuth credential whoseexpires - Date.now() < refreshSkewMs(default 5 min) everyrefreshIntervalMs(default 60 s).tokenprints the cached bearer or generates a new one.--regeneraterotates it.login [<provider>]runs the per-provider OAuth flow locally — when no provider is supplied, it falls back to an interactive numbered picker. With--via=user@hostit shells outssh -L <callback-port>:127.0.0.1:<callback-port> user@host omp auth-broker login <provider>so the OAuth callback hits the local browser but the credential is written on the broker host (--viarequires<provider>). Built-in callback ports:anthropic:54545,openai-codex:1455,google-gemini-cli:8085,google-antigravity:51121,gitlab-duo:8080. The OAuth dance is driven in-process viaAuthStorage.login()— there is no longer api-aibin to spawn.logout [<provider>]deletes every credential row for<provider>. With no argument it shows an interactive numbered picker of currently-stored providers.listenumerates every registered OAuth provider id/name (the union of built-ins +registerOAuthProvidercustom providers).--jsonemits a machine-readable array.import <file|dir>imports CLIProxyAPI-style JSON credentials into the local SQLite store. Mapstypefield → omp provider (claude → anthropic,codex → openai-codex,gemini → google-gemini-cli,antigravity → google-antigravity,gemini-cli → google-gemini-cli).migrate --from-localuploads local SQLite credentials to the configured broker (POST /v1/credential). Local API keys are included by default; local OAuth rows are skipped unless--include-oauthis set; environment-derived API keys are skipped unless--include-envis set. Re-runs are idempotent against the broker snapshot.statushealth-pings the configured remote broker.
Endpoints
| Method | Path | Auth | Purpose |
|---|---|---|---|
GET | /v1/healthz | none | Liveness + version |
GET | /v1/snapshot | bearer | Redacted snapshot (refresh tokens replaced by sentinel) |
GET | /v1/snapshot/stream | bearer | SSE snapshot stream with delta events and keepalives |
POST | /v1/credential | bearer | Upsert one OAuth or API-key credential |
POST | /v1/credential/:id/refresh | bearer | Force-refresh one OAuth credential |
POST | /v1/credential/:id/disable | bearer | Disable one credential with a recorded cause |
GET | /v1/usage | bearer | Aggregate UsageReport[] across credentials |
Requests use Authorization: Bearer <token>. The server compares against an in-memory token allow-list; the gateway’s implementation uses a timing-safe comparison.
Background refresher
AuthBrokerRefresher iterates active OAuth credentials at refreshIntervalMs cadence and refreshes any within refreshSkewMs of expiry. Refreshes are single-flighted per credential id so a slow refresh cannot be retriggered. The refresher distinguishes:
- definitive failures (
invalid_grant,invalid_token,revoked, unauthorized refresh-token, 401/403 not from a network blip) — credentials are passed toAuthStorage.disableCredentialById(id, cause)so the next snapshot pull surfaces a clean delete on the client; - transient failures (timeout / ECONNREFUSED / fetch failed) — left in place for the next sweep.
auth-gateway
CLI
omp auth-gateway serve [--bind=host:port] [--no-auth]
omp auth-gateway token [--regenerate] [--json]
omp auth-gateway status [--json]
omp auth-gateway check [--strict] [--json]
serverequiresOMP_AUTH_BROKER_URL(orauth.broker.urlinconfig.yml) — the gateway is itself a broker client. It callsAuthBrokerClient.fetchSnapshot(), wraps it inRemoteAuthCredentialStore, and constructs anAuthStoragethat resolves access tokens through the broker. Default bind is127.0.0.1:4000. The gateway token is stored at<config-dir>/auth-gateway.token(0600);--no-authdisables the bearer check entirely (loopback-only use).token/statusmanage and inspect the gateway bearer token and upstream broker readiness.checkprobes broker-backed credentials through the gateway store. Without--strictit uses provider usage probes;--strictalso exercises each credential against its chat-completion endpoint and can consume a small amount of quota.
Endpoints
| Method | Path | Auth | Purpose |
|---|---|---|---|
GET | /healthz | none | Liveness + version |
GET | /v1/usage | bearer | Aggregate UsageReport[] (proxied through AuthStorage) |
GET | /v1/models | bearer | Bundled-model catalog filtered to providers with credentials |
GET | /v1/credentials/check | bearer | Per-credential auth health probe |
POST | /v1/chat/completions | bearer | OpenAI Chat Completions wire format |
POST | /v1/messages | bearer | Anthropic Messages wire format |
POST | /v1/responses | bearer | OpenAI Responses wire format |
POST | /v1/pi/stream | bearer | Native pi-ai stream wire format |
The model id is read from the top-level model field for foreign wire formats and from the pi-native request body for /v1/pi/stream. The gateway picks the first bundled Model<Api> matching that id, parses the inbound wire format into an omp Context, resolves the provider credential from broker-backed AuthStorage, dispatches through streamSimple(), and re-encodes the result to the inbound format (SSE for streamed responses).
There is no raw provider passthrough path. All supported routes go through pi-ai provider logic so credential-specific request shaping, OAuth refresh-on-auth-error, and provider quirks stay centralized.
idleTimeout on the underlying Bun.serve is set to 255 s so long thinking-budget calls do not get killed by Bun’s default idle timeout.
Usage cache: server-side 5-min jitter + client-side 15 s single-flight
Two layers cache the aggregate provider-usage report. Both are intentional and stacked.
Server-side cache (broker AuthStorage)
AuthStorage caches each credential’s UsageReport in the broker’s SQLite store at a 5-minute per-credential TTL with ±25 % jitter. Anthropic and OpenAI rate-limit /usage aggressively per source IP, and a synchronized 5-credential fan-out trips 429s every cycle; the jitter decorrelates refresh times within a few cycles. On fetch failure the store keeps the last-good report for up to 24 h with a short jittered re-poll window — so a transient upstream blip never blanks out the widget.
Constants: USAGE_REPORT_TTL_MS = 5 * 60_000, USAGE_LAST_GOOD_RETENTION_MS = 24 * 60 * 60_000 (packages/ai/src/auth-storage.ts).
Client-side single-flight (RemoteAuthCredentialStore)
When the gateway (or any other broker client) calls fetchUsageReports() / getUsageReport(provider, credential), RemoteAuthCredentialStore coalesces concurrent calls into a single GET /v1/usage round-trip and caches the result for 15 s in memory.
USAGE_CACHE_TTL_MS = 15_000(packages/ai/src/auth-broker/remote-store.ts).- A single
#usageInflightpromise is shared across all callers; a per-callerAbortSignalis raced against the shared promise, not threaded into it, so one caller’s abort never cascades into a peer’s in-flight request. - On fetch failure the rejected promise is logged and the awaited value is
null— callers (AuthStorage.fetchUsageReports,#getUsageReport) treat anullreport as “no usage signal for this cycle” and proceed without it. This is the 15 s TTL fallback: the client absorbs transient broker outages by suppressing the error, returningnullto ranking, and re-attempting after the 15 s window.
The 15 s client window deliberately sits below the broker’s 5 min server cache, so almost every client poll is served from the broker’s already-cached value; the client cache exists to absorb the parallel fan-out generated by AuthStorage.#rankOAuthSelections into a single broker round-trip.
Client snapshot cache
discoverAuthStorage() persists the broker snapshot to ~/.omp/cache/auth-broker-snapshot.enc after the initial /v1/snapshot fetch and after later broker-sourced full snapshots. The file is AES-256-GCM encrypted with SHA-256(OMP_AUTH_BROKER_TOKEN) and authenticated with the broker URL as additional data, so changing either the token or URL makes the cache unreadable. The file is written atomically with mode 0600.
Freshness is anchored to the broker-stamped snapshot.generatedAt, not local write time. Default TTL is 1 h (OMP_AUTH_BROKER_SNAPSHOT_TTL_MS); 0 disables the cache and restores the old always-fetch boot path. When the cached snapshot is still fresh, omp boots from it and skips the blocking /v1/snapshot query. RemoteAuthCredentialStore still starts its normal SSE / long-poll background sync immediately, so deleted or rotated credentials reconcile after startup, and expired OAuth access tokens still refresh through POST /v1/credential/:id/refresh.
If the broker is down at boot and a fresh cache exists, startup now succeeds from the cached snapshot. If the cache is missing, expired, corrupt, written for a different URL, or encrypted with a different token, startup falls back to the live fetch and fails the same way it did before if the broker is unreachable.
Operator opt-in
The broker is off unless OMP_AUTH_BROKER_URL (or auth.broker.url in config.yml) is set. When set, discoverAuthStorage in packages/coding-agent/src/sdk.ts swaps the local SQLite credential store for RemoteAuthCredentialStore and every API call resolves credentials through the broker.
Environment variables
| Variable | Purpose | Required when |
|---|---|---|
OMP_AUTH_BROKER_URL | Base URL of the remote auth-broker (e.g. https://broker.tailnet:8765). Selecting this puts the client in broker mode — local SQLite is bypassed. | Any time the omp client should resolve credentials through a broker (and required by omp auth-gateway serve). |
OMP_AUTH_BROKER_TOKEN | Bearer token used for every broker endpoint except /v1/healthz. | When OMP_AUTH_BROKER_URL is set and no token is available from auth.broker.token or <config-dir>/auth-broker.token. |
OMP_AUTH_BROKER_SNAPSHOT_TTL_MS | Freshness window for the encrypted local snapshot cache. Default 3600000 (1 h); 0 disables cache reads and writes. | Optional in broker mode. |
OMP_AUTH_BROKER_SNAPSHOT_CACHE | Path override for the encrypted local snapshot cache. Default ~/.omp/cache/auth-broker-snapshot.enc (or XDG cache equivalent). | Optional in broker mode. |
Resolution order in resolveAuthBrokerConfig():
OMP_AUTH_BROKER_URLenv (elseauth.broker.urlfromconfig.yml, resolved throughresolveConfigValue);OMP_AUTH_BROKER_TOKENenv (elseauth.broker.tokenfromconfig.yml, else<config-dir>/auth-broker.token);- URL set but no token resolvable → hard error pointing at the token file path.
The gateway has no dedicated env vars — it inherits OMP_AUTH_BROKER_* because it is itself a broker client.
config.yml keys
| Key | Default | Purpose |
|---|---|---|
auth.broker.url | unset | Same as OMP_AUTH_BROKER_URL; env wins. Hidden from the settings UI. Values are resolved as a literal, an environment variable name, or !<shell command> to use trimmed stdout. |
auth.broker.token | unset | Same as OMP_AUTH_BROKER_TOKEN; env wins. Values are resolved the same way. |
Token files
| Path | Owner | Mode |
|---|---|---|
<config-dir>/auth-broker.token | omp auth-broker serve (created at first start) | 0600 in a 0700 parent dir |
<config-dir>/auth-gateway.token | omp auth-gateway serve (skipped under --no-auth) | 0600 in a 0700 parent dir |
<config-dir> resolves to ~/.omp/ (respecting PI_CONFIG_DIR).
Interaction with the local API-key resolution order
The broker only owns OAuth credentials and provider-API-key credentials that were uploaded to it. The standard credential ladder in models.md (Auth and API key resolution order) is preserved, with one addition committed alongside the gateway:
AuthStorage.setConfigApiKey / removeConfigApiKey / clearConfigApiKeyslet amodels.ymlapiKeybeat a stored OAuth token without overriding an explicit--api-key. This is what allows a broker-resolved OAuth credential to be reliably shadowed by a per-environmentmodels.ymlconfig key when both are present.
See also
secrets.md— secret obfuscation around tokens that do leak through (e.g.OMP_AUTH_BROKER_TOKENin shell output).models.md— provider auth resolution order; the broker plugs in at layers 2–3 (stored credentials).environment-variables.md— full env reference includingOMP_AUTH_BROKER_URL/OMP_AUTH_BROKER_TOKEN.