Как получить секреты (например, ключи API) в блокнотах Snowflake

У меня была та же проблема с получением объекта секретов имени пользователя и пароля. Мой обходной путь состоял в том, чтобы создать функцию для получения секрета, а затем вызвать ее в моем блокноте. Адаптируйте этот код для использования ключей API.

Сначала создайте интеграцию:

CREATE OR REPLACE EXTERNAL ACCESS INTEGRATION my_external_endpoints
ALLOWED_AUTHENTICATION_SECRETS = ('database.schema.your_secret_string_name')
ENABLED = true;

Затем создайте эту функцию, которая будет получать информацию для вас:

CREATE OR REPLACE FUNCTION get_secret_username_password()
RETURNS STRING
LANGUAGE PYTHON
RUNTIME_VERSION = 3.8
HANDLER = 'get_secret_username_password'
EXTERNAL_ACCESS_INTEGRATIONS = (my_external_endpoints)
SECRETS = ('cred' = your_secret_string_name)
AS
$$
import _snowflake
import json

def get_secret_username_password():
  username_password_object = _snowflake.get_username_password('cred')

  username_password_dictionary = {
      "Username": username_password_object.username,
      "Password": username_password_object.password
  }

  return json.dumps(username_password_dictionary)
$$;

В своем блокноте извлеките информацию:

from snowflake.snowpark import Session
import json

session = get_active_session()

result = session.sql(f"""
    SELECT get_secret_username_password()
""").collect()

creds = json.loads(result[0][0])

username = creds["Username"]
password = creds["Password"]

Хотя @Amito дал правильный ответ, вот немного более короткая версия, если вам просто нужно вернуть строку, например ключ API:

CREATE OR REPLACE FUNCTION retrieve_my_api_key()
RETURNS STRING
LANGUAGE PYTHON
RUNTIME_VERSION = 3.8
HANDLER = 'retrieve_my_api_key'
EXTERNAL_ACCESS_INTEGRATIONS = (my_integration)
SECRETS = ('cred' = my_api_key)
AS
$$
import _snowflake

def retrieve_my_api_key():   
  my_api_key = _snowflake.get_generic_secret_string('cred') 
  return my_api_key
$$;

и в тетради:

from snowflake.snowpark.context import get_active_session
session = get_active_session()
results = session.sql('SELECT retrieve_my_api_key()').collect()
# this returns [Row(RETRIEVE_MY_API_KEY()='sUpErSecRetApiKey')] - why??
secret_my_api_key = results[0][0]