Ai ml api

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