Route new bugs to the right team

from clickruntime import on_event, http, credentials, ai, notify

def linear(query, variables=None):
    r = http.post(
        "https://api.linear.app/graphql",
        headers={"Authorization": credentials.secret("LINEAR_API_KEY")},
        json={"query": query, "variables": variables or {}},
    )
    r.raise_for_status()
    return r.json()["data"]

@on_event
def handle(event):
    body = event.json or {}
    issue = body.get("data", {})
    if body.get("action") != "create" or (issue.get("state") or {}).get("type") != "triage":
        return {"triaged": False}
    teams = linear("{ teams { nodes { id name } } }")["teams"]["nodes"]
    labels = linear("{ issueLabels(first: 100) { nodes { id name } } }")["issueLabels"]["nodes"]
    pick = ai.generate(
        f"Issue: {issue.get('title')}\n{(issue.get('description') or '')[:1500]}\n"
        f"Teams: {[t['name'] for t in teams]}\nLabels: {[l['name'] for l in labels]}",
        system="Pick the best team and label and give a one-line reason.",
        json_schema={"type": "object", "properties": {"team": {"type": "string"}, "label": {"type": "string"}, "reason": {"type": "string"}}},
        max_output_tokens=120,
    ).json
    team_id = next(t["id"] for t in teams if t["name"] == pick["team"])
    label_ids = [l["id"] for l in labels if l["name"] == pick["label"]]
    linear("""mutation($id: String!, $input: IssueUpdateInput!) {
      issueUpdate(id: $id, input: $input) { success } }""",
           {"id": issue["id"], "input": {"teamId": team_id, "labelIds": label_ids}})
    notify.slack(f"{issue.get('title')} → {pick['team']} / {pick['label']}: {pick['reason']}", channel="#triage")
    return pick