feat(api-key): add allowed_models/allowed_providers and enforce in routing (#85, #86)

[yacosta738]

Summary

Implements two issues from Phase 2: Auth & Provider Management:

• Add allowed_models and allowed_providers to ApiKeyRecord — schema + CRUD #85 — Data layer: allowed_models and allowed_providers restriction fields on ApiKeyRecord
• Enforce allowed_models and allowed_providers in the routing pipeline #86 — Enforcement layer: reject completion requests where the API key's restrictions exclude the requested model or provider

Changes

Data layer (#85)

• ApiKeyRecord gains allowed_models: Vec<ModelId> and allowed_providers: Vec<ProviderId> (empty = unrestricted)
• Migration V1__allowed_models_providers.sql adds two TEXT columns to api_keys (default '[]')
• CreateApiKeyRequest / UpdateApiKeyRequest accept these fields
• POST /api/api-keys, PUT /api/api-keys/:id, GET /api/api-keys, GET /api/api-keys/:id pass them through

Enforcement layer (#86)

• CortexError::forbidden() + forbidden_code() distinguish model_not_allowed vs provider_not_allowed
• RouteRequest checks model restriction before any provider interaction, and provider restriction after router.select()
• Same checks apply to execute_stream
• HTTP 403 returned for both error codes (OpenAI handler returns JSON with code; Anthropic handler returns plain text matching its existing style)
• Authz middleware stamps new x-authz-allowed-models / x-authz-allowed-providers headers from ApiKeySubject so handlers can hydrate CompletionRequest.restrictions

Testing

• 343 tests passing, 0 failed (full workspace)
• 7 new unit tests in route_request.rs covering model/provider restrictions for both execute and execute_stream
• 3 new tests in auth-sqlite for create/update with restrictions
• clippy --workspace --all-targets -- -D warnings clean

Architecture

• Restrictions live on ApiKeyRecord (domain) → DTOs (transport) → wire format via ApiKeySubject
• Enforcement happens in the use case (RouteRequest) so it applies regardless of which handler invokes it (OpenAI, Anthropic, etc.)
• Headers-based transport from authz to handlers keeps the use case signature unchanged (the existing CompletionRequest already had restrictions field with #[serde(default)] so internal callers are unaffected)

Notes

• Branched off feat/api-key-scope-enforcement (PR feat(api-key): typed scopes with canonical values and route enforcement #89) — will rebase once that merges
• Empty list semantics match OmniRoute-style "all allowed" so migration of existing keys is automatic (new column defaults to '[]')

Closes #85, closes #86

[coderabbitai]

Warning

Review limit reached

@yacosta738, we couldn't start this review because you've reached your PR review rate limit.

More reviews will be available in 14 minutes and 38 seconds. Learn how PR review limits work.

Your organization has run out of usage credits. Purchase more in the billing tab.

⌛ How to resolve this issue?

After more reviews become available, a review can be triggered using the @coderabbitai review command as a PR comment. Alternatively, push new commits to this PR.

We recommend that you space out your commits to avoid hitting the rate limit.

🚦 How do rate limits work?

CodeRabbit enforces hourly rate limits for each developer per organization.

Our paid plans include higher PR review limits than trial, open-source, and free plans. In all cases, reviews become available again over time. During sustained high-volume PR review activity, CodeRabbit may temporarily slow when the next review becomes available.

Please see our Fair Usage Limits Policy for further information.

ℹ️ Review info

⚙️ Run configuration

Configuration used: Organization UI

Review profile: ASSERTIVE

Plan: Pro

Run ID: 5fea54fd-e69b-4f21-859d-73e7f3735627

📥 Commits

Reviewing files that changed from the base of the PR and between 76bbb96 and d4740cb.

⛔ Files ignored due to path filters (1)

• Cargo.lock is excluded by !**/*.lock

📒 Files selected for processing (29)

• crates/application/rook-usecases/Cargo.toml
• crates/application/rook-usecases/src/auth/bootstrap_status.rs
• crates/application/rook-usecases/src/auth/login.rs
• crates/application/rook-usecases/src/auth/validate_session.rs
• crates/application/rook-usecases/src/authenticate_client_api.rs
• crates/application/rook-usecases/src/manage_api_keys.rs
• crates/application/rook-usecases/src/route_request.rs
• crates/application/rook-usecases/src/router_impl.rs
• crates/domain/rook-core/src/api_key.rs
• crates/domain/rook-core/src/lib.rs
• crates/domain/rook-core/src/model.rs
• crates/domain/shared-kernel/src/error.rs
• crates/infrastructure/auth-sqlite/src/lib.rs
• crates/infrastructure/db-migration/src/migrations/V1__allowed_models_providers.sql
• crates/infrastructure/providers-anthropic/tests/provider.rs
• crates/infrastructure/providers-gemini/tests/provider.rs
• crates/infrastructure/providers-groq/tests/provider.rs
• crates/infrastructure/providers-ollama/tests/provider.rs
• crates/infrastructure/providers-openai/tests/provider.rs
• crates/infrastructure/transport-axum/Cargo.toml
• crates/infrastructure/transport-axum/src/anthropic_adapter.rs
• crates/infrastructure/transport-axum/src/authz.rs
• crates/infrastructure/transport-axum/src/format_registry.rs
• crates/infrastructure/transport-axum/src/handlers/api_key.rs
• crates/infrastructure/transport-axum/src/handlers/auth.rs
• crates/infrastructure/transport-axum/src/openai_adapter.rs
• crates/infrastructure/transport-axum/src/routes.rs
• crates/infrastructure/transport-axum/tests/api_key_routes.rs
• crates/infrastructure/transport-axum/tests/format_translation_integration.rs

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create PR with unit tests
• Commit unit tests in branch feat/allowed-models-providers

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

[socket-security]

Review the following changes in direct dependencies. Learn more about Socket for GitHub.

Diff	Package	Supply Chain Security	Vulnerability	Quality	Maintenance	License
	cargo/​hex@​0.4.3

View full report