From 085bf5b61b009a9cb83bf99b977b34293300c281 Mon Sep 17 00:00:00 2001 From: Arijeet Chakrabarty Date: Thu, 16 May 2024 13:46:53 +0200 Subject: [PATCH] feat: Add private key params to Snowflake config Signed-off-by: Arijeet Chakrabarty --- sdk/python/feast/infra/materialization/snowflake_engine.py | 6 ++++++ sdk/python/feast/infra/offline_stores/snowflake.py | 6 ++++++ sdk/python/feast/infra/online_stores/snowflake.py | 6 ++++++ sdk/python/feast/infra/registry/snowflake.py | 6 ++++++ 4 files changed, 24 insertions(+) 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 """