metr-sdk 0.1.0__tar.gz

metr_sdk-0.1.0/PKG-INFO

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: metr-sdk
+Version: 0.1.0
+Summary: One line of code to monetize any FastAPI endpoint
+License: MIT
+Keywords: metr,api,monetization,metering,billing,stripe,fastapi
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.25.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21; extra == "dev"
+Requires-Dist: fastapi>=0.104.0; extra == "dev"
+Requires-Dist: uvicorn>=0.24.0; extra == "dev"

metr_sdk-0.1.0/metr/__init__.py

@@ -0,0 +1,16 @@
+"""
+metr - One line of code to monetize any FastAPI endpoint.
+
+Usage:
+    from metr import require_payment
+
+    @app.post("/api/summarize")
+    @require_payment(price=0.001)
+    async def summarize():
+        return {"summary": "Your text summary..."}
+"""
+
+from .middleware import require_payment, MetrConfig, MetrPaymentInfo
+
+__all__ = ["require_payment", "MetrConfig", "MetrPaymentInfo"]
+__version__ = "0.1.0"

metr_sdk-0.1.0/metr/middleware.py

@@ -0,0 +1,149 @@
+"""metr FastAPI middleware - one decorator to monetize any endpoint."""
+
+import os
+from dataclasses import dataclass
+from functools import wraps
+from typing import Optional
+
+import httpx
+from fastapi import HTTPException, Request
+from fastapi.responses import JSONResponse
+
+
+DEFAULT_GATEWAY_URL = "https://api.metr.dev"
+
+
+@dataclass
+class MetrConfig:
+    """Configuration for metr middleware."""
+    price: float
+    unit_type: str = "request"
+    api_key: Optional[str] = None
+    endpoint_id: Optional[str] = None
+    gateway_url: Optional[str] = None
+
+
+@dataclass
+class MetrPaymentInfo:
+    """Payment info attached to verified requests."""
+    session_token: str
+    buyer_id: str
+    balance_usd: float
+
+
+def require_payment(
+    price: float,
+    unit_type: str = "request",
+    api_key: Optional[str] = None,
+    endpoint_id: Optional[str] = None,
+    gateway_url: Optional[str] = None,
+):
+    """
+    Decorator that requires payment for a FastAPI endpoint.
+
+    Usage:
+        @app.post("/api/summarize")
+        @require_payment(price=0.001)
+        async def summarize(request: Request):
+            payment = request.state.metr  # MetrPaymentInfo
+            return {"summary": "..."}
+    """
+    _api_key = api_key or os.environ.get("METR_API_KEY")
+    _endpoint_id = endpoint_id or os.environ.get("METR_ENDPOINT_ID")
+    _gateway_url = gateway_url or os.environ.get("METR_GATEWAY_URL", DEFAULT_GATEWAY_URL)
+
+    if not _api_key:
+        raise ValueError("metr: API key required. Set api_key or METR_API_KEY env var.")
+
+    def decorator(func):
+        @wraps(func)
+        async def wrapper(*args, **kwargs):
+            # Find the Request object in args/kwargs
+            request = kwargs.get("request")
+            if request is None:
+                for arg in args:
+                    if isinstance(arg, Request):
+                        request = arg
+                        break
+
+            if request is None:
+                raise HTTPException(status_code=500, detail="metr: Request object not found")
+
+            # Extract session token
+            auth_header = request.headers.get("authorization", "")
+            session_token = None
+            if auth_header.startswith("Bearer metr_sess_"):
+                session_token = auth_header[7:]  # Strip "Bearer "
+
+            if not session_token:
+                # No session — create checkout
+                async with httpx.AsyncClient() as client:
+                    checkout_res = await client.post(
+                        f"{_gateway_url}/api/v1/checkout/create",
+                        headers={"X-Metr-Api-Key": _api_key},
+                        json={
+                            "endpoint_id": _endpoint_id,
+                            "amount_usd": price,
+                        },
+                    )
+                    checkout = checkout_res.json()
+
+                return JSONResponse(
+                    status_code=402,
+                    content={
+                        "error": "payment_required",
+                        "message": "This API requires payment. Complete checkout to get access.",
+                        "checkout_url": checkout.get("checkout_url"),
+                        "price": price,
+                        "unit_type": unit_type,
+                    },
+                )
+
+            # Verify session
+            async with httpx.AsyncClient() as client:
+                verify_res = await client.post(
+                    f"{_gateway_url}/api/v1/verify",
+                    headers={"X-Metr-Api-Key": _api_key},
+                    json={"session_token": session_token},
+                )
+                session = verify_res.json()
+
+            if not session.get("valid"):
+                return JSONResponse(
+                    status_code=402,
+                    content={
+                        "error": "session_expired",
+                        "message": "Your billing session has expired.",
+                    },
+                )
+
+            # Attach payment info to request
+            request.state.metr = MetrPaymentInfo(
+                session_token=session_token,
+                buyer_id=session.get("buyer_id", ""),
+                balance_usd=session.get("remaining_balance_usd", 0),
+            )
+
+            # Execute handler
+            result = await func(*args, **kwargs)
+
+            # Record usage (fire and forget)
+            try:
+                async with httpx.AsyncClient() as client:
+                    await client.post(
+                        f"{_gateway_url}/api/v1/meter",
+                        headers={"X-Metr-Api-Key": _api_key},
+                        json={
+                            "endpoint_id": _endpoint_id,
+                            "session_token": session_token,
+                            "units": 1,
+                            "unit_type": unit_type,
+                        },
+                    )
+            except Exception:
+                pass  # Don't fail the request if metering fails
+
+            return result
+
+        return wrapper
+    return decorator

metr_sdk-0.1.0/metr_sdk.egg-info/PKG-INFO

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: metr-sdk
+Version: 0.1.0
+Summary: One line of code to monetize any FastAPI endpoint
+License: MIT
+Keywords: metr,api,monetization,metering,billing,stripe,fastapi
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.25.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21; extra == "dev"
+Requires-Dist: fastapi>=0.104.0; extra == "dev"
+Requires-Dist: uvicorn>=0.24.0; extra == "dev"

metr_sdk-0.1.0/metr_sdk.egg-info/SOURCES.txt

@@ -0,0 +1,8 @@
+pyproject.toml
+metr/__init__.py
+metr/middleware.py
+metr_sdk.egg-info/PKG-INFO
+metr_sdk.egg-info/SOURCES.txt
+metr_sdk.egg-info/dependency_links.txt
+metr_sdk.egg-info/requires.txt
+metr_sdk.egg-info/top_level.txt

metr_sdk-0.1.0/metr_sdk.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

metr_sdk-0.1.0/metr_sdk.egg-info/requires.txt

@@ -0,0 +1,7 @@
+httpx>=0.25.0
+
+[dev]
+pytest>=7.0
+pytest-asyncio>=0.21
+fastapi>=0.104.0
+uvicorn>=0.24.0

metr_sdk-0.1.0/metr_sdk.egg-info/top_level.txt

@@ -0,0 +1 @@
+metr

metr_sdk-0.1.0/pyproject.toml

@@ -0,0 +1,23 @@
+[build-system]
+requires = ["setuptools>=68.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "metr-sdk"
+version = "0.1.0"
+description = "One line of code to monetize any FastAPI endpoint"
+readme = "README.md"
+license = {text = "MIT"}
+requires-python = ">=3.9"
+keywords = ["metr", "api", "monetization", "metering", "billing", "stripe", "fastapi"]
+dependencies = [
+    "httpx>=0.25.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0",
+    "pytest-asyncio>=0.21",
+    "fastapi>=0.104.0",
+    "uvicorn>=0.24.0",
+]

metr_sdk-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+