diff --git a/sdk/python/feast/infra/materialization/snowflake_engine.py b/sdk/python/feast/infra/materialization/snowflake_engine.py index 4a81982dcd..f77239398e 100644 --- a/sdk/python/feast/infra/materialization/snowflake_engine.py +++ b/sdk/python/feast/infra/materialization/snowflake_engine.py @@ -67,6 +67,12 @@ class SnowflakeMaterializationEngineConfig(FeastConfigBaseModel): authenticator: Optional[str] = None """ Snowflake authenticator name """ + private_key: Optional[str] = None + """ Snowflake private key file path""" + + private_key_passphrase: Optional[str] = None + """ Snowflake private key file passphrase""" + database: StrictStr """ Snowflake database name """ diff --git a/sdk/python/feast/infra/offline_stores/snowflake.py b/sdk/python/feast/infra/offline_stores/snowflake.py index cc59804467..64c84318c9 100644 --- a/sdk/python/feast/infra/offline_stores/snowflake.py +++ b/sdk/python/feast/infra/offline_stores/snowflake.py @@ -105,6 +105,12 @@ class SnowflakeOfflineStoreConfig(FeastConfigBaseModel): authenticator: Optional[str] = None """ Snowflake authenticator name """ + private_key: Optional[str] = None + """ Snowflake private key file path""" + + private_key_passphrase: Optional[str] = None + """ Snowflake private key file passphrase""" + database: StrictStr """ Snowflake database name """ diff --git a/sdk/python/feast/infra/online_stores/snowflake.py b/sdk/python/feast/infra/online_stores/snowflake.py index f5600249c9..57e3bbbb8d 100644 --- a/sdk/python/feast/infra/online_stores/snowflake.py +++ b/sdk/python/feast/infra/online_stores/snowflake.py @@ -51,6 +51,12 @@ class SnowflakeOnlineStoreConfig(FeastConfigBaseModel): authenticator: Optional[str] = None """ Snowflake authenticator name """ + private_key: Optional[str] = None + """ Snowflake private key file path""" + + private_key_passphrase: Optional[str] = None + """ Snowflake private key file passphrase""" + database: StrictStr """ Snowflake database name """ diff --git a/sdk/python/feast/infra/registry/snowflake.py b/sdk/python/feast/infra/registry/snowflake.py index 326d2e0226..169c8ae43e 100644 --- a/sdk/python/feast/infra/registry/snowflake.py +++ b/sdk/python/feast/infra/registry/snowflake.py @@ -93,6 +93,12 @@ class SnowflakeRegistryConfig(RegistryConfig): authenticator: Optional[str] = None """ Snowflake authenticator name """ + private_key: Optional[str] = None + """ Snowflake private key file path""" + + private_key_passphrase: Optional[str] = None + """ Snowflake private key file passphrase""" + database: StrictStr """ Snowflake database name """