x402-neurobro/examples/06_batch_queries.py at main · axioma-ai-labs/x402-neurobro · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#!/usr/bin/env python3
"""
Batch queries — run many questions sequentially or in parallel.

Each prompt costs $1 USDC. Three prompts = $3 USDC total.
Pass concurrent=True to fire them in parallel instead of one after another.

Requires WALLET_PRIVATE_KEY in .env and enough USDC on Base mainnet
to cover every prompt in the batch.

Usage:
    python 06_batch_queries.py
"""

import asyncio
import sys
from dataclasses import dataclass

from neurobro_client import NeurobroClient, QueryResult


@dataclass
class BatchResult:
    prompt: str
    result: QueryResult | None
    error: str | None


async def run_batch(
    client: NeurobroClient,
    prompts: list[str],
    concurrent: bool = False,
) -> list[BatchResult]:
    """Run `prompts` against the API, sequentially or in parallel."""
    results: list[BatchResult] = []

    if concurrent:
        tasks = [client.query_async(p) for p in prompts]
        responses = await asyncio.gather(*tasks, return_exceptions=True)

        for prompt, response in zip(prompts, responses):
            if isinstance(response, Exception):
                results.append(BatchResult(prompt, None, str(response)))
            else:
                results.append(BatchResult(prompt, response, None))
    else:
        for prompt in prompts:
            try:
                result = await client.query_async(prompt)
                results.append(BatchResult(prompt, result, None))
            except Exception as e:
                results.append(BatchResult(prompt, None, str(e)))

    return results


async def main() -> None:
    try:
        client = NeurobroClient()
    except ValueError as e:
        print(f"Setup error: {e}")
        sys.exit(1)

    print(f"Wallet: {client.wallet_address}")
    print("=" * 60)

    prompts = [
        "What is Bitcoin in one sentence?",
        "What is Ethereum in one sentence?",
        "What is Solana in one sentence?",
    ]

    print("\n[Sequential Queries]")
    print("-" * 40)

    results = await run_batch(client, prompts, concurrent=False)

    for i, batch in enumerate(results, 1):
        print(f"\n{i}. {batch.prompt}")
        if batch.result:
            text = batch.result.text[:150].replace("\n", " ")
            print(f"   → {text}...")
        else:
            print(f"   → Error: {batch.error}")

    print("\n" + "=" * 60)
    success = sum(1 for r in results if r.result)
    print(f"Completed: {success}/{len(prompts)} queries")


if __name__ == "__main__":
    asyncio.run(main())


# Example output:
#
# Wallet: 0xA1b2C3d4E5F67890a1b2c3D4e5f6789012345678
# ============================================================
#
# [Sequential Queries]
# ----------------------------------------
#
# 1. What is Bitcoin in one sentence?
#    → Bitcoin is a decentralized digital currency with a fixed supply of
#      21 million coins, secured by proof-of-work mining on a public...
#
# 2. What is Ethereum in one sentence?
#    → Ethereum is a programmable blockchain whose native asset ETH powers
#      a global network of smart contracts, DeFi, and L2 rollups...
#
# 3. What is Solana in one sentence?
#    → Solana is a high-throughput proof-of-stake blockchain designed for
#      low-latency applications, using a Proof-of-History clock to...
#
# ============================================================
# Completed: 3/3 queries