Pydantic AI

# provision.py — run once. No SDK needed; these are plain REST calls.
import os, requests

NAIVE = "https://api.usenaive.ai/v1"
HEAD = {
    "Authorization": f"Bearer {os.environ['NAIVE_API_KEY']}",
    "Content-Type": "application/json",
}

# Control plane: a reusable policy template.
kit = requests.post(f"{NAIVE}/account-kits", headers=HEAD, json={
    "name": "Pro",
    "primitives_config": {
        "cards": {"enabled": True, "requiresApproval": True,
                   "defaults": {"spending_limit_cents": 50000}},
        "vault": {"enabled": True},
    },
    "connections_config": {"mode": "allowlist",
                            "toolkits": ["github", "gmail", "stripe"]},
}).json()

# Provision one of your end-users and assign the kit.
alice = requests.post(f"{NAIVE}/users", headers=HEAD, json={
    "external_id": "user_123",
    "email": "alice@acme.com",
    "account_kit_id": kit["id"],
}).json()

print(alice["id"])  # → set as NAIVE_USER_ID for the agent

# agent.py
import os, requests

NAIVE = "https://api.usenaive.ai/v1"
API_HEAD = {"Authorization": f"Bearer {os.environ['NAIVE_API_KEY']}"}
USER_ID = os.environ["NAIVE_USER_ID"]  # Alice, from provisioning

session = requests.post(
    f"{NAIVE}/users/{USER_ID}/sessions",
    headers=API_HEAD,
    json={"ttl_ms": 15 * 60 * 1000},
).json()

# session["mcp"] = {
#   "url": "https://api.usenaive.ai/mcp/sse/sess_...",
#   "headers": {"Authorization": "Bearer nv_sess_..."},
#   "expires_at": "..."
# }

import asyncio
from pydantic import BaseModel
from pydantic_ai import Agent
from pydantic_ai.mcp import MCPServerSSE


class SetupReport(BaseModel):
    github_connect_url: str | None = None
    card_status: str  # "issued" | "pending_approval" | "failed"
    notes: str


naive = MCPServerSSE(
    url=session["mcp"]["url"],
    headers=session["mcp"]["headers"],  # scoped bearer — never in the URL
)

agent = Agent(
    "openai:gpt-5.1",
    toolsets=[naive],
    output_type=SetupReport,
    instructions=(
        "You are Alice's operations agent. Use the Naive tools to act on her "
        "real account. Report the GitHub connect link and the card's status."
    ),
)


async def main():
    async with agent:  # opens the MCP connection for the whole run
        result = await agent.run(
            "Connect Alice's GitHub, then issue a $50 virtual card called "
            "'Ads budget' for our marketing spend."
        )
    print(result.output)  # → a typed SetupReport


asyncio.run(main())