| """OpenRouter model catalog: tool-capable models only (design doc: 'any |
| OpenRouter model' is only true of tool-capable ones).""" |
| from __future__ import annotations |
|
|
| import httpx |
|
|
| |
| |
| PINNED_MODELS = [ |
| {"id": "anthropic/claude-sonnet-4.5", "name": "Claude Sonnet 4.5 (pinned)", |
| "prompt_price": 0.0, "completion_price": 0.0}, |
| {"id": "openai/gpt-4.1-mini", "name": "GPT-4.1 mini (pinned)", |
| "prompt_price": 0.0, "completion_price": 0.0}, |
| ] |
|
|
|
|
| def fetch_key_headroom(base_url: str, api_key: str, |
| transport: httpx.BaseTransport | None = None, |
| ) -> float | None: |
| """Remaining credit on this OpenRouter key (GET /key), or None when |
| unknowable — no key, endpoint error, or an uncapped key. None means |
| "don't warn", never "zero left": the estimate must fail open.""" |
| if not api_key: |
| return None |
| try: |
| with httpx.Client(base_url=base_url, timeout=10.0, transport=transport, |
| headers={"Authorization": f"Bearer {api_key}"} |
| ) as client: |
| resp = client.get("/key") |
| resp.raise_for_status() |
| data = resp.json().get("data") |
| except (httpx.HTTPError, ValueError): |
| return None |
| if not isinstance(data, dict): |
| return None |
| remaining = data.get("limit_remaining") |
| if remaining is None and data.get("limit") is not None: |
| remaining = float(data["limit"]) - float(data.get("usage") or 0.0) |
| return float(remaining) if remaining is not None else None |
|
|
|
|
| def fetch_models(base_url: str, |
| transport: httpx.BaseTransport | None = None) -> list[dict]: |
| with httpx.Client(base_url=base_url, timeout=30.0, |
| transport=transport) as client: |
| resp = client.get("/models") |
| resp.raise_for_status() |
| out = [] |
| for m in resp.json().get("data", []): |
| if "tools" not in (m.get("supported_parameters") or []): |
| continue |
| pricing = m.get("pricing") or {} |
| out.append({"id": m["id"], "name": m.get("name", m["id"]), |
| "prompt_price": float(pricing.get("prompt", 0) or 0), |
| "completion_price": float(pricing.get("completion", 0) or 0)}) |
| return sorted(out, key=lambda m: m["id"]) |
|
|
