ultralytics-actions 0.0.26__tar.gz → 0.0.28__tar.gz

{ultralytics_actions-0.0.26 → ultralytics_actions-0.0.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics-actions
-Version: 0.0.26
+Version: 0.0.28
 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.26 → ultralytics_actions-0.0.28}/actions/__init__.py

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

{ultralytics_actions-0.0.26 → ultralytics_actions-0.0.28}/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},
     ]

{ultralytics_actions-0.0.26 → ultralytics_actions-0.0.28}/actions/summarize_pr.py

@@ -10,6 +10,7 @@ from .utils import (
     GITHUB_REPOSITORY,
     PR,
     get_completion,
+    get_github_username,
     get_pr_diff,
 )
 
@@ -19,6 +20,62 @@ SUMMARY_START = (
 )
 
 
+def generate_merge_message(pr_author, contributors, pr_summary=None):
+    """Generates an AI thank you message for merged PRs using OpenAI."""
+    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"Make the message relevant to the specific contributions in this PR. "
+            f"We want them to feel their hard work is acknowledged and will make a difference in the world.",
+        },
+    ]
+    return get_completion(messages)
+
+
+def post_merge_message(pr_number, pr_author, contributors, summary):
+    """Posts AI-generated 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 AI 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 comment for a GitHub issue where a potential fix has been merged in 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. Testing options:\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 these 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 }
+            reviews(first: 50) {
+                nodes { author { login } }
+            }
+            comments(first: 50) {
+                nodes { author { login } }
+            }
         }
     }
 }
@@ -94,21 +159,44 @@ 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"]
+        data = response.json()["data"]["repository"]["pullRequest"]
+        issues = data["closingIssuesReferences"]["nodes"]
+        author = data["author"]["login"]
+
+        # Get unique contributors from reviews and comments
+        contributors = {review["author"]["login"] for review in data["reviews"]["nodes"]}
+        contributors.update(comment["author"]["login"] for comment in data["comments"]["nodes"])
+        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 issues:
             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):
@@ -138,12 +226,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__":

{ultralytics_actions-0.0.26 → ultralytics_actions-0.0.28}/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",
 )

{ultralytics_actions-0.0.26 → ultralytics_actions-0.0.28}/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.26 → ultralytics_actions-0.0.28}/ultralytics_actions.egg-info/PKG-INFO

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