Skip to content

Add direct-mode bundle support for AI Gateway model_provider_service - #6634

Open
alex-khakhlyuk wants to merge 5 commits into
mainfrom
aigov-dabs-model-provider-service
Open

alex-khakhlyuk wants to merge 5 commits into
mainfrom
aigov-dabs-model-provider-service

Conversation

@alex-khakhlyuk

Copy link
Copy Markdown
Member

Third and last of the stacked split of #6410 (stacked on the mcp_service PR). Direct engine only; provider_type is immutable (recreate) and write-only provider secrets + server-computed fields are suppressed to avoid phantom drift.

Changes

Why

Tests

@alex-khakhlyuk
alex-khakhlyuk force-pushed the aigov-dabs-model-provider-service branch from 9a78c32 to eb8fe3b Compare September 11, 2026 13:33
@eng-dev-ecosystem-bot

eng-dev-ecosystem-bot commented Sep 11, 2026

Copy link
Copy Markdown
Collaborator

Integration test report

Commit: ab540af

Run: 34967236156

Env 🟨​KNOWN ✅​pass 🙈​skip Time
🟨​ aws linux 5 277 25 8:07
🟨​ aws windows 5 279 23 8:32
🟨​ azure linux 5 276 25 11:32
🟨​ azure windows 5 278 23 8:54
🟨​ gcp linux 5 277 25 9:49
🟨​ gcp windows 5 279 23 9:25
Test Name aws linux aws windows azure linux azure windows gcp linux gcp windows
🟨​ TestAccept 🟨​K 🟨​K 🟨​K 🟨​K 🟨​K 🟨​K
🟨​ TestAccept/bundle/resources/mcp_services/lifecycle 🟨​K 🟨​K 🟨​K 🟨​K 🟨​K 🟨​K
🟨​ TestAccept/bundle/resources/mcp_services/lifecycle/DATABRICKS_BUNDLE_ENGINE=direct/DMS=true 🟨​K 🟨​K 🟨​K 🟨​K 🟨​K 🟨​K
🟨​ TestAccept/bundle/resources/model_provider_services/lifecycle 🟨​K 🟨​K 🟨​K 🟨​K 🟨​K 🟨​K
🟨​ TestAccept/bundle/resources/model_provider_services/lifecycle/DATABRICKS_BUNDLE_ENGINE=direct/DMS=true 🟨​K 🟨​K 🟨​K 🟨​K 🟨​K 🟨​K

Comment thread bundle/direct/dresources/resources.yml Outdated
# rejects changing it via Update, so a local change must recreate.
- field: config.provider_type
reason: immutable
ignore_remote_changes:

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

see #6654

@alex-khakhlyuk
alex-khakhlyuk force-pushed the aigov-dabs-model-provider-service branch from eb8fe3b to d1948b6 Compare September 14, 2026 12:51
@alex-khakhlyuk
alex-khakhlyuk force-pushed the aigov-dabs-model-provider-service branch 2 times, most recently from 5448793 to 1cd813f Compare September 14, 2026 17:55
@alex-khakhlyuk
alex-khakhlyuk force-pushed the aigov-dabs-model-provider-service branch from 1cd813f to e95c91a Compare September 14, 2026 19:21

@janniklasrose janniklasrose left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

See nits in #6633, they apply here too

@janniklasrose
janniklasrose added this pull request to stack #6683 September 15, 2026 10:47
@alex-khakhlyuk
alex-khakhlyuk force-pushed the aigov-dabs-model-provider-service branch from e95c91a to ab540af Compare September 15, 2026 12:08
@janniklasrose
janniklasrose added this pull request to the merge queue Sep 15, 2026
@alex-khakhlyuk
alex-khakhlyuk removed this pull request from the merge queue due to a manual request Sep 15, 2026
Base automatically changed from aigov-dabs-mcp-service to main September 15, 2026 13:34
@alex-khakhlyuk
alex-khakhlyuk force-pushed the aigov-dabs-model-provider-service branch from ab540af to e960e17 Compare September 15, 2026 13:34
alex-khakhlyuk and others added 5 commits September 15, 2026 13:35
Third and last of the stacked split of #6410 (stacked on the mcp_service PR). Direct engine only; provider_type is immutable (recreate) and write-only provider secrets + server-computed fields are suppressed to avoid phantom drift.

Co-authored-by: Isaac <no-reply@databricks.com>
Follow-up mirroring #6654 (model_services) and the mcp_services cleanup, now
that #6618 enabled generating field behaviors from the response type. Added
catalog.ModelProviderService to apitypes.yml, removed the hand-written
recreate_on_changes (config.provider_type) and ignore_remote_changes
(inference_table + write-only provider credentials) from resources.yml, and
regenerated resources.generated.yml. The generator now derives all of these
from the OpenAPI schema (and additionally picks up config.anthropic.relayed.
plan_type as immutable); no suppression is lost.

Co-authored-by: Isaac <no-reply@databricks.com>
…t_tag)

Main upgraded databricks-sdk-go v0.177 -> v0.178, the GA-sync SDK that dropped
`owner` (effective_owner only now), the request_tag rate-limit feature, and the
anthropic relayed plan_type from the AI Gateway securable types. Adapted the
model_provider_service resource to v0.178: removed the stale Owner field from
modelProviderServiceBody, and regenerated schema/validation/refschema/pydabs.
RATE_LIMIT_KEY_REQUEST_TAG is now absent from model_provider_services (and
model_services / mcp_services), matching the v0.178 spec.

Co-authored-by: Isaac <no-reply@databricks.com>
… configs/ layout

Main #6679 split the monolithic resources.yml / resources.generated.yml into one
file per resource under configs/. Recreate the model_provider_services rules in
that layout: configs/model_provider_services.yml (provided_id_fields for the
immutable parent + model_provider_service_id) and the regenerated
configs/model_provider_services.generated.yml (provider_type recreate + the
write-only provider-credential and inference_table suppressions, derived from
catalog.ModelProviderService via apitypes.yml). No behavior change.

Co-authored-by: Isaac <no-reply@databricks.com>
Addresses jannik's review nit on #6633 (deferred as a follow-up): the helper
resolves the compound schemas/{catalog}.{schema} parent shared by the AI Gateway
securables, so the more specific name makes the signature clear. Renames the
definition and all three call sites (model_services, mcp_services,
model_provider_services). No behavior change.

Co-authored-by: Isaac <no-reply@databricks.com>
@alex-khakhlyuk
alex-khakhlyuk force-pushed the aigov-dabs-model-provider-service branch from e960e17 to 13e7f9a Compare September 15, 2026 13:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants