Function main

utils/label_issues.py:86–119

()

def main():
    try:
        title = os.environ.get("ISSUE_TITLE", "")
        body = os.environ.get("ISSUE_BODY", "")

        client = InferenceClient(api_key=os.environ["HF_TOKEN"])

        completion = client.chat.completions.create(
            model=os.environ.get("HF_MODEL", "Qwen/Qwen3.5-35B-A3B"),
            messages=[
                {"role": "system", "content": SYSTEM_PROMPT},
                {"role": "user", "content": USER_TEMPLATE.format(title=title, body=body)},
            ],
            response_format={"type": "json_object"},
            temperature=0,
        )

        response = completion.choices[0].message.content.strip()
        result = json.loads(response)

        labels = [l for l in result["labels"] if l in VALID_LABELS]
        model_name = result.get("model_name")

        if model_name:
            existing = get_existing_components()
            if not any(model_name.lower() in name for name in existing):
                labels.append("new-pipeline/model")

        if "bug" in labels and "Diffusers version:" not in body:
            labels.append("needs-env-info")

        print(json.dumps(labels))
    except Exception:
        print("Labeling failed", file=sys.stderr)


if __name__ == "__main__":