yggdrasil.databricks.ai.service

service

Databricks AI umbrella service.

Holds the AI-shaped sub-services (vector search today; model serving and registered models are on the roadmap — see ygg-mlops). Reach each via client.ai.<service>::

client.ai.vector_search          # vector search endpoints + indexes
client.ai.vector_search.endpoint("rag").ensure_created()
client.ai.vector_search.index("main.rag.docs").query(query_text="…", columns=["id", "text"])

DatabricksAI

DatabricksAI(client=None)

Bases: DatabricksService

Umbrella for Databricks AI sub-services on a single client.

vector_search property

vector_search: 'VectorSearch'

Vector search service (lazy + cached on this :class:DatabricksAI).

wheels property

wheels: 'Wheels'

Wheel registry service (shorthand for client.wheels).

environments property

environments: 'Environments'

Base-environment service (shorthand for client.environments).

tables property

tables: 'Tables'

Collection-level Unity Catalog table service (shorthand for client.tables).

views property

views: 'Tables'

Alias for :attr:tables — :class:Table covers both managed/external tables and view-shaped securables.

catalogs property

catalogs: 'Catalogs'

Collection-level Unity Catalog hierarchy service (shorthand for client.catalogs).

schemas property

schemas: 'Schemas'

Collection-level Unity Catalog schema service (shorthand for client.schemas).

volumes property

volumes: 'Volumes'

Collection-level Unity Catalog volume service (shorthand for client.volumes).

genie property

genie: 'Genie'

Genie service (shorthand for client.genie).

ai property

ai: 'DatabricksAI'

Databricks AI umbrella service (shorthand for client.ai).

default_tags

default_tags(update: bool = True) -> dict[str, str]

Return default resource tags for Databricks assets.

Returns:

Type	Description
dict[str, str]	A dict of default tags.