management/handlers + wiring — HTTP API + gRPC delivery
Risk level: Medium — the surface is mostly additive, but two changes are load-bearing:
injectAllProxyPoliciesruns on every per-peer compute, andshallowCloneMappingmust round-tripPrivate(a missed field silently breaks every MODIFIED). Backward-compat impact: Additive on the wire (new routes, new RPCs, new proto fields, new gorm column onAccessLogEntry). One management-internal break:nbhttp.NewAPIHandlergains a trailingagentNetworkManagerparameter;nilis tolerated and silently skips route registration.
Module boundary
This module is the seam between the public Agent Network HTTP API and the proxy fleet that serves agent traffic. North side: a /api/agent-network/* surface (providers, policies, guardrails, budget rules, settings, consumption) on the existing gorilla router, delegating to agentnetwork.Manager. Handlers are thin — they translate api.* ↔ types.*, validate shape, forward. RBAC and event emission stay inside the manager (manager.go:680-682).
South side: ProxyServiceServer (proxy.go) learns to (a) ship synth services to a proxy on initial snapshot, (b) resolve agent-network domains in getServiceByDomain for OIDC/session/tunnel-peer flows, (c) gate LLM requests via CheckLLMPolicyLimits + RecordLLMUsage, (d) preserve Private through shallowCloneMapping so per-proxy live updates don't silently flip services public. The network_map controller prepends synth services to account.Services on every per-peer compute; accesslogentry.go gains an indexed AgentNetwork column so the dashboard can filter cheaply.
Files
| Path | Role |
|---|---|
handlers/agentnetwork/providers_handler.go |
Catalog + provider CRUD + central AddEndpoints |
handlers/agentnetwork/policies_handler.go |
Policy CRUD + shared validatePolicy* |
handlers/agentnetwork/guardrails_handler.go |
Guardrail CRUD |
handlers/agentnetwork/budget_handler.go |
Account-level budget rule CRUD |
handlers/agentnetwork/settings_handler.go |
GET (200+null if unbootstrapped) + PUT toggles |
handlers/agentnetwork/consumption_handler.go |
Read-only consumption rows |
handlers/agentnetwork/handlers_test.go |
Real-store fixture; wire round-trip + validation |
handlers/agentnetwork/budget_handler_test.go |
Budget-rule + settings toggles |
server/http/handler.go |
New agentNetworkManager arg; conditional AddEndpoints |
server/permissions/modules/module.go |
New AgentNetwork module key |
internals/server/boot.go |
Wires synthesiser adapter + limits service into proxy server |
internals/server/modules.go |
AgentNetworkManager() lazy-create node |
internals/controllers/network_map/controller/controller.go |
injectAllProxyPolicies replaces 4 InjectProxyPolicies calls |
internals/controllers/network_map/controller/repository.go |
SynthesizeAgentNetworkServices repo method |
internals/modules/reverseproxy/service/service.go |
MiddlewareConfig, capture limits, AgentNetwork, DisableAccessLog + proto |
internals/modules/reverseproxy/accesslogs/accesslogentry.go |
Indexed AgentNetwork bool from proto |
internals/shared/grpc/proxy.go |
Synth wiring, 2 RPCs, domain fallback, Private in clone |
internals/shared/grpc/proxy_clone_test.go |
Locks every ProxyMapping field minus AuthToken |
server/activity/codes.go |
13 new activity codes (125-137) |
HTTP routes added
All routes inherit the platform's auth middleware. Perms enforced inside agentnetwork.Manager.requirePermission (manager.go:680-682) on modules.AgentNetwork. Permission column shows the op passed to requirePermission — read = Read, etc.
| Method | Path | Perm | Handler |
|---|---|---|---|
| GET | /agent-network/catalog/providers |
authn only | providers_handler.go:43 |
| GET | /agent-network/providers |
read | providers_handler.go:57 |
| POST | /agent-network/providers |
create | providers_handler.go:97 |
| GET | /agent-network/providers/{providerId} |
read | providers_handler.go:77 |
| PUT | /agent-network/providers/{providerId} |
update | providers_handler.go:132 |
| DELETE | /agent-network/providers/{providerId} |
delete | providers_handler.go:172 |
| GET | /agent-network/policies |
read | policies_handler.go:32 |
| POST | /agent-network/policies |
create | policies_handler.go:72 |
| GET | /agent-network/policies/{policyId} |
read | policies_handler.go:52 |
| PUT | /agent-network/policies/{policyId} |
update | policies_handler.go:102 |
| DELETE | /agent-network/policies/{policyId} |
delete | policies_handler.go:142 |
| GET | /agent-network/guardrails |
read | guardrails_handler.go:25 |
| POST | /agent-network/guardrails |
create | guardrails_handler.go:65 |
| GET | /agent-network/guardrails/{guardrailId} |
read | guardrails_handler.go:45 |
| PUT | /agent-network/guardrails/{guardrailId} |
update | guardrails_handler.go:95 |
| DELETE | /agent-network/guardrails/{guardrailId} |
delete | guardrails_handler.go:135 |
| GET | /agent-network/budget-rules |
read | budget_handler.go:24 |
| POST | /agent-network/budget-rules |
create | budget_handler.go:64 |
| GET | /agent-network/budget-rules/{ruleId} |
read | budget_handler.go:44 |
| PUT | /agent-network/budget-rules/{ruleId} |
update | budget_handler.go:95 |
| DELETE | /agent-network/budget-rules/{ruleId} |
delete | budget_handler.go:135 |
| GET | /agent-network/settings |
read | settings_handler.go:53 (200+null if no row) |
| PUT | /agent-network/settings |
update | settings_handler.go:27 |
| GET | /agent-network/consumption |
read | consumption_handler.go:21 |
gRPC RPCs added (or modified)
| RPC | Direction | Trigger |
|---|---|---|
CheckLLMPolicyLimits |
proxy→mgmt unary | Pre-flight gate; returns allow/deny, selected policy, attribution group, window, deny code+reason (proxy.go:259-301). Unimplemented when limits service is nil. |
RecordLLMUsage |
proxy→mgmt unary | Post-flight write of tokens+cost against policy-window dimensions + every applicable account budget rule (proxy.go:303-349). window_seconds==0 ⇒ no policy cap, only account fan-out runs. |
GetMappingUpdate/SendServiceUpdate (stream) |
mgmt→proxy | Snapshot (proxy.go:752-780) now appends SynthesizeServicesForCluster. Live updates use SendServiceUpdateToCluster + shallowCloneMapping. |
Architecture & flow
HTTP request lifecycle
sequenceDiagram
participant DB as Dashboard
participant R as gorilla.Router (/api)
participant H as handler (agentnetwork)
participant M as agentnetwork.Manager
participant S as store.Store
participant AM as accountManager (StoreEvent)
DB->>R: POST /api/agent-network/providers
R->>H: createProvider (auth mw sets UserAuth)
H->>H: GetUserAuthFromContext + validate(req)
H->>M: CreateProvider(userID, provider, bootstrapCluster)
M->>M: requirePermission(AgentNetwork, Create)
M->>S: SaveAgentNetworkProvider
M->>AM: StoreEvent(AgentNetworkProviderCreated)
M-->>H: created provider
H-->>DB: 200 + api.AgentNetworkProvider JSON
Synth-service delivery via gRPC
sequenceDiagram
participant P as Proxy
participant G as ProxyServiceServer
participant SM as service.Manager (persisted)
participant SA as synthesizerAdapter
participant AN as SynthesizeServicesForCluster
participant ST as store.Store
Note over P,G: Initial snapshot
P->>G: GetMappingUpdate (stream open)
G->>SM: GetServicesForCluster(conn.address)
SM-->>G: persisted []*Service
G->>SA: SynthesizeServicesForCluster(conn.address)
SA->>AN: SynthesizeServicesForCluster(store, clusterAddr)
AN->>ST: walk every account; read providers/policies/settings
AN-->>SA: in-memory []*Service
SA-->>G: []*Service
G->>P: response (persisted + synth)
Note over G,P: Per-request live update
G->>G: SendServiceUpdateToCluster(update, clusterAddr)
G->>G: shallowCloneMapping(update) %% Private MUST survive
G->>P: response with single mapping
End-to-end: HTTP write persists rows and emits an activity event; the manager then triggers proxyController.SendServiceUpdate so proxies re-render. The snapshot path is the only one that calls into the synthesiser — on stream open it pulls persisted services then appends synth services for the cluster. Synth services are never persisted. For OIDC/session/tunnel-peer flows, getServiceByDomain falls back to SynthesizeServicesForCluster(clusterFromDomain(domain)) when persisted lookup misses (proxy.go:1763-1793). The network_map contribution is orthogonal: per-peer compute prepends the same synth services to account.Services before InjectProxyPolicies.
Permissions model added
permissions/modules/module.go:22addsAgentNetwork Module = "agent_network", registered inAll(module.go:42). Standardoperations.{Read,Create,Update,Delete}matrix.- Handlers don't call
permissionsManagerdirectly — they extractUserAuthand delegate toagentnetwork.Manager, which gates every mutation throughrequirePermission(manager.go:168, 308, 549, etc.). Confirm your role-set provider hasagent_networkrows for owner/admin/user/billing-admin before merging. getCatalogProviders(providers_handler.go:43) intentionally skips RBAC — catalog is global static data.
Activity codes added
activity/codes.go:244-274 adds Activities 125-137 + string/code mappings (codes.go:428-444), following <domain>.<resource>.<action> (e.g., agent_network.provider.create). Audit-log exporters / SIEM forwarders need to know the new codes.
Invariants
- Synth services are never persisted. Snapshot appends after
serviceManager.GetServicesForCluster(proxy.go:761-770); network_map prepends beforeInjectProxyPolicies(controller.go:117-126). shallowCloneMappingmust round-trip everyProxyMappingfield exceptAuthToken—proxy_clone_test.go:50-58enforces viagproto.Equal. The bug it guards: a missingPrivatemade every MODIFIED arriveprivate=false, the proxy skippedValidateTunnelPeer,UserGroupsstayed empty,llm_routerdeniedno_authorised_provider; a restart "fixed" it because the snapshot uses the original mapping.- Limit-window floor is 60s (
policies_handler.go:189-220); enabled cap with both per-group and per-user at zero is rejected. Budget rules reuse the same validator (budget_handler.go:170). - Manager is optional at boot.
NewAPIHandlerregisters routes only when non-nil (handler.go:129);ProxyServiceServerreturnsUnimplementedfrom both RPCs when limits service is unwired (proxy.go:262-265, 306-309). - Settings GET on an unbootstrapped account returns 200 +
null(settings_handler.go:65-72) — not 404.
Things to scrutinize
Correctness
injectAllProxyPoliciesruns on every per-peer compute:controller.go:163, 309, 415, 681.sendUpdateAccountPeersis the target of the buffered fan-out — synth runs once per debounced account-update tick and once per directUpdateAccountPeer. Cost is O(providers + policies × users-per-group) per account underLockingStrengthNone. No per-account synth cache — verify it fits the buffer interval for your largest tenant.clusterFromDomainstrips at the first.(proxy.go:1784-1792). A zero-dot domain returns""and the synth call walks every account. Confirm no path reaches this with a malformed/internal domain.- Account-budget
RecordConsumptionfans out even whenwindow_seconds == 0(proxy.go:341-348) — intentional. Verify the proxy never sendsRecordLLMUsagefor a request that wasn't actually allowed.
Security
- Every handler extracts
UserAuthvianbcontext.GetUserAuthFromContextbefore any work. Routes live behind the standard/apimux; bypass list is not extended. CheckLLMPolicyLimits/RecordLLMUsageride the existing proxy → mgmt gRPC connection auth. No additional token check inside the RPCs — they trust the connection. Confirm the proxy-side token-verification interceptor in this package gates both.RecordLLMUsageonly validatesaccount_id != ""(proxy.go:317-319). A compromised proxy can attribute cost to any account in its cluster — was already true for prior RPCs but is louder now that data drives denials.
Concurrency
SetAgentNetworkSynthesizer/SetAgentNetworkLimitsServicewrite unders.mu.Lock; read paths copy the interface under read lock (proxy.go:236-247, 260-263, 304-307). Same pattern as existingserviceManager/proxyControllersetters.- Manager writes use
LockingStrengthUpdate; synth reads useLockingStrengthNone— read-after-write via the proxy snapshot can observe a stale view by up to one fan-out tick. - Network_map controller is single-threaded per account; cross-account is parallel.
Backward compatibility
proxy_clone_test.gois the regression net; any newProxyMappingfield must be cloned or explicitly nulled in the test.AccessLogEntryadds indexedAgentNetwork bool— implicit AutoMigrate; deploy story must handle table-rewrite cost on high-volume access-log tables.TargetOptionsgains sevenomitemptyJSON fields (service.go:69-94); on-wire shape stays compatible.targetOptionsToPrototests all fields when deciding nil (service.go:551-556).NewAPIHandlersignature changes — every caller must passagentNetworkManager;nilis supported.
Observability
- 13 new activity codes via
accountManager.StoreEventin the manager — confirm dashboard's audit-log UI maps them. AccessLogEntry.AgentNetworkis indexed for the dashboard's agent-network log filter.- New RPCs log at error level on store/selector failures (
proxy.go:284, 327, 332, 348). Snapshot synth failures degrade to warnings — stream is not aborted (proxy.go:765).
Test coverage
| Test | Locks down |
|---|---|
handlers_test.go::TestPolicyHandler_WindowSecondsRoundTrip |
GET carries window_seconds; legacy window_hours/window_days absent. |
handlers_test.go::TestPolicyHandler_RejectsSubMinuteWindow |
POST <60s returns 4xx. |
handlers_test.go::TestConsumptionHandler_EmptyAccountReturnsArray |
/consumption returns [] — never null. |
handlers_test.go::TestConsumptionHandler_PopulatedAccountListsRows |
RecordConsumption×2 surfaces both with correct tokens/cost/window. |
budget_handler_test.go::TestBudgetRuleHandler_RoundTrip |
Targets + PolicyLimits shape round-trip. |
budget_handler_test.go::TestBudgetRuleHandler_ListReturnsArray |
Empty-list shape. |
budget_handler_test.go::TestBudgetRuleHandler_{RejectsMissingName,RejectsSubMinuteWindow} |
Validation rejections are 4xx. |
budget_handler_test.go::TestSettingsHandler_GetExposesCollectionToggles |
All four toggles + computed Endpoint. |
proxy_clone_test.go::TestShallowCloneMapping_PreservesAllFieldsExceptAuthToken |
Future-proofs clone; every field round-trips, AuthToken dropped. |
Handler tests use a real sqlite store + real manager + always-allow permissions mock (handlers_test.go:53-75). Create/update/delete success paths flow through accountManager.StoreEvent which the fixture doesn't wire — covered by manager-level no-mock tests outside this module.
Known limitations / explicit non-goals
- No pagination on any list endpoint; no bulk endpoints.
- Synth result is not cached — every snapshot and every per-peer compute repeats the store walk.
getSettingsreturning200 + nullis a deliberate dashboard concession.- No rate-limiting beyond the global
/apirate limiter.
Cross-references
- Upstream: shared/api, management/agentnetwork, management/store
- Downstream: proxy/runtime
- End-to-end flow: ../01-end-to-end-flows.md
- Top-level: ../00-overview.md