3_create_report.py

"""Create a research report and poll until complete."""

import os
import time
import requests

api_key = os.environ["ELICIT_API_KEY"]

# Create
response = requests.post(
    "https://elicit.com/api/v1/reports",
    headers={"Authorization": f"Bearer {api_key}"},
    json={
        "researchQuestion": "What is the evidence for melatonin supplementation improving sleep quality in adults?",
        "maxSearchPapers": 50,
        "maxExtractPapers": 10,
    },
)
response.raise_for_status()
report = response.json()

report_id = report["reportId"]
print(f"Report created: {report['url']}")

# Poll (reports take 5-15 minutes)
while True:
    time.sleep(30)
    report = requests.get(
        f"https://elicit.com/api/v1/reports/{report_id}",
        headers={"Authorization": f"Bearer {api_key}"},
    ).json()
    print(f"Status: {report['status']}")
    if report["status"] == "completed":
        print(f"\n{report['result']['title']}")
        print(report["result"]["summary"])
        print(f"\n{report['url']}")
        break
    elif report["status"] == "failed":
        print(f"Report failed: {report.get('error')}")
        break