import json
import os
from pathlib import Path
import httpx

request = {
    "input": "Write a short poem about a peaceful garden. Use the whole word roses exactly three times, calm exactly once, and the letter e exactly twelve times (case-insensitive).",
    "policy": {
        "engine": "opa",
        "entrypoint": "data.poem.allow",
        "modules": {
            "poem.rego": Path("poem.rego").read_text(encoding="utf-8"),
        },
        "feedback_entrypoint": "data.poem.feedback",
    },
    "policy_max_retries": 8,
    "policy_feedback": "chat",
    "schema": json.loads(Path("poem.schema.json").read_text(encoding="utf-8")),
}
with httpx.Client(
    base_url="https://api.blatent.ai",
    headers={"Authorization": f"Bearer {os.environ['BLATENT_API_KEY']}"},
    timeout=195,
) as client:
    response = client.post("/v1/resolve", json=request)
    response.raise_for_status()
    print(json.dumps(response.json()["output"]))