BrainstormRouter Python SDK

Drop-in replacement for the OpenAI Python SDK with intelligent routing, memory, guardrails, and governance across 247 model endpoints.

Install

pip install brainstormrouter

Quick Start

from brainstormrouter import BrainstormRouter

client = BrainstormRouter(api_key="brk_...")

# OpenAI-compatible — works with LangChain, LlamaIndex, CrewAI, etc.
response = client.chat.completions.create(
    model="anthropic/claude-sonnet-4",
    messages=[{"role": "user", "content": "Hello"}],
)
print(response.choices[0].message.content)

Use "auto" as the model to let BrainstormRouter's Thompson sampling pick the best model:

response = client.chat.completions.create(
    model="auto",
    messages=[{"role": "user", "content": "Explain quantum computing"}],
)

Streaming

stream = client.chat.completions.create(
    model="auto",
    messages=[{"role": "user", "content": "Write a poem"}],
    stream=True,
)
for chunk in stream:
    content = chunk.choices[0].delta.content
    if content:
        print(content, end="")

Memory (4-Block RMM)

# Append a memory entry
client.memory.append("User prefers concise answers", block="semantic")

# List memory entries
result = client.memory.entries()

# List memory blocks with entry counts
result = client.memory.blocks()

# Bootstrap memory from documents
client.memory.init([{"content": "Product requirements doc...", "source": "prd.md"}])

Guardrails

# Test content against guardrail pipeline
result = client.guardrails.test(
    content="Check this message for PII",
    direction="inbound",
)

Provider Keys (BYOK)

# Register your own provider key
client.providers.register("anthropic", api_key="sk-ant-...")

# Validate a key with a live request
result = client.providers.test("anthropic", "sk-ant-...")

Configuration

client = BrainstormRouter(
    api_key="brk_...",                     # or set BRAINSTORMROUTER_API_KEY env var
    base_url="https://api.brainstormrouter.com",  # default
)

Resources

• Documentation
• API Reference
• Full API Reference (llms-full.txt)
• GitHub
• Dashboard

License

MIT