Snowflake

1
from composio import Composio
2
from openai import OpenAI
3
import json
4
5
openai = OpenAI()
6
composio = Composio()
7
8
# User ID must be a valid UUID format
9
user_id = "0000-0000-0000"  # Replace with actual user UUID from your database
10
11
tools = composio.tools.get(user_id=user_id, toolkits=["SNOWFLAKE"])
12
13
print("[!] Tools:")
14
print(json.dumps(tools))
15
16
def 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
33
invoke_llm()