Skip to content

Commit

Permalink
feat: Add private key params to Snowflake config
Browse files Browse the repository at this point in the history
Signed-off-by: Arijeet Chakrabarty <arijeet995@gmail.com>
  • Loading branch information
ArijeetC committed May 17, 2024
1 parent a417ea8 commit 085bf5b
Show file tree
Hide file tree
Showing 4 changed files with 24 additions and 0 deletions.
6 changes: 6 additions & 0 deletions sdk/python/feast/infra/materialization/snowflake_engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 """

Expand Down
6 changes: 6 additions & 0 deletions sdk/python/feast/infra/offline_stores/snowflake.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 """

Expand Down
6 changes: 6 additions & 0 deletions sdk/python/feast/infra/online_stores/snowflake.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 """

Expand Down
6 changes: 6 additions & 0 deletions sdk/python/feast/infra/registry/snowflake.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 """

Expand Down

0 comments on commit 085bf5b

Please sign in to comment.