ultralytics-actions 0.0.25__tar.gz → 0.0.27__tar.gz

{ultralytics_actions-0.0.25 → ultralytics_actions-0.0.27}/PKG-INFO

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

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

{ultralytics_actions-0.0.25 → ultralytics_actions-0.0.27}/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.25 → ultralytics_actions-0.0.27}/actions/summarize_pr.py

@@ -13,12 +13,34 @@ from .utils import (
     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_issue_comment(pr_url, pr_body):
+    """Generates a personalized issue comment using AI based on the PR context."""
+    messages = [
+        {
+            "role": "system",
+            "content": "You are the Ultralytics AI assistant. Generate friendly GitHub issue comments. No @ mentions or direct addressing.",
+        },
+        {
+            "role": "user",
+            "content": f"Write a comment for a fixed GitHub issue using this merged PR context:\n\n{pr_body}\n\n"
+            f"Include:\n"
+            f"1. Reference to fix PR: {pr_url}\n"
+            f"2. Key changes in the PR and instructions to test the fix with:\n"
+            f"   - pip install git+https://github.com/ultralytics/ultralytics.git@main # immediate testing\n"
+            f"   - or await next release\n"
+            f"3. Request verification that PR fix works\n"
+            f"4. Thank 🙏 for reporting the issue and encourage reporting any new issues in the future\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:
@@ -72,8 +94,7 @@ def update_pr_description(repo_name, pr_number, new_summary, max_retries=2):
 
 
 def label_fixed_issues(pr_number):
-    """Labels issues that are closed by this PR when it's merged."""
-    # GraphQL query to get closing issues
+    """Labels issues closed by this PR when merged and notifies users about the fix with AI-generated comments."""
     query = """
 query($owner: String!, $repo: String!, $pr_number: Int!) {
     repository(owner: $owner, name: $repo) {
@@ -83,6 +104,8 @@ query($owner: String!, $repo: String!, $pr_number: Int!) {
                     number
                 }
             }
+            url
+            body
         }
     }
 }
@@ -97,26 +120,48 @@ query($owner: String!, $repo: String!, $pr_number: Int!) {
         return
 
     try:
-        issues = response.json()["data"]["repository"]["pullRequest"]["closingIssuesReferences"]["nodes"]
+        data = response.json()["data"]["repository"]["pullRequest"]
+        issues = data["closingIssuesReferences"]["nodes"]
+
+        # Generate personalized comment
+        comment = generate_issue_comment(pr_url=data["url"], pr_body=data["body"])
+
         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 AI-generated 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}"
+                )
     except KeyError as e:
         print(f"Error parsing GraphQL response: {e}")
         return
 
 
+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."""
     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...")
@@ -133,7 +178,9 @@ def main():
     # Update linked issues
     if PR.get("merged"):
         print("PR is merged, labeling fixed issues...")
-        label_fixed_issues(PR["number"])
+        label_fixed_issues(pr_number)
+        print("Removing TODO label from PR...")
+        remove_todos_on_merge(pr_number)
 
 
 if __name__ == "__main__":

{ultralytics_actions-0.0.25 → ultralytics_actions-0.0.27}/ultralytics_actions.egg-info/PKG-INFO

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