implement a bot to remind prs to link issues if not.

Author: sayakpaul

.github/workflows/pr_link_issue_reminder.yml

@@ -0,0 +1,33 @@
+name: PR Issue Link Reminder
+
+on:
+  schedule:
+    - cron: "30 7 * * *"
+  workflow_dispatch:
+
+jobs:
+  remind_or_close:
+    name: Remind or close PRs without a linked issue
+    if: github.repository == 'huggingface/diffusers'
+    runs-on: ubuntu-22.04
+    permissions:
+      contents: read
+      pull-requests: write
+      issues: write
+    env:
+      GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+    steps:
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+
+      - name: Setup Python
+        uses: actions/setup-python@a309ff8b426b58ec0e2a45f0f869d46889d02405  # v6
+        with:
+          python-version: 3.10
+
+      - name: Install requirements
+        run: |
+          pip install PyGithub requests
+
+      - name: Run reminder script
+        run: |
+          python utils/remind_link_issue.py

utils/remind_link_issue.py

@@ -0,0 +1,152 @@
+# Copyright 2025 The HuggingFace Team. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""
+Script to remind PR authors to link an issue, and close PRs that ignore the reminders.
+
+Behavior:
+- Scans open, non-draft PRs.
+- A PR is considered "linked" if GitHub's GraphQL `closingIssuesReferences` returns > 0
+  (covers both `Fixes #N` keywords in the body and issues linked via the GitHub UI).
+- If a PR is not linked, the script posts up to 3 reminder comments spaced 7 days apart.
+- If the 3rd reminder is older than 7 days and the PR is still not linked, the PR is closed.
+- PRs labeled `no-issue-needed` and bot-authored PRs are skipped.
+"""
+
+import os
+from datetime import datetime, timedelta, timezone
+
+import requests
+from github import Github
+
+
+REPO = "huggingface/diffusers"
+REMINDER_MARKER = "<!-- pr-link-issue-reminder -->"
+CLOSE_MARKER = "<!-- pr-link-issue-close -->"
+REMINDER_INTERVAL = timedelta(days=7)
+MAX_REMINDERS = 3
+BYPASS_LABELS = {"no-issue-needed"}
+
+GRAPHQL_URL = "https://api.github.com/graphql"
+GRAPHQL_QUERY = """
+query($owner: String!, $name: String!, $number: Int!) {
+  repository(owner: $owner, name: $name) {
+    pullRequest(number: $number) {
+      closingIssuesReferences(first: 1) {
+        totalCount
+      }
+    }
+  }
+}
+"""
+
+
+def has_linked_issue(token, owner, name, number):
+    response = requests.post(
+        GRAPHQL_URL,
+        json={"query": GRAPHQL_QUERY, "variables": {"owner": owner, "name": name, "number": number}},
+        headers={"Authorization": f"Bearer {token}"},
+        timeout=30,
+    )
+    response.raise_for_status()
+    payload = response.json()
+    return payload["data"]["repository"]["pullRequest"]["closingIssuesReferences"]["totalCount"] > 0
+
+
+def reminder_history(pr):
+    reminders = [c for c in pr.get_issue_comments() if REMINDER_MARKER in (c.body or "")]
+    reminders.sort(key=lambda c: c.created_at)
+    return reminders
+
+
+def reminder_body(author, count):
+    remaining = MAX_REMINDERS - count
+    lines = [
+        REMINDER_MARKER,
+        f"Hi @{author}, this PR does not appear to link an issue it fixes. "
+        "If this PR addresses an existing issue, please add a closing keyword "
+        "(e.g. `Fixes #1234`) to the PR description so the issue is linked.",
+        "",
+        f"Reminder **{count}/{MAX_REMINDERS}**. ",
+    ]
+    if remaining > 0:
+        lines[-1] += (
+            f"If no linked issue is added within {REMINDER_INTERVAL.days} days, "
+            f"you will receive {remaining} more reminder(s)."
+        )
+    else:
+        lines[-1] += (
+            f"This is the final reminder. If no linked issue is added within "
+            f"{REMINDER_INTERVAL.days} days, this PR will be closed automatically. "
+            "If this PR intentionally does not fix a tracked issue, a maintainer "
+            "can add the `no-issue-needed` label to bypass this check."
+        )
+    return "\n".join(lines)
+
+
+def close_body(author):
+    return (
+        f"{CLOSE_MARKER}\n"
+        f"Closing this PR because @{author} did not add a linked issue after "
+        f"{MAX_REMINDERS} reminders spaced {REMINDER_INTERVAL.days} days apart. "
+        "Please reopen once the PR description references the issue it fixes "
+        "(e.g. `Fixes #1234`), or ask a maintainer to add the `no-issue-needed` "
+        "label if this PR is intentionally unrelated to a tracked issue."
+    )
+
+
+def aware(ts):
+    return ts if ts.tzinfo is not None else ts.replace(tzinfo=timezone.utc)
+
+
+def main():
+    token = os.environ["GITHUB_TOKEN"]
+    g = Github(token)
+    repo = g.get_repo(REPO)
+    owner, name = REPO.split("/", 1)
+
+    now = datetime.now(timezone.utc)
+
+    for pr in repo.get_pulls(state="open"):
+        if pr.draft:
+            continue
+        if pr.user is None:
+            continue
+        author = pr.user.login
+        if not author or author.endswith("[bot]") or pr.user.type == "Bot":
+            continue
+        labels = {label.name for label in pr.labels}
+        if labels & BYPASS_LABELS:
+            continue
+        if has_linked_issue(token, owner, name, pr.number):
+            continue
+
+        reminders = reminder_history(pr)
+        count = len(reminders)
+
+        if count == 0:
+            pr.create_issue_comment(reminder_body(author, 1))
+            continue
+
+        if now - aware(reminders[-1].created_at) < REMINDER_INTERVAL:
+            continue
+
+        if count >= MAX_REMINDERS:
+            pr.create_issue_comment(close_body(author))
+            pr.edit(state="closed")
+        else:
+            pr.create_issue_comment(reminder_body(author, count + 1))
+
+
+if __name__ == "__main__":
+    main()