Upload folder using huggingface_hub

app/api/routes_governance.py

@@ -307,7 +307,7 @@ async def evaluate_intent_endpoint(
 
     record = UsageRecord(
         api_key=api_key,
-        tier=None,
+        tier=quota["tier"],
         timestamp=start_time,
         endpoint="/api/v1/intents/evaluate",
         request_body=intent_req.model_dump(),
@@ -661,7 +661,7 @@ async def evaluate_healing_decision_endpoint(
 
     record = UsageRecord(
         api_key=api_key,
-        tier=None,
+        tier=quota["tier"],
         timestamp=start_time,
         endpoint="/api/v1/healing/evaluate",
         request_body=decision_req.model_dump(),

tests/test_governance.py

@@ -1,7 +1,11 @@
 """
 Tests for governance endpoints: /api/v1/intents/evaluate
 """
+import tempfile
+
 import pytest
+import app.core.usage_tracker as usage_tracker_module
+from app.core.usage_tracker import UsageTracker
 from app.database.models_intents import TenantDB
 
 
@@ -113,3 +117,36 @@ def test_evaluate_with_criticality(client):
     # context_hash is computed by the governance loop (a 64‑char hex string)
     ctx_hash = healing.get("context_hash")
     assert isinstance(ctx_hash, str) and len(ctx_hash) == 64
+
+
+def test_evaluate_intent_against_real_tracker_does_not_crash(client, monkeypatch):
+    """Every other test in this file runs against tests/conftest.py's
+    MockTracker, whose consume_quota_and_log ignores record.tier entirely
+    and always returns (True, None) -- so none of them would have noticed
+    that this endpoint hardcoded tier=None instead of using quota["tier"]
+    (already resolved by the enforce_quota dependency). Against the real
+    UsageTracker, consume_quota_and_log evaluates tier.monthly_evaluation_limit
+    unconditionally, so a None tier raised AttributeError on every real
+    call, outside any try/except, surfacing as a raw 500. This test swaps
+    in a real UsageTracker (a throwaway SQLite file; the same real
+    Postgres the CI service provides backs api_keys/monthly_counts) to
+    prove the endpoint no longer crashes."""
+    with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
+        monkeypatch.setattr(usage_tracker_module, "tracker", UsageTracker(db_path=tmp.name))
+
+        payload = {
+            "intent_type": "provision_resource",
+            "environment": "prod",
+            "resource_type": "database",
+            "region": "eastus",
+            "size": "Standard",
+            "estimated_cost": 1200,
+            "policy_violations": [],
+            "requester": "alice",
+            "provenance": {},
+            "configuration": {}
+        }
+        response = client.post("/api/v1/intents/evaluate", json=payload,
+                               headers={"X-Tenant-ID": "test-tenant"})
+        assert response.status_code == 200, response.text
+        assert "risk_score" in response.json()

tests/test_healing_endpoint.py

@@ -1,5 +1,9 @@
+import tempfile
+
 from fastapi.testclient import TestClient
 from app.main import app
+import app.core.usage_tracker as usage_tracker_module
+from app.core.usage_tracker import UsageTracker
 
 client = TestClient(app)
 
@@ -18,3 +22,27 @@ def test_healing_evaluate_endpoint():
     response = client.post("/api/v1/healing/evaluate", json=payload,
                            headers={"X-Tenant-ID": "test-tenant"})
     assert response.status_code == 200, f"Expected 200, got {response.status_code}: {response.text}"
+
+
+def test_healing_evaluate_against_real_tracker_does_not_crash(monkeypatch):
+    """Same regression as test_governance.py's equivalent test, for this
+    endpoint's own hardcoded tier=None (routes_governance.py's
+    /healing/evaluate handler) -- see that test's docstring for the full
+    explanation. Swaps in a real UsageTracker to prove
+    consume_quota_and_log no longer crashes on a None tier here either."""
+    payload = {
+        "event": {
+            "component": "my-service",
+            "latency_p99": 450.0,
+            "error_rate": 0.25,
+            "service_mesh": "default",
+            "cpu_util": 0.85,
+            "memory_util": 0.90
+        }
+    }
+    with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
+        monkeypatch.setattr(usage_tracker_module, "tracker", UsageTracker(db_path=tmp.name))
+
+        response = client.post("/api/v1/healing/evaluate", json=payload,
+                               headers={"X-Tenant-ID": "test-tenant"})
+        assert response.status_code == 200, f"Expected 200, got {response.status_code}: {response.text}"