Needle

1
from composio import Composio
2
3
# Replace these with your actual values
4
needle_auth_config_id = "ac_YOUR_NEEDLE_CONFIG_ID" # Auth config ID created above
5
user_id = "0000-0000-0000"  # UUID from database/app
6
7
composio = Composio()
8
9
def authenticate_toolkit(user_id: str, auth_config_id: str):
10
    # Replace this with a method to retrieve an API key from the user.
11
    # Or supply your own.
12
    user_api_key = input("[!] Enter API key")
13
14
    connection_request = composio.connected_accounts.initiate(
15
        user_id=user_id,
16
        auth_config_id=auth_config_id,
17
        config={"auth_scheme": "API_KEY", "val": {"generic_api_key": user_api_key}}
18
    )
19
20
    # API Key authentication is immediate - no redirect needed
21
    print(f"Successfully connected Needle for user {user_id}")
22
    print(f"Connection status: {connection_request.status}")
23
    
24
    return connection_request.id
25
26
27
connection_id = authenticate_toolkit(user_id, needle_auth_config_id)
28
29
# You can verify the connection using:
30
connected_account = composio.connected_accounts.get(connection_id)
31
print(f"Connected account: {connected_account}")