runtime zmeny

skills/usage/scripts/ollama_usage_poll.py

@@ -11,14 +11,9 @@ import json
 import sqlite3
 import sys
 import urllib.error
-import urllib.request
 from datetime import UTC, datetime
-from pathlib import Path
 
-from ollama_usage import API_URL, load_api_key
-
-WORKSPACE = Path(__file__).resolve().parents[3]
-DB_PATH = WORKSPACE / "db" / "ollama_usage.sqlite"
+from ollama_usage import DB_PATH, fetch_usage
 
 SCHEMA = """
 CREATE TABLE IF NOT EXISTS samples (
@@ -36,15 +31,6 @@ CREATE TABLE IF NOT EXISTS meta (
 """
 
 
-def fetch_usage() -> dict:
-    request = urllib.request.Request(API_URL, headers={
-        "Authorization": f"Bearer {load_api_key()}",
-        "Accept": "application/json",
-    })
-    with urllib.request.urlopen(request, timeout=15) as response:
-        return json.load(response)
-
-
 def canonical_models(models: list[dict]) -> str:
     """Stable JSON for change detection — the API returns an unordered list."""
     return json.dumps(
@@ -72,8 +58,7 @@ def record_poll(conn: sqlite3.Connection, ts: str, status: str) -> None:
 
 def latest_sample(conn: sqlite3.Connection) -> tuple | None:
     row = conn.execute(
-        "SELECT session_usage, weekly_usage, session_models, weekly_models "
-        "FROM samples ORDER BY ts DESC LIMIT 1"
+        "SELECT session_usage, weekly_usage, session_models, weekly_models FROM samples ORDER BY ts DESC LIMIT 1"
     ).fetchone()
     return row