Fill forms from email

import os
import requests

def sage(path, body):
    r = requests.post(
        "https://sage.levanto.ai" + path,
        headers={"Authorization": f"Bearer {os.environ['SAGE_API_KEY']}"},
        json=body,
        timeout=30,
    )
    r.raise_for_status()
    return r.json()

FORM = {
    "policyholder": "the person making the claim",
    "policy_number": "the insurance policy number",
    "phone": "the phone number of the person making the claim",
    "incident_date": {"type": "date", "definition": "the date the accident happened"},
    "incident_time": {"type": "time", "definition": "the time the accident happened"},
    "vehicle": "the make and model of the policyholder's car",
    "registration": "the policyholder's car registration",
    "damage": ["a part of the policyholder's car that was damaged"],
    "repair_estimate": {"type": "money", "definition": "the quoted cost of the repair"},
    "police_report": "the police report or crime reference number",
}

email = """Hi,

I need to report an accident. My name is Daniel Okoro and my policy number is AU-77120-K.

On Friday 16 October 2026, at around 6:40 pm, I was stopped at the lights on Baker Street when a white van went into the back of my car. It's an Aurora C5, registration AB12 CDE.

The van driver was Sam Whitfield. Nobody was hurt, thankfully, but the rear bumper and the boot lid are smashed. The garage quoted £2,340 to fix it.

A police officer came but I didn't get a report number. They said they would send it by post.

Thanks,
Daniel
07700 900123"""

results = sage("/v1/extract", {"content": email, "extract": FORM})["results"]

for field, r in results.items():
    if "values" in r:  # a field with every mention
        sure = all((v["probability"] or 0) >= 0.6 for v in r["values"])
        print(field, [v["value"] for v in r["values"]], "" if sure else "-> check")
    elif r["value"] is None:
        print(field, "-> ask the customer" if r["probability"] is not None else "-> check: too close to call")
    else:
        print(field, r["value"], "" if r["probability"] >= 0.6 else "-> check")