Snowflake connector

1. Install the SDK

   Node.js

   npm install @scalekit-sdk/node

   Python

   pip install scalekit

   Full SDK reference: Node.js | Python

2. Set your credentials

   Add your Scalekit credentials to your .env file. Find values in app.scalekit.com > Developers > API Credentials.

   .env

   SCALEKIT_ENVIRONMENT_URL=<your-environment-url>
   SCALEKIT_CLIENT_ID=<your-client-id>
   SCALEKIT_CLIENT_SECRET=<your-client-secret>

3. Set up the connector

   Register your Snowflake credentials with Scalekit so it handles the token lifecycle. You do this once per environment.

   Dashboard setup steps

   Register your Scalekit environment with the Snowflake connector so Scalekit handles the authentication flow and token lifecycle for you. The connection name you create will be used to identify and invoke the connection programmatically. You’ll need to create an OAuth Security Integration in your Snowflake account.

   1. Set up auth redirects

      • In Scalekit dashboard, go to AgentKit > Connections > Create Connection.

      • Find Snowflake from the list of providers and click Create. Copy the redirect URI. It looks like https://<SCALEKIT_ENVIRONMENT_URL>/sso/v1/oauth/<CONNECTION_ID>/callback.

        

      • Log into your Snowflake account (Snowsight) and run the following SQL to create an OAuth Security Integration, replacing <redirect_uri> with the URI you copied:

        CREATE OR REPLACE SECURITY INTEGRATION scalekit_oauth
          TYPE = OAUTH
          OAUTH_CLIENT = CUSTOM
          OAUTH_CLIENT_TYPE = 'CONFIDENTIAL'
          OAUTH_REDIRECT_URI = '<redirect_uri>'
          ENABLED = TRUE;

   2. Get client credentials

      • After creating the integration, run the following SQL to retrieve the client credentials:

        SELECT SYSTEM$SHOW_OAUTH_CLIENT_SECRETS('SCALEKIT_OAUTH');

      • This returns a JSON object containing:

        • Client ID — value of OAUTH_CLIENT_ID
        • Client Secret — value of OAUTH_CLIENT_SECRET_2 (or OAUTH_CLIENT_SECRET_1)

   3. Add credentials in Scalekit

      • In Scalekit dashboard, go to AgentKit > Connections and open the connection you created.

      • Enter your credentials:

        • Client ID (from the SQL output)
        • Client Secret (from the SQL output)

        

      • Click Save.

4. Authorize and make your first call

   Node.js

   quickstart.ts

   import { ScalekitClient } from '@scalekit-sdk/node'
   import 'dotenv/config'
   
   const scalekit = new ScalekitClient(
     process.env.SCALEKIT_ENV_URL,
     process.env.SCALEKIT_CLIENT_ID,
     process.env.SCALEKIT_CLIENT_SECRET,
   )
   const actions = scalekit.actions
   
   const connector = 'snowflake'
   const identifier = 'user_123'
   
   // Generate an authorization link for the user
   const { link } = await actions.getAuthorizationLink({ connectionName: connector, identifier })
   console.log('Authorize Snowflake:', link)
   process.stdout.write('Press Enter after authorizing...')
   await new Promise(r => process.stdin.once('data', r))
   
   // Make your first call
   const result = await actions.executeTool({
     connector,
     identifier,
     toolName: 'snowflake_cancel_query',
     toolInput: { statement_handle: 'YOUR_STATEMENT_HANDLE' },
   })
   console.log(result)

   Python

   quickstart.py

   import os
   from scalekit.client import ScalekitClient
   from dotenv import load_dotenv
   load_dotenv()
   
   scalekit_client = ScalekitClient(
       env_url=os.getenv("SCALEKIT_ENV_URL"),
       client_id=os.getenv("SCALEKIT_CLIENT_ID"),
       client_secret=os.getenv("SCALEKIT_CLIENT_SECRET"),
   )
   actions = scalekit_client.actions
   
   connection_name = "snowflake"
   identifier = "user_123"
   
   # Generate an authorization link for the user
   link_response = actions.get_authorization_link(
       connection_name=connection_name,
       identifier=identifier,
   )
   print("Authorize Snowflake:", link_response.link)
   input("Press Enter after authorizing...")
   
   # Make your first call
   result = actions.execute_tool(
       tool_input={"statement_handle":"YOUR_STATEMENT_HANDLE"},
       tool_name="snowflake_cancel_query",
       connection_name=connection_name,
       identifier=identifier,
   )
   print(result)

What you can do

Connect this agent connector to let your agent:

• Grants show — Run SHOW GRANTS in common modes (to role, to user, of role, on object)
• Warehouses show — Run SHOW WAREHOUSES
• Schemas show databases — Run SHOW DATABASES or SHOW SCHEMAS
• Keys show imported exported, show primary — Run SHOW IMPORTED KEYS or SHOW EXPORTED KEYS for a table
• Get referential constraints, table constraints, schemata — Query INFORMATION_SCHEMA.REFERENTIAL_CONSTRAINTS
• Query cancel — Cancel a running Snowflake SQL API statement by statement handle

Common workflows

Don’t worry about your Snowflake account domain in the path. Scalekit automatically resolves {{domain}} from the connected account’s configuration. For example, a request with path="/api/v2/statements" will be sent to https://myorg-myaccount.snowflakecomputing.com/api/v2/statements automatically.

Proxy API call

Node.js

const result = await actions.request({
  connectionName: 'snowflake',
  identifier: 'user_123',
  path: '/api/v2/statements',
  method: 'POST',
});
console.log(result);

Python

result = actions.request(
    connection_name='snowflake',
    identifier='user_123',
    path="/api/v2/statements",
    method="POST"
)
print(result)

Execute a tool

Node.js

const result = await actions.executeTool({
  connector: 'snowflake',
  identifier: 'user_123',
  toolName: 'snowflake_cancel_query',
  toolInput: {},
});
console.log(result);

Python

result = actions.execute_tool(
    tool_input={},
    tool_name='snowflake_cancel_query',
    connection_name='snowflake',
    identifier='user_123',
)
print(result)

Tool list

Use the exact tool names from the Tool list below when you call execute_tool. If you’re not sure which name to use, list the tools available for the current user first.

snowflake_cancel_query#Cancel a running Snowflake SQL API statement by statement handle.2 params▾

Cancel a running Snowflake SQL API statement by statement handle.

NameTypeRequiredDescription

statement_handlestringrequiredSnowflake statement handle to cancel

request_idstringoptionalOptional request ID used when the statement was submitted

snowflake_execute_query#Execute one or more SQL statements against Snowflake using the SQL API. Requires a valid Snowflake OAuth2 connection. Use semicolons to submit multiple statements.12 params▾

Execute one or more SQL statements against Snowflake using the SQL API. Requires a valid Snowflake OAuth2 connection. Use semicolons to submit multiple statements.

NameTypeRequiredDescription

statementstringrequiredSQL statement to execute. Use semicolons to send multiple statements in one request.

asyncbooleanoptionalExecute statement asynchronously and return a statement handle

bindingsobjectoptionalBind variables object for '?' placeholders in the SQL statement

databasestringoptionalDatabase to use when executing the statement

nullablebooleanoptionalWhen false, SQL NULL values are returned as the string "null"

parametersobjectoptionalStatement-level Snowflake parameters as a JSON object

request_idstringoptionalUnique request identifier (UUID) used for idempotent retries

retrybooleanoptionalSet true when resubmitting a previously sent request with the same request_id

rolestringoptionalRole to use when executing the statement

schemastringoptionalSchema to use when executing the statement

timeoutintegeroptionalMaximum number of seconds to wait for statement execution

warehousestringoptionalWarehouse to use when executing the statement

snowflake_get_columns#Query INFORMATION_SCHEMA.COLUMNS for column metadata.7 params▾

Query INFORMATION_SCHEMA.COLUMNS for column metadata.

NameTypeRequiredDescription

databasestringrequiredDatabase name

column_name_likestringoptionalOptional column name pattern

limitintegeroptionalMaximum rows

rolestringoptionalOptional role

schemastringoptionalOptional schema filter

tablestringoptionalOptional table filter

warehousestringoptionalOptional warehouse

snowflake_get_query_partition#Get a specific result partition for a Snowflake SQL API statement.3 params▾

Get a specific result partition for a Snowflake SQL API statement.

NameTypeRequiredDescription

partitionintegerrequiredPartition index to fetch (0-based)

statement_handlestringrequiredSnowflake statement handle returned by Execute Query

request_idstringoptionalOptional request ID used when the statement was submitted

snowflake_get_query_status#Get Snowflake SQL API statement status and first partition result metadata by statement handle.2 params▾

Get Snowflake SQL API statement status and first partition result metadata by statement handle.

NameTypeRequiredDescription

statement_handlestringrequiredSnowflake statement handle returned by Execute Query

request_idstringoptionalOptional request ID used when the statement was submitted

snowflake_get_referential_constraints#Query INFORMATION_SCHEMA.REFERENTIAL_CONSTRAINTS.6 params▾

Query INFORMATION_SCHEMA.REFERENTIAL_CONSTRAINTS.

NameTypeRequiredDescription

databasestringrequiredDatabase name

limitintegeroptionalMaximum rows

rolestringoptionalOptional role

schemastringoptionalOptional schema filter

tablestringoptionalOptional table filter

warehousestringoptionalOptional warehouse

snowflake_get_schemata#Query INFORMATION_SCHEMA.SCHEMATA for schema metadata.5 params▾

Query INFORMATION_SCHEMA.SCHEMATA for schema metadata.

NameTypeRequiredDescription

databasestringrequiredDatabase name

limitintegeroptionalMaximum rows

rolestringoptionalOptional role

schema_likestringoptionalOptional schema pattern

warehousestringoptionalOptional warehouse

snowflake_get_table_constraints#Query INFORMATION_SCHEMA.TABLE_CONSTRAINTS.7 params▾

Query INFORMATION_SCHEMA.TABLE_CONSTRAINTS.

NameTypeRequiredDescription

databasestringrequiredDatabase name

constraint_typestringoptionalOptional constraint type filter

limitintegeroptionalMaximum rows

rolestringoptionalOptional role

schemastringoptionalOptional schema filter

tablestringoptionalOptional table filter

warehousestringoptionalOptional warehouse

snowflake_get_tables#Query INFORMATION_SCHEMA.TABLES for table metadata in a Snowflake database.6 params▾

Query INFORMATION_SCHEMA.TABLES for table metadata in a Snowflake database.

NameTypeRequiredDescription

databasestringrequiredDatabase name

limitintegeroptionalMaximum number of rows

rolestringoptionalOptional role

schemastringoptionalOptional schema filter

table_name_likestringoptionalOptional table name pattern

warehousestringoptionalOptional warehouse

snowflake_show_databases_schemas#Run SHOW DATABASES or SHOW SCHEMAS.5 params▾

Run SHOW DATABASES or SHOW SCHEMAS.

NameTypeRequiredDescription

object_typestringrequiredObject type to show

database_namestringoptionalOptional database scope for SHOW SCHEMAS

like_patternstringoptionalOptional LIKE pattern

rolestringoptionalOptional role

warehousestringoptionalOptional warehouse

snowflake_show_grants#Run SHOW GRANTS in common modes (to role, to user, of role, on object).7 params▾

Run SHOW GRANTS in common modes (to role, to user, of role, on object).

NameTypeRequiredDescription

grant_viewstringrequiredSHOW GRANTS variant

object_namestringoptionalObject name for on_object

object_typestringoptionalObject type for on_object

rolestringoptionalOptional execution role

role_namestringoptionalRole name (for to_role/of_role)

user_namestringoptionalUser name (for to_user)

warehousestringoptionalOptional warehouse

snowflake_show_imported_exported_keys#Run SHOW IMPORTED KEYS or SHOW EXPORTED KEYS for a table. For reliable execution in this environment, use fully-qualified scope (database_name + schema_name + table_name).6 params▾

Run SHOW IMPORTED KEYS or SHOW EXPORTED KEYS for a table. For reliable execution in this environment, use fully-qualified scope (database_name + schema_name + table_name).

NameTypeRequiredDescription

key_directionstringrequiredWhich command to run

table_namestringrequiredTable name (use with schema_name and database_name for fully-qualified scope)

database_namestringoptionalOptional database name (recommended with schema_name)

rolestringoptionalOptional role

schema_namestringoptionalOptional schema name (recommended with database_name)

warehousestringoptionalOptional warehouse

snowflake_show_primary_keys#Run SHOW PRIMARY KEYS with optional scope. When using schema_name (or schema_name + table_name), database_name is required for fully-qualified scope.5 params▾

Run SHOW PRIMARY KEYS with optional scope. When using schema_name (or schema_name + table_name), database_name is required for fully-qualified scope.

NameTypeRequiredDescription

database_namestringoptionalOptional database name for scope (required when schema_name is set)

rolestringoptionalOptional role

schema_namestringoptionalOptional schema name for scope

table_namestringoptionalOptional table name for scope

warehousestringoptionalOptional warehouse

snowflake_show_warehouses#Run SHOW WAREHOUSES.3 params▾

Run SHOW WAREHOUSES.

NameTypeRequiredDescription

like_patternstringoptionalOptional LIKE pattern

rolestringoptionalOptional role

warehousestringoptionalOptional warehouse