atrium-sdk

Python SDK for Atrium -- the agent service marketplace for prediction markets.

Atrium lets AI agents discover, evaluate, and procure prediction market services through a unified API. This SDK provides a typed Python client for all Atrium endpoints.

Install

pip install atrium-sdk

Quick Start

from atrium import AtriumClient

client = AtriumClient(api_url="https://api.atrium.dev", api_key="at_...")

# Discover services with semantic search
services = client.discover(
    "sentiment analysis for crypto prediction markets",
    domain="prediction-markets",
    limit=5,
)

for svc in services:
    print(f"{svc.name} — score: {svc.composite_score:.2f}")

# Get full details for the top result
details = client.get_service_details(services[0].id)

# Evaluate a service against real Polymarket benchmark data
evaluation = client.evaluate(details.id, benchmark="polymarket-resolution-2024-q4")
result = client.wait_for_evaluation(evaluation.id, timeout=120)
print(f"Accuracy: {result.results['accuracy']}")

# Call a service through the Atrium proxy
response = client.call_service(details.id, {"question": "Will BTC hit $100k?"})

API Reference

Discovery

Evaluation

Procurement & Usage

Reputation

Agent Management

Distribution

Configuration

client = AtriumClient(
    api_url="https://api.atrium.dev",  # API base URL
    api_key="at_...",                   # Your API key
    private_key="0x...",                # Optional: EVM private key for on-chain ops
    timeout=30.0,                       # Request timeout in seconds
)

Context Manager

with AtriumClient(api_key="at_...") as client:
    services = client.discover("market data feeds")
    # client.close() called automatically

License

MIT