Snowflake

Learn how to use Snowflake with Composio

Overview

SLUG: SNOWFLAKE

Description

Snowflake is a cloud-based data warehouse offering elastic scaling, secure data sharing, and SQL analytics across multiple cloud environments

Authentication Details

client_id
stringRequired
client_secret
stringRequired
oauth_redirect_uri
stringDefaults to https://backend.composio.dev/api/v1/auth-apps/add
scopes
stringDefaults to session:role:PUBLIC
subdomain
stringRequired
username
stringRequired
password
stringRequired
subdomain
stringRequired

Connecting to Snowflake

Create an auth config

Use the dashboard to create an auth config for the Snowflake toolkit. This allows you to connect multiple Snowflake accounts to Composio for agents to use.

1

Select App

Navigate to Snowflake.

2

Configure Auth Config Settings

Select among the supported auth schemes of and configure them here.

3

Create and Get auth config ID

Click “Create Snowflake Auth Config”. After creation, copy the displayed ID starting with ac_. This is your auth config ID. This is not a sensitive ID — you can save it in environment variables or a database. This ID will be used to create connections to the toolkit for a given user.

Connect Your Account

Using OAuth2

1from composio import Composio
2
3# Replace these with your actual values
4snowflake_auth_config_id = "ac_YOUR_SNOWFLAKE_CONFIG_ID" # Auth config ID created above
5user_id = "0000-0000-0000" # UUID from database/application
6
7composio = Composio()
8
9
10def authenticate_toolkit(user_id: str, auth_config_id: str):
11 connection_request = composio.connected_accounts.initiate(
12 user_id=user_id,
13 auth_config_id=auth_config_id,
14 )
15
16 print(
17 f"Visit this URL to authenticate Snowflake: {connection_request.redirect_url}"
18 )
19
20 # This will wait for the auth flow to be completed
21 connection_request.wait_for_connection(timeout=15)
22 return connection_request.id
23
24
25connection_id = authenticate_toolkit(user_id, snowflake_auth_config_id)
26
27# You can also verify the connection status using:
28connected_account = composio.connected_accounts.get(connection_id)
29print(f"Connected account: {connected_account}")

Using Basic Auth

1from composio import Composio
2from composio.types import auth_scheme
3
4# Replace these with your actual values
5snowflake_auth_config_id = "ac_YOUR_SNOWFLAKE_CONFIG_ID"
6user_id = "user@example.com"
7username = "your_snowflake_username"
8password = "your_snowflake_password"
9
10composio = Composio()
11
12# Create a new connected account for Snowflake using Basic Auth
13connection_request = composio.connected_accounts.initiate(
14 user_id=user_id,
15 auth_config_id=snowflake_auth_config_id,
16 config=auth_scheme.basic_auth(
17 username=username,
18 password=password
19 )
20)
21
22# Basic authentication is immediate - no redirect needed
23print(f"Successfully connected Snowflake for user {user_id}")
24
25# You can verify the connection using:
26# connected_account = composio.connected_accounts.get(user_id=user_id, app_id="SNOWFLAKE")

Tools

Executing tools

To prototype you can execute some tools to see the responses and working on the Snowflake toolkit’s playground

Python
1from composio import Composio
2from openai import OpenAI
3import json
4
5openai = OpenAI()
6composio = Composio()
7
8# User ID must be a valid UUID format
9user_id = "0000-0000-0000" # Replace with actual user UUID from your database
10
11tools = composio.tools.get(user_id=user_id, toolkits=["SNOWFLAKE"])
12
13print("[!] Tools:")
14print(json.dumps(tools))
15
16def invoke_llm(task = "What can you do?"):
17 completion = openai.chat.completions.create(
18 model="gpt-4o",
19 messages=[
20 {
21 "role": "user",
22 "content": task, # Your task here!
23 },
24 ],
25 tools=tools,
26 )
27
28 # Handle Result from tool call
29 result = composio.provider.handle_tool_calls(user_id=user_id, response=completion)
30 print(f"[!] Completion: {completion}")
31 print(f"[!] Tool call result: {result}")
32
33invoke_llm()

Tool List

Tool Name: Cancel Statement Execution

Description

Cancels the execution of a running sql statement. use this action to stop a long-running query.

Action Parameters

request_id
string
statementHandle
stringRequired

Action Response

data
objectRequired
error
string
successful
booleanRequired

Tool Name: Check Statement Status

Description

Retrieves the status of a previously submitted sql statement. replace '{statementhandle}' with the unique identifier of the statement.

Action Parameters

partition
integer
requestId
integer
statementHandle
stringRequired

Action Response

code
stringRequired
data
array
error
string
message
stringRequired
statementHandle
stringRequired
statementStatusUrl
stringRequired
successful
booleanRequired

Tool Name: Execute SQL

Description

Tool to execute a sql statement and return the resulting data. use when you need to query data from snowflake.

Action Parameters

bindings
object
database
string
parameters
object
role
string
schema
string
statement
stringRequired
timeout
integer
warehouse
string

Action Response

data
arrayRequired
error
string
successful
booleanRequired

Tool Name: Fetch Catalog Integration

Description

Fetches details of a specific catalog integration.

Action Parameters

name
stringRequired

Action Response

data
objectRequired
error
string
successful
booleanRequired

Tool Name: Get Active Scheduled Maintenances

Description

Retrieves a list of any active scheduled maintenances currently in the in progress or verifying state.

Action Parameters

Action Response

data
objectRequired
error
string
successful
booleanRequired

Tool Name: Get All Scheduled Maintenances

Description

Retrieves a list of the 50 most recent scheduled maintenances, including those in the completed state.

Action Parameters

Action Response

data
objectRequired
error
string
successful
booleanRequired

Tool Name: Get Component Status

Description

Retrieves the status of individual components, each listed with its current status.

Action Parameters

limit
integer

Action Response

data
objectRequired
error
string
successful
booleanRequired

Tool Name: Get Status Rollup

Description

Retrieves the status rollup for the entire page, including indicators and human-readable descriptions of the blended component status.

Action Parameters

Action Response

data
objectRequired
error
string
successful
booleanRequired

Tool Name: Get Status Summary

Description

Retrieves a summary of the status page, including status indicators, component statuses, unresolved incidents, and upcoming or in-progress scheduled maintenances.

Action Parameters

Action Response

data
objectRequired
error
string
successful
booleanRequired

Tool Name: Get Unresolved Incidents

Description

Retrieves a list of any unresolved incidents currently in the investigating, identified, or monitoring state.

Action Parameters

Action Response

data
objectRequired
error
string
successful
booleanRequired

Tool Name: Get Upcoming Scheduled Maintenances

Description

Retrieves a list of any upcoming scheduled maintenances still in the scheduled state.

Action Parameters

test_id
string

Action Response

data
objectRequired
error
string
successful
booleanRequired

Tool Name: Show Databases

Description

Lists all databases for which you have access privileges. shows database metadata including name, creation date, owner, retention time, and more. can filter results and include dropped databases within time travel retention period.

Action Parameters

from_name
string
history
boolean
like_pattern
string
limit
integer
role
string
starts_with
string
terse
boolean
timeout
integer
warehouse
string

Action Response

columns
array
data
arrayRequired
error
string
successful
booleanRequired

Tool Name: Show Schemas

Description

Lists all schemas for which you have access privileges. shows schema metadata including name, creation date, owner, database, retention time, and more. can filter results and include dropped schemas within time travel retention period.

Action Parameters

database
string
from_name
string
history
boolean
in_scope
string
like_pattern
string
limit
integer
role
string
starts_with
string
terse
boolean
timeout
integer
warehouse
string

Action Response

columns
array
data
arrayRequired
error
string
successful
booleanRequired

Tool Name: Show Tables

Description

Lists all tables for which you have access privileges. shows table metadata including name, creation date, owner, database, schema, row count, size in bytes, clustering keys, and more. can filter results and include dropped tables within time travel retention period.

Action Parameters

database
string
from_name
string
history
boolean
in_scope
string
like_pattern
string
limit
integer
role
string
schema
string
starts_with
string
terse
boolean
timeout
integer
warehouse
string

Action Response

columns
array
data
arrayRequired
error
string
successful
booleanRequired

Tool Name: Submit SQL Statement

Description

Submits a sql statement for execution.

Action Parameters

bindings
object
database
string
parameters
object
role
string
schema
string
statement
stringRequired
timeout
integerDefaults to 604800
warehouse
string

Action Response

code
stringRequired
createdOn
integerRequired
data
array
error
string
message
stringRequired
resultSetMetaData
object
sqlState
stringRequired
statementHandle
stringRequired
statementHandles
array
statementStatusUrl
stringRequired
stats
object
successful
booleanRequired