Tag and categorise new posts automatically

from clickruntime import on_trigger, credentials, http, state, ai, report_fetched

SITE = "https://blog.example.com"
SCHEMA = {"type": "object", "required": ["tags", "category"], "properties": {
    "tags": {"type": "array", "items": {"type": "string"}, "maxItems": 5}, "category": {"type": "string"}}}

def wp(method, path, **kw):
    resp = http.request(method, f"{SITE}/wp-json/wp/v2/{path}", **kw,
                        auth=("admin", credentials.secret("WP_APP_PASSWORD")))
    resp.raise_for_status()
    return resp.json()

@on_trigger
def run(event):
    done = set(state.get("done") or [])
    posts = [p for p in wp("GET", "posts", params={"per_page": 20, "context": "edit", "status": "any"})
             if not p["tags"] and p["id"] not in done]
    report_fetched(len(posts))
    if not posts:
        return {"summary": "Nothing new"}
    tags = {t["name"]: t["id"] for t in wp("GET", "tags", params={"per_page": 100})}
    cats = {c["name"]: c["id"] for c in wp("GET", "categories", params={"per_page": 100})}
    for p in posts:
        pick = ai.generate(f"Title: {p['title']['raw']}\n\n{p['content']['raw'][:2500]}\n\nTags: {', '.join(tags)}\n"
                           f"Categories: {', '.join(cats)}", json_schema=SCHEMA, max_output_tokens=100,
                           system="Choose up to five tags and one category from the lists given.").json
        tag_ids = [tags[t] for t in pick["tags"] if t in tags]
        wp("POST", f"posts/{p['id']}", json={"tags": tag_ids, "categories": [cats.get(pick["category"], p["categories"][0])]})
        done.add(p["id"])
    state.set("done", sorted(done))
    return {"tagged": len(posts)}