Get Form Filling Structured Data

from typing import Any, TypedDict, cast

import requests

BASE_URL = "https://sdp.suki-stage.com"


class GeneratedFormInstance(TypedDict, total=False):
    # Shared shape for static templates and dynamic schemas.
    id: str  # Dynamic: partner form_filling.values[].id when provided
    data: dict[str, Any]
    correlation_id: str
    created_at: str
    type: str
    title: str  # Dynamic: Context name when provided
    patient_id: str
    # Usually set for Suki templates; often empty for partner schemas.
    status: str
    form_template_id: str
    metadata: dict[str, Any]


class NonGeneratedFormInstance(TypedDict, total=False):
    # Sparse placeholder when a form did not produce output.
    form_template_id: str


class FormFillingStructuredData(TypedDict, total=False):
    generated_values: list[GeneratedFormInstance]
    non_generated_values: list[NonGeneratedFormInstance]


class FormFillingStructuredDataResponse(TypedDict):
    structured_data: FormFillingStructuredData


class ApiHttpError(RuntimeError):
    """Wrong HTTP status; OpenAPI errors usually include JSON with message + code."""

    def __init__(self, status: int, url: str, detail: str) -> None:
        super().__init__(f"HTTP {status} {url}: {detail}")
        self.status = status
        self.url = url


def _get_expect_json_object(url: str, headers: dict[str, str], expect_status: int) -> dict[str, Any]:
    r = requests.get(url, headers=headers, timeout=60)
    if r.status_code == expect_status:
        data = r.json()
        if isinstance(data, dict):
            return data
        raise ApiHttpError(expect_status, url, "response JSON was not an object")

    detail = ""
    try:
        err = r.json()
        if isinstance(err, dict) and isinstance(err.get("message"), str):
            detail = err["message"]
    except ValueError:
        detail = (r.text or "")[:500]
    raise ApiHttpError(r.status_code, url, detail or "(no body)")


def get_form_filling_structured_data(suki_token: str, ambient_session_id: str) -> FormFillingStructuredDataResponse:
    """GET /api/v1/form-filling/session/{ambient_session_id}/structured-data (sdp_suki_token header required). HTTP 200."""
    url = f"{BASE_URL}/api/v1/form-filling/session/{ambient_session_id}/structured-data"
    data = _get_expect_json_object(url, {"sdp_suki_token": suki_token, "sdp_provider_id": "<sdp_provider_id>"}, 200)
    sd = data.get("structured_data")
    if not isinstance(sd, dict):
        raise ValueError(f"{url}: 200 response missing structured_data object")
    return cast(FormFillingStructuredDataResponse, {"structured_data": sd})


if __name__ == "__main__":
    try:
        out = get_form_filling_structured_data("YOUR_SUKI_TOKEN", "YOUR_AMBIENT_SESSION_ID")
        structured = out["structured_data"]
        generated = structured.get("generated_values") or []

        # For dynamic forms, match by id (Context form_filling.values[].id) or title (name).
        by_id = {item.get("id"): item for item in generated if item.get("id")}
        print(structured)
        print("by_id keys:", list(by_id.keys()))
    except (ApiHttpError, ValueError) as e:
        print(e)