wealthschema (Python)

Official Python SDK for WealthSchema — the data layer for building and testing financial software. Zero dependencies (standard library only).

It wraps WealthSchema's public, free surfaces:

• Rule Sets — cited, current U.S. planning figures (IRS, SSA, CMS, Treasury) with issuer, document, section, and effective date.
• Evaluation samples — one deliberately-insufficient synthetic household per archetype.
• Free Planning Benchmark — tasks with ground-truth answers for evaluating AI financial-advisor agents.
• Catalog — the product catalog.

The full synthetic-household corpus is a one-time purchase, delivered as files after checkout — not a metered API. This SDK intentionally has no method to download it. Samples are banded and truncated for evaluation only.

Install

pip install wealthschema

Quickstart

from wealthschema import WealthSchema

ws = WealthSchema()  # or WealthSchema(api_key="...") ; falls back to WEALTHSCHEMA_API_KEY

# A cited planning figure, with its source and effective date:
fact = ws.get_fact("us-federal-2026", "retirement.401k.elective_deferral.under_50")
print(fact["value"], fact["source"])  # 24500 {'issuer': 'IRS', 'document': 'Notice 2025-67', ...}

# A sample synthetic household (evaluation only):
sample = ws.get_sample_household("young-family-first-home")

# The free Planning Benchmark for agent evals:
bench = ws.get_benchmark(withhold_answers=True)  # omit for the answer key

Get a free API key at https://www.wealthschema.com/account/api-keys. The free tier works without a key.

API

Requests raise WealthSchemaError with .status and .code (e.g. code="rate_limited" on HTTP 429).

MCP

Rule Sets is also an MCP server:

{ "mcpServers": { "wealthschema": { "url": "https://www.wealthschema.com/api/mcp" } } }

License

MIT. Data and API use are governed by the feed terms and per-purchase license.