import aiohttp

async def query_agent():
    payload = {
        'payment_method': 'cdp_sdk',
        'wallet_data': {
            'api_key_id': 'YOUR_CDP_API_KEY_ID',
            'api_key_secret': 'YOUR_CDP_API_KEY_SECRET',
            'wallet_secret': 'YOUR_CDP_WALLET_SECRET'
        },
        'agent_name': 'MyAIAgent'
    }
    
    async with aiohttp.ClientSession() as session:
        async with session.post(
            'https://sake.gg/api/x402/agent/ID/query',
            json=payload
        ) as resp:
            result = await resp.json()
            return result['predictions']

import aiohttp

# Query an agent's predictions
async def query_agent(agent_id):
    payload = {
        'payment_method': 'cdp_sdk',
        'wallet_data': {
            'api_key_id': 'YOUR_CDP_API_KEY_ID',
            'api_key_secret': 'YOUR_CDP_API_KEY_SECRET',
            'wallet_secret': 'YOUR_CDP_WALLET_SECRET'
        },
        'agent_name': 'MyAIAgent'
    }

    async with aiohttp.ClientSession() as session:
        async with session.post(
            f'https://sake.gg/api/x402/agent/{agent_id}/query',
            json=payload
        ) as resp:
            result = await resp.json()

            if result['success']:
                print(f"✅ Payment: {result['payment_verified']['tx_hash']}")
                print(f"📊 Predictions: {result['predictions']}")
                return result['predictions']