Azure CosmosDB Online Store Spec

class databricks.feature_store.online_store_spec.azure_cosmosdb_online_store_spec.AzureCosmosDBSpec(*, account_uri: str, database_name: Optional[str] = None, container_name: Optional[str] = None, read_secret_prefix: Optional[str] = None, write_secret_prefix: str)

Bases: databricks.feature_store.online_store_spec.online_store_spec.OnlineStoreSpec

This OnlineStoreSpec implementation is intended for publishing features to Azure Cosmos DB.

If database_name and container_name are not provided, FeatureStoreClient.publish_table will use the offline store’s database and table name as the Cosmos DB database and container name.

The expected read or write secret for Cosmos DB for a given {prefix} string is ${prefix}-authorization-key.

The authorization key can be either the Cosmos DB account primary or secondary key.

Note

Available starting with Databricks Runtime 11.1 for ML.

Parameters
  • account_uri – URI of the Cosmos DB account.

  • database_name – Database name.

  • container_name – Container name.

  • read_secret_prefix – Prefix for read secret.

  • write_secret_prefix – Prefix for write secret.

Methods

auth_type()

Publish Auth type.

Properties

account_uri

Account URI of the online store.

database_name

Database name.

container_name

Container name.

cloud

Define the cloud property for the data store.

store_type

Define the data store type.

property account_uri

Account URI of the online store.

property database_name

Database name.

property container_name

Container name.

auth_type()

Publish Auth type.