ultralytics-actions 0.0.25__py3-none-any.whl → 0.0.30__py3-none-any.whl

actions/__init__.py

@@ -22,4 +22,4 @@
 #     ├── test_summarize_pr.py
 #     └── ...
 
-__version__ = "0.0.25"
+__version__ = "0.0.30"

actions/first_interaction.py

@@ -335,7 +335,7 @@ INSTRUCTIONS:
 - Adapt the example {issue_type} response below as appropriate, keeping all badges, links and references provided
 - For bug reports, specifically request a minimum reproducible example (MRE) if not provided
 - INCLUDE ALL LINKS AND INSTRUCTIONS IN THE EXAMPLE BELOW, customized as appropriate
-- In your response, mention to the user that this is an automated response and that an Ultralytics engineer will also assist soon
+- Mention to the user that this is an automated response and that an Ultralytics engineer will also assist soon
 - Do not add a sign-off or valediction like "best regards" at the end of your response
 - Do not add spaces between bullet points or numbered lists
 - Only link to files or URLs in the example below, do not add external links
@@ -359,7 +359,7 @@ YOUR {issue_type.upper()} RESPONSE:
     messages = [
         {
             "role": "system",
-            "content": f"You are a helpful assistant responding to GitHub {issue_type}s for the {org_name} organization.",
+            "content": f"You are a helpful assistant responding to GitHub {issue_type}s for {org_name}.",
         },
         {"role": "user", "content": prompt},
     ]

actions/summarize_pr.py

@@ -10,15 +10,72 @@ from .utils import (
     GITHUB_REPOSITORY,
     PR,
     get_completion,
+    get_github_username,
     get_pr_diff,
 )
 
-# Action settings
+# Constants
 SUMMARY_START = (
     "## 🛠️ PR Summary\n\n<sub>Made with ❤️ by [Ultralytics Actions](https://github.com/ultralytics/actions)<sub>\n\n"
 )
 
 
+def generate_merge_message(pr_author, contributors, pr_summary=None):
+    """Generates a thank-you message for merged PR contributors."""
+    contributors_str = ", ".join(f"@{c}" for c in contributors if c != pr_author)
+    mention_str = f"@{pr_author}"
+    if contributors_str:
+        mention_str += f" and {contributors_str}"
+
+    messages = [
+        {
+            "role": "system",
+            "content": "You are an Ultralytics AI assistant. Generate meaningful, inspiring messages to GitHub users.",
+        },
+        {
+            "role": "user",
+            "content": f"Write a friendly thank you for a merged PR by these GitHub contributors: {mention_str}. "
+            f"Context from PR:\n{pr_summary}\n\n"
+            f"Start with the exciting message that this PR is now merged, and weave in an inspiring quote "
+            f"from a famous figure in science, philosophy or stoicism. "
+            f"Keep the message concise yet relevant to the specific contributions in this PR. "
+            f"We want the contributors to feel their effort is appreciated and will make a difference in the world.",
+        },
+    ]
+    return get_completion(messages)
+
+
+def post_merge_message(pr_number, pr_author, contributors, summary):
+    """Posts thank you message on PR after merge."""
+    message = generate_merge_message(pr_author, contributors, summary)
+    comment_url = f"{GITHUB_API_URL}/repos/{GITHUB_REPOSITORY}/issues/{pr_number}/comments"
+    response = requests.post(comment_url, json={"body": message}, headers=GITHUB_HEADERS)
+    return response.status_code == 201
+
+
+def generate_issue_comment(pr_url, pr_summary):
+    """Generates a personalized issue comment using based on the PR context."""
+    messages = [
+        {
+            "role": "system",
+            "content": "You are an Ultralytics AI assistant. Generate friendly GitHub issue comments. No @ mentions or direct addressing.",
+        },
+        {
+            "role": "user",
+            "content": f"Write a GitHub issue comment announcing a potential fix has been merged in linked PR {pr_url}\n\n"
+            f"Context from PR:\n{pr_summary}\n\n"
+            f"Include:\n"
+            f"1. An explanation of key changes from the PR that may resolve this issue\n"
+            f"2. Options for testing if PR changes have resolved this issue:\n"
+            f"   - pip install git+https://github.com/ultralytics/ultralytics.git@main # test latest changes\n"
+            f"   - or await next official PyPI release\n"
+            f"3. Request feedback on whether the PR changes resolve the issue\n"
+            f"4. Thank 🙏 for reporting the issue and welcome any further feedback if the issue persists\n\n",
+        },
+    ]
+    return get_completion(messages)
+
+
 def generate_pr_summary(repo_name, diff_text):
     """Generates a concise, professional summary of a PR using OpenAI's API for Ultralytics repositories."""
     if not diff_text:
@@ -71,9 +128,8 @@ def update_pr_description(repo_name, pr_number, new_summary, max_retries=2):
     return update_response.status_code
 
 
-def label_fixed_issues(pr_number):
-    """Labels issues that are closed by this PR when it's merged."""
-    # GraphQL query to get closing issues
+def label_fixed_issues(pr_number, pr_summary):
+    """Labels issues closed by this PR when merged, notifies users, and returns PR contributors."""
     query = """
 query($owner: String!, $repo: String!, $pr_number: Int!) {
     repository(owner: $owner, name: $repo) {
@@ -83,6 +139,15 @@ query($owner: String!, $repo: String!, $pr_number: Int!) {
                     number
                 }
             }
+            url
+            body
+            author { login, __typename }
+            reviews(first: 50) {
+                nodes { author { login, __typename } }
+            }
+            comments(first: 50) {
+                nodes { author { login, __typename } }
+            }
         }
     }
 }
@@ -94,29 +159,59 @@ query($owner: String!, $repo: String!, $pr_number: Int!) {
     response = requests.post(graphql_url, json={"query": query, "variables": variables}, headers=GITHUB_HEADERS)
     if response.status_code != 200:
         print(f"Failed to fetch linked issues. Status code: {response.status_code}")
-        return
+        return [], None
 
     try:
-        issues = response.json()["data"]["repository"]["pullRequest"]["closingIssuesReferences"]["nodes"]
-        for issue in issues:
+        data = response.json()["data"]["repository"]["pullRequest"]
+        comments = data["reviews"]["nodes"] + data["comments"]["nodes"]  # merge lists
+        author = data["author"]["login"]
+
+        # Get unique contributors from reviews and comments
+        contributors = {x["author"]["login"] for x in comments if x["author"]["__typename"] != "Bot"}
+        contributors.discard(author)  # Remove author from contributors list
+
+        # Generate personalized comment
+        comment = generate_issue_comment(pr_url=data["url"], pr_summary=pr_summary)
+
+        # Update linked issues
+        for issue in data["closingIssuesReferences"]["nodes"]:
             issue_number = issue["number"]
+            # Add fixed label
             label_url = f"{GITHUB_API_URL}/repos/{GITHUB_REPOSITORY}/issues/{issue_number}/labels"
             label_response = requests.post(label_url, json={"labels": ["fixed"]}, headers=GITHUB_HEADERS)
-            if label_response.status_code == 200:
-                print(f"Added 'fixed' label to issue #{issue_number}")
+
+            # Add comment
+            comment_url = f"{GITHUB_API_URL}/repos/{GITHUB_REPOSITORY}/issues/{issue_number}/comments"
+            comment_response = requests.post(comment_url, json={"body": comment}, headers=GITHUB_HEADERS)
+
+            if label_response.status_code == 200 and comment_response.status_code == 201:
+                print(f"Added 'fixed' label and comment to issue #{issue_number}")
             else:
-                print(f"Failed to add label to issue #{issue_number}. Status: {label_response.status_code}")
+                print(
+                    f"Failed to update issue #{issue_number}. Label status: {label_response.status_code}, "
+                    f"Comment status: {comment_response.status_code}"
+                )
+
+        return contributors, author
     except KeyError as e:
         print(f"Error parsing GraphQL response: {e}")
-        return
+        return [], None
+
+
+def remove_todos_on_merge(pr_number):
+    """Removes specified labels from PR."""
+    for label in ["TODO"]:  # Can be extended with more labels in the future
+        requests.delete(
+            f"{GITHUB_API_URL}/repos/{GITHUB_REPOSITORY}/issues/{pr_number}/labels/{label}", headers=GITHUB_HEADERS
+        )
 
 
 def main():
-    """Summarize a pull request and update its description with an AI-generated summary."""
+    """Summarize a pull request and update its description with a summary."""
     pr_number = PR["number"]
 
     print(f"Retrieving diff for PR {pr_number}")
-    diff = get_pr_diff(PR["number"])
+    diff = get_pr_diff(pr_number)
 
     # Generate PR summary
     print("Generating PR summary...")
@@ -130,10 +225,17 @@ def main():
     else:
         print(f"Failed to update PR description. Status code: {status_code}")
 
-    # Update linked issues
+    # Update linked issues and post thank you message if merged
     if PR.get("merged"):
         print("PR is merged, labeling fixed issues...")
-        label_fixed_issues(PR["number"])
+        contributors, author = label_fixed_issues(pr_number, summary)
+        print("Removing TODO label from PR...")
+        remove_todos_on_merge(pr_number)
+        username = get_github_username()  # get GITHUB_TOKEN username
+        if author and author != username:
+            print("Posting PR author thank you message...")
+            contributors.discard(username)
+            post_merge_message(pr_number, author, contributors, summary)
 
 
 if __name__ == "__main__":

actions/utils/__init__.py

@@ -14,6 +14,7 @@ from .github_utils import (
     PR,
     check_pypi_version,
     get_github_data,
+    get_github_username,
     get_pr_diff,
     graphql_request,
     ultralytics_actions_info,
@@ -38,6 +39,7 @@ __all__ = (
     "OPENAI_API_KEY",
     "OPENAI_MODEL",
     "get_completion",
+    "get_github_username",
     "check_pypi_version",
     "ultralytics_actions_info",
 )

actions/utils/github_utils.py

@@ -24,6 +24,27 @@ PR = EVENT_DATA.get("pull_request", {})
 DISCUSSION = EVENT_DATA.get("discussion", {})
 
 
+def get_github_username():
+    """Gets username associated with the GitHub token in GITHUB_HEADERS."""
+    query = """
+    query {
+        viewer {
+            login
+        }
+    }
+    """
+    response = requests.post("https://api.github.com/graphql", json={"query": query}, headers=GITHUB_HEADERS)
+    if response.status_code != 200:
+        print(f"Failed to fetch authenticated user. Status code: {response.status_code}")
+        return None
+
+    try:
+        return response.json()["data"]["viewer"]["login"]
+    except KeyError as e:
+        print(f"Error parsing authenticated user response: {e}")
+        return None
+
+
 def get_pr_diff(pr_number: int) -> str:
     """Retrieves the diff content for a specified pull request in a GitHub repository."""
     url = f"{GITHUB_API_URL}/repos/{GITHUB_REPOSITORY}/pulls/{pr_number}"

{ultralytics_actions-0.0.25.dist-info → ultralytics_actions-0.0.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics-actions
-Version: 0.0.25
+Version: 0.0.30
 Summary: Ultralytics Actions for GitHub automation and PR management.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>

ultralytics_actions-0.0.30.dist-info/RECORD

@@ -0,0 +1,15 @@
+actions/__init__.py,sha256=vI2djkUXlHUwTefYfOmHPdRKls05ZS0all3rQSdIKXw,749
+actions/first_interaction.py,sha256=ehHkFwWr-14LgesGmO4lOthu8z626FgvV9e6fiyzN8Q,17648
+actions/summarize_pr.py,sha256=vkMVfy8ukoWMf_BrS1Ykh34GHFDkREaw9hNCiUUIClE,10529
+actions/summarize_release.py,sha256=l8NBdTAXLysfNKl1Kf_1tyuBRmeEBLyzTDXS6s5_eQg,8350
+actions/update_markdown_code_blocks.py,sha256=WBNcMD_KKsZS-qSPBn6O1G0ggQ_VrT-jTQffbg7xH_M,6369
+actions/utils/__init__.py,sha256=e3vKraD3_YpFiVUn3B3KR0diqG1ZrXMV9eXQYikObxo,1025
+actions/utils/common_utils.py,sha256=XT8GG0SWBtlZLruA0nKrY4AJpkitvPbM8zndE8etuDo,3548
+actions/utils/github_utils.py,sha256=QBNBx_qb3cbMwYmAQqEmUmGj33jir8Hjc6dr8Jk3vws,6286
+actions/utils/openai_utils.py,sha256=O0sYtLFTPqC4inw9_NWcVNlhElcp0iJ1FvAX3L3arKo,1834
+ultralytics_actions-0.0.30.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+ultralytics_actions-0.0.30.dist-info/METADATA,sha256=TZDWraFY1tpuTQMnkkF2FHmTM8u5JiP-TLE6oHtvDJc,10591
+ultralytics_actions-0.0.30.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+ultralytics_actions-0.0.30.dist-info/entry_points.txt,sha256=GowvOFplj0C7JmsjbKcbpgLpdf2r921pcaOQkAHWZRA,378
+ultralytics_actions-0.0.30.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
+ultralytics_actions-0.0.30.dist-info/RECORD,,

ultralytics_actions-0.0.25.dist-info/RECORD

@@ -1,15 +0,0 @@
-actions/__init__.py,sha256=t0B8aBSaP2FlwHHA-wL74q6apk3Kgkh3m6RddLrLwaw,749
-actions/first_interaction.py,sha256=cLXo5hmhOgTfk7F9LbGJeRdg6vvrKUe-0taCa_TkpAg,17683
-actions/summarize_pr.py,sha256=R-GqxWF24CYRb-jPNF2tckXUV9g5BG_e3tKZwHlhtNQ,5737
-actions/summarize_release.py,sha256=l8NBdTAXLysfNKl1Kf_1tyuBRmeEBLyzTDXS6s5_eQg,8350
-actions/update_markdown_code_blocks.py,sha256=WBNcMD_KKsZS-qSPBn6O1G0ggQ_VrT-jTQffbg7xH_M,6369
-actions/utils/__init__.py,sha256=0vRjFc7i2WOlphuxdUxQo5BuNipgwGw2Bs-fdUBDeUw,973
-actions/utils/common_utils.py,sha256=XT8GG0SWBtlZLruA0nKrY4AJpkitvPbM8zndE8etuDo,3548
-actions/utils/github_utils.py,sha256=O9hTo8amqT890YIixMvALNur7R8SGRMJ8xXJT3nwpTc,5659
-actions/utils/openai_utils.py,sha256=O0sYtLFTPqC4inw9_NWcVNlhElcp0iJ1FvAX3L3arKo,1834
-ultralytics_actions-0.0.25.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-ultralytics_actions-0.0.25.dist-info/METADATA,sha256=lp9CnsStoZ6FQdgpEpCpxcLCZbna4D-cKCC5mOAAVys,10591
-ultralytics_actions-0.0.25.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-ultralytics_actions-0.0.25.dist-info/entry_points.txt,sha256=GowvOFplj0C7JmsjbKcbpgLpdf2r921pcaOQkAHWZRA,378
-ultralytics_actions-0.0.25.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
-ultralytics_actions-0.0.25.dist-info/RECORD,,