nanobot: 2026-09-14 20:57:58

skills/usage/scripts/ollama_usage.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+"""Ollama Cloud usage — GET https://ollama.com/api/usage with Bearer key."""
+
+from __future__ import annotations
+
+import json
+import os
+import sys
+import urllib.error
+import urllib.request
+from pathlib import Path
+
+API_URL = "https://ollama.com/api/usage"
+
+
+def load_api_key() -> str:
+    env = os.environ.get("OLLAMA_API_KEY")
+    if env:
+        return env.strip()
+    # workspace root = three levels above this script
+    env_file = Path(__file__).resolve().parents[3] / ".env"
+    if env_file.is_file():
+        for line in env_file.read_text().splitlines():
+            line = line.strip()
+            if line.startswith("OLLAMA_API_KEY="):
+                return line.split("=", 1)[1].strip().strip('"').strip("'")
+    sys.exit("OLLAMA_API_KEY nenalezen (ani env, ani workspace/.env)")
+
+
+def fmt_models(models: list[dict]) -> str:
+    if not models:
+        return ""
+    parts = [f'{m["name"]} ({m["request_count"]} req)' for m in models]
+    return ", ".join(parts)
+
+
+def main() -> None:
+    req = urllib.request.Request(API_URL, headers={
+        "Authorization": f"Bearer {load_api_key()}",
+        "Accept": "application/json",
+    })
+    try:
+        with urllib.request.urlopen(req, timeout=15) as resp:
+            data = json.load(resp)
+    except urllib.error.HTTPError as e:
+        sys.exit(f"ollama.com API chyba: HTTP {e.code}")
+    except urllib.error.URLError as e:
+        sys.exit(f"ollama.com nedostupné: {e.reason}")
+
+    limits = data.get("limits", {})
+    session = limits.get("session", {})
+    weekly = limits.get("weekly", {})
+
+    print("Ollama Cloud — spotřeba")
+    s_models = fmt_models(session.get("models", []))
+    print(f'Session: ${session.get("usage", 0):.3f}' + (f" — {s_models}" if s_models else ""))
+    w_models = fmt_models(weekly.get("models", []))
+    print(f'Weekly:  ${weekly.get("usage", 0):.3f}' + (f" — {w_models}" if w_models else ""))
+
+    activity = data.get("activity", {})
+    cost = activity.get("cost")
+    if cost is not None:
+        period = activity.get("period", {})
+        print(
+            f'Aktivita: ${float(cost):.5f}'
+            f' ({period.get("type", "?")} {period.get("starting_at", "")[:10]}'
+            f' → {period.get("ending_at", "")[:10]})'
+        )
+
+
+if __name__ == "__main__":
+    main()