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

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

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

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

{ultralytics_actions-0.0.22 → 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.22 → 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:
@@ -71,12 +93,75 @@ 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 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) {
+        pullRequest(number: $pr_number) {
+            closingIssuesReferences(first: 50) {
+                nodes {
+                    number
+                }
+            }
+            url
+            body
+        }
+    }
+}
+"""
+
+    owner, repo = GITHUB_REPOSITORY.split("/")
+    variables = {"owner": owner, "repo": repo, "pr_number": pr_number}
+    graphql_url = "https://api.github.com/graphql"
+    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
+
+    try:
+        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)
+
+            # 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 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...")
@@ -90,6 +175,13 @@ def main():
     else:
         print(f"Failed to update PR description. Status code: {status_code}")
 
+    # Update linked issues
+    if PR.get("merged"):
+        print("PR is merged, labeling fixed issues...")
+        label_fixed_issues(pr_number)
+        print("Removing TODO label from PR...")
+        remove_todos_on_merge(pr_number)
+
 
 if __name__ == "__main__":
     main()

{ultralytics_actions-0.0.22 → ultralytics_actions-0.0.27}/actions/utils/github_utils.py

@@ -23,8 +23,6 @@ if GITHUB_EVENT_PATH:
 PR = EVENT_DATA.get("pull_request", {})
 DISCUSSION = EVENT_DATA.get("discussion", {})
 
-INPUTS = {k[6:].lower(): v for k, v in os.environ.items() if k.startswith("INPUT_")}  # actions inputs dictionary
-
 
 def get_pr_diff(pr_number: int) -> str:
     """Retrieves the diff content for a specified pull request in a GitHub repository."""
@@ -57,8 +55,12 @@ def graphql_request(query: str, variables: dict = None) -> dict:
 
 def check_pypi_version(pyproject_toml="pyproject.toml"):
     """Compares local and PyPI package versions to determine if a new version should be published."""
+    import re
+
     import tomllib  # requires Python>=3.11
 
+    version_pattern = re.compile(r"^\d+\.\d+\.\d+$")  # e.g. 0.0.0
+
     with open(pyproject_toml, "rb") as f:
         pyproject = tomllib.load(f)
 
@@ -73,6 +75,9 @@ def check_pypi_version(pyproject_toml="pyproject.toml"):
             local_version = next(line.split("=")[1].strip().strip("'\"") for line in f if line.startswith(attr_name))
 
     print(f"Local Version: {local_version}")
+    if not bool(version_pattern.match(local_version)):
+        print("WARNING: Incorrect local version pattern")
+        return "0.0.0", "0.0.0", False
 
     # Get online version from PyPI
     response = requests.get(f"https://pypi.org/pypi/{package_name}/json")

{ultralytics_actions-0.0.22 → 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.22
+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>