Upload services/governance-gateway/app/api/routes/gateway.py with huggingface_hub

services/governance-gateway/app/api/routes/gateway.py

@@ -1,59 +1,60 @@
-from datetime import UTC, datetime
-
-from fastapi import APIRouter, HTTPException
-
-from app.core.config import get_settings
-from app.schemas.gateway import GatewayAnswerRequest, GatewayAnswerResponse, GatewayMetricsResponse
-from app.services.audit import AuditStore
-from app.services.gateway_answer_store import GatewayAnswerStore
-from app.services.orchestrator import GatewayOrchestrator
-
-
-router = APIRouter(prefix="/gateway", tags=["gateway"])
-_settings = get_settings()
-_shared_audit = AuditStore(_settings.audit_log_path)
-_shared_answer_store = GatewayAnswerStore(
-    database_url=_settings.database_url,
-    enabled=_settings.gateway_answers_persist,
-)
-_orchestrator = GatewayOrchestrator(
-    audit_store=_shared_audit,
-    answer_store=_shared_answer_store,
-)
-
-
-@router.post("/answer", response_model=GatewayAnswerResponse)
-def answer(payload: GatewayAnswerRequest) -> GatewayAnswerResponse:
-    return _orchestrator.answer(payload)
-
-
-@router.get("/audit/{audit_id}")
-def get_audit(audit_id: str) -> dict:
-    record = _shared_audit.get(audit_id)
-    if not record:
-        raise HTTPException(status_code=404, detail=f"Audit record {audit_id} not found")
-    return record
-
-
-@router.get("/history")
-def get_history(limit: int = 20) -> list[dict]:
-    """List recent gateway interactions from Postgres."""
-    return _shared_answer_store.list_history(limit=limit)
-
-
-@router.get("/answers/{answer_id}")
-def get_answer(answer_id: str) -> dict:
-    """Load a stored gateway response from Postgres ``gateway_answers``."""
-    record = _shared_answer_store.get(answer_id)
-    if not record:
-        raise HTTPException(status_code=404, detail=f"Gateway answer {answer_id} not found")
-    return record
-
-
-@router.get("/metrics", response_model=GatewayMetricsResponse)
-def metrics() -> GatewayMetricsResponse:
-    snapshot = _shared_audit.metrics()
-    return GatewayMetricsResponse(
-        generated_at=datetime.now(UTC),
-        **snapshot,
-    )
+from datetime import UTC, datetime
+from typing import Optional
+
+from fastapi import APIRouter, HTTPException
+
+from app.core.config import get_settings
+from app.schemas.gateway import GatewayAnswerRequest, GatewayAnswerResponse, GatewayMetricsResponse
+from app.services.audit import AuditStore
+from app.services.gateway_answer_store import GatewayAnswerStore
+from app.services.orchestrator import GatewayOrchestrator
+
+
+router = APIRouter(prefix="/gateway", tags=["gateway"])
+_settings = get_settings()
+_shared_audit = AuditStore(_settings.audit_log_path)
+_shared_answer_store = GatewayAnswerStore(
+    database_url=_settings.database_url,
+    enabled=_settings.gateway_answers_persist,
+)
+_orchestrator = GatewayOrchestrator(
+    audit_store=_shared_audit,
+    answer_store=_shared_answer_store,
+)
+
+
+@router.post("/answer", response_model=GatewayAnswerResponse)
+def answer(payload: GatewayAnswerRequest) -> GatewayAnswerResponse:
+    return _orchestrator.answer(payload)
+
+
+@router.get("/audit/{audit_id}")
+def get_audit(audit_id: str) -> dict:
+    record = _shared_audit.get(audit_id)
+    if not record:
+        raise HTTPException(status_code=404, detail=f"Audit record {audit_id} not found")
+    return record
+
+
+@router.get("/history")
+def get_history(limit: int = 20, user_role: Optional[str] = None) -> list[dict]:
+    """List recent gateway interactions from Postgres."""
+    return _shared_answer_store.list_history(limit=limit, user_role=user_role)
+
+
+@router.get("/answers/{answer_id}")
+def get_answer(answer_id: str) -> dict:
+    """Load a stored gateway response from Postgres ``gateway_answers``."""
+    record = _shared_answer_store.get(answer_id)
+    if not record:
+        raise HTTPException(status_code=404, detail=f"Gateway answer {answer_id} not found")
+    return record
+
+
+@router.get("/metrics", response_model=GatewayMetricsResponse)
+def metrics() -> GatewayMetricsResponse:
+    snapshot = _shared_audit.metrics()
+    return GatewayMetricsResponse(
+        generated_at=datetime.now(UTC),
+        **snapshot,
+    )