__init__.py

from fastapi import HTTPException
import httpx
from nilai_common import AttestationReport

ATTESTATION_URL = "http://nilcc-attester/v2/report"


async def get_attestation_report(nonce: str) -> AttestationReport:
    """Get the attestation report"""

    try:
        async with httpx.AsyncClient() as client:
            response: httpx.Response = await client.get(ATTESTATION_URL)
            response_json = response.json()
            return AttestationReport(
                nonce=nonce,
                gpu_attestation=response_json["gpu_token"],
                cpu_attestation=response_json["report"],
                verifying_key="",  # Added later by the API
            )
    except httpx.HTTPStatusError as e:
        raise HTTPException(
            status_code=e.response.status_code,
            detail=str("Error getting attestation report" + str(e)),
        )
    except Exception as e:
        raise HTTPException(
            status_code=500, detail=str("Error getting attestation report" + str(e))
        )