ultralytics-actions 0.1.1__tar.gz → 0.1.3__tar.gz

{ultralytics_actions-0.1.1 → ultralytics_actions-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-actions
-Version: 0.1.1
+Version: 0.1.3
 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.1.1 → ultralytics_actions-0.1.3}/actions/__init__.py

@@ -23,4 +23,4 @@
 #     ├── test_summarize_pr.py
 #     └── ...
 
-__version__ = "0.1.1"
+__version__ = "0.1.3"

{ultralytics_actions-0.1.1 → ultralytics_actions-0.1.3}/actions/first_interaction.py

@@ -5,12 +5,14 @@ from __future__ import annotations
 import os
 import time
 
+from . import review_pr
 from .utils import Action, filter_labels, get_completion, get_pr_open_response, remove_html_comments
 
 SUMMARY_START = (
     "## 🛠️ PR Summary\n\n<sub>Made with ❤️ by [Ultralytics Actions](https://github.com/ultralytics/actions)<sub>\n\n"
 )
 BLOCK_USER = os.getenv("BLOCK_USER", "false").lower() == "true"
+AUTO_PR_REVIEW = os.getenv("REVIEW", "true").lower() == "true"
 
 
 def apply_and_check_labels(event, number, node_id, issue_type, username, labels, label_descriptions):
@@ -176,19 +178,27 @@ YOUR {issue_type.upper()} RESPONSE:
 def main(*args, **kwargs):
     """Executes auto-labeling and custom response generation for new GitHub issues, PRs, and discussions."""
     event = Action(*args, **kwargs)
+    if event.should_skip_openai():
+        return
+
     number, node_id, title, body, username, issue_type, action = get_event_content(event)
     available_labels = event.get_repo_data("labels")
     label_descriptions = {label["name"]: label.get("description", "") for label in available_labels}
 
     # Use unified PR open response for new PRs (summary + labels + first comment in 1 API call)
     if issue_type == "pull request" and action == "opened":
+        if event.should_skip_pr_author():
+            return
+
         print("Processing PR open with unified API call...")
         diff = event.get_pr_diff()
         response = get_pr_open_response(event.repository, diff, title, body, label_descriptions)
 
         if summary := response.get("summary"):
             print("Updating PR description with summary...")
-            event.update_pr_description(number, SUMMARY_START + summary)
+            event.update_pr_description(number, SUMMARY_START + summary + "\n\n" + body)
+        else:
+            summary = body
 
         if relevant_labels := response.get("labels", []):
             apply_and_check_labels(event, number, node_id, issue_type, username, relevant_labels, label_descriptions)
@@ -197,6 +207,14 @@ def main(*args, **kwargs):
             print("Adding first interaction comment...")
             time.sleep(1)  # sleep to ensure label added first
             event.add_comment(number, node_id, first_comment, issue_type)
+
+        # Automatic PR review after first interaction
+        if AUTO_PR_REVIEW:
+            print("Starting automatic PR review...")
+            review_number = review_pr.dismiss_previous_reviews(event)
+            review_data = review_pr.generate_pr_review(event.repository, diff, title, summary)
+            review_pr.post_review_summary(event, review_data, review_number)
+            print("PR review completed")
         return
 
     # Handle issues and discussions (NOT PRs)

{ultralytics_actions-0.1.1 → ultralytics_actions-0.1.3}/actions/review_pr.py

@@ -79,49 +79,32 @@ def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_descri
     diff_truncated = len(diff_text) > limit
     lines_changed = sum(len(lines) for lines in diff_files.values())
 
-    comment_guidance = (
-        "Provide up to 1-3 comments only if critical issues exist"
-        if lines_changed < 50
-        else "Provide up to 3-5 comments only if high-impact issues exist"
-        if lines_changed < 200
-        else "Provide up to 5-10 comments only for the most critical issues"
-    )
-
     content = (
         "You are an expert code reviewer for Ultralytics. Provide detailed inline comments on specific code changes.\n\n"
-        "Focus on: Code quality, style, best practices, bugs, edge cases, error handling, performance, security, documentation, test coverage\n\n"
-        "FORMATTING: Use backticks for code, file names, branch names, function names, variable names, packages\n\n"
+        "Focus on: Bugs, security, performance, best practices, edge cases, error handling\n\n"
+        "FORMATTING: Use backticks for code: `x=3`, `file.py`, `function()`\n\n"
         "CRITICAL RULES:\n"
-        "1. Quality over quantity: Zero comments is fine for clean code - only flag truly important issues\n"
-        f"2. {comment_guidance} - these are maximums, not targets\n"
-        "3. CRITICAL: Do not post separate comments on adjacent/nearby lines (within 10 lines). Combine all related issues into ONE comment\n"
-        "4. When combining issues from multiple lines, use 'start_line' (first line) and 'line' (last line) to highlight the entire range\n"
-        "5. Each comment must reference separate areas - no overlapping line ranges\n"
-        "6. Prioritize: CRITICAL bugs/security > HIGH impact issues > code quality\n"
-        "7. Keep comments concise, friendly, and easy to understand - avoid jargon when possible\n"
-        "8. DO not comment on routine changes: adding imports, adding dependencies, updating version numbers, standard refactoring\n"
-        "9. Trust the developer - only flag issues with clear evidence of problems, not hypothetical concerns\n\n"
-        "SUMMARY GUIDELINES:\n"
-        "- Keep summary brief, clear, and actionable - avoid overly detailed explanations\n"
-        "- Highlight only the most important findings\n"
-        "- Do NOT include file names or line numbers in the summary - inline comments already show exact locations\n"
-        "- Focus on what needs to be fixed, not where\n\n"
-        "CODE SUGGESTIONS:\n"
+        "1. Quality over quantity - zero comments is fine for clean code, only flag truly important issues\n"
+        "2. Combine issues that are directly related to the same problem\n"
+        "3. Use 'start_line' and 'line' to highlight multi-line ranges when issues span multiple lines\n"
+        "4. Prioritize: CRITICAL bugs/security > HIGH impact > code quality improvements\n"
+        "5. Keep comments concise and friendly - avoid jargon\n"
+        "6. Skip routine changes: imports, version updates, standard refactoring\n\n"
+        "SUMMARY:\n"
+        "- Brief and actionable - what needs fixing, not where (locations shown in inline comments)\n\n"
+        "SUGGESTIONS:\n"
         "- ONLY provide 'suggestion' field when you have high certainty the code is problematic AND sufficient context for a confident fix\n"
         "- If uncertain about the correct fix, omit 'suggestion' field and explain the concern in 'message' only\n"
         "- Suggestions must be ready-to-merge code with NO comments, placeholders, or explanations\n"
         "- Suggestions replace ONLY the single line at 'line' - for multi-line fixes, describe the change in 'message' instead\n"
         "- Do NOT provide 'start_line' when including a 'suggestion' - suggestions are always single-line only\n"
         "- Suggestion content must match the exact indentation of the original line\n"
-        "- Never include triple backticks (```) in suggestions as they break markdown formatting\n"
+        "- Avoid triple backticks (```) in suggestions as they break markdown formatting\n"
         "- It's better to flag an issue without a suggestion than provide a wrong or uncertain fix\n\n"
         "Return JSON: "
         '{"comments": [{"file": "exact/path", "line": N, "severity": "HIGH", "message": "...", "suggestion": "..."}], "summary": "..."}\n\n'
         "Rules:\n"
-        "- Only comment on NEW lines (starting with + in diff)\n"
-        "- Use exact file paths from diff (no ./ prefix)\n"
-        "- Line numbers must match NEW file line numbers from @@ hunks\n"
-        "- When '- old' then '+ new', new line keeps SAME line number\n"
+        "- Only NEW lines (+ in diff), exact paths (no ./), correct line numbers from @@ hunks\n"
         "- Severity: CRITICAL, HIGH, MEDIUM, LOW, SUGGESTION\n"
         f"- Files changed: {len(file_list)} ({', '.join(file_list[:10])}{'...' if len(file_list) > 10 else ''})\n"
         f"- Lines changed: {lines_changed}\n"
@@ -236,19 +219,17 @@ def post_review_summary(event: Action, review_data: dict, review_number: int) ->
 
     review_title = f"{REVIEW_MARKER} {review_number}" if review_number > 1 else REVIEW_MARKER
     comments = review_data.get("comments", [])
-    event_type = (
-        "REQUEST_CHANGES" if any(c.get("severity") not in ["LOW", "SUGGESTION", None] for c in comments) else "APPROVE"
-    )
+    event_type = "COMMENT" if any(c.get("severity") not in ["LOW", "SUGGESTION", None] for c in comments) else "APPROVE"
 
     body = (
         f"## {review_title}\n\n"
         "<sub>Made with ❤️ by [Ultralytics Actions](https://github.com/ultralytics/actions)</sub>\n\n"
-        f"{review_data.get('summary', 'Review completed')}\n\n"
+        f"{review_data.get('summary', 'Review completed')[:1000]}\n\n"  # Clip summary length
     )
 
     if comments:
         shown = min(len(comments), 10)
-        body += f"💬 Posted {shown} inline comment{'s' if shown != 1 else ''}{' (10 shown, more available)' if len(comments) > 10 else ''}\n"
+        body += f"💬 Posted {shown} inline comment{'s' if shown != 1 else ''}\n"
 
     if review_data.get("diff_truncated"):
         body += "\n⚠️ **Large PR**: Review focused on critical issues. Some details may not be covered.\n"
@@ -258,14 +239,15 @@ def post_review_summary(event: Action, review_data: dict, review_number: int) ->
 
     # Build inline comments for the review
     review_comments = []
-    for comment in comments[:10]:
+    for comment in comments[:10]:  # Limit to 10 comments
         if not (file_path := comment.get("file")) or not (line := comment.get("line", 0)):
             continue
 
         severity = comment.get("severity", "SUGGESTION")
-        comment_body = f"{EMOJI_MAP.get(severity, '💭')} **{severity}**: {comment.get('message', '')}"
+        comment_body = f"{EMOJI_MAP.get(severity, '💭')} **{severity}**: {comment.get('message', '')[:1000]}"
 
         if suggestion := comment.get("suggestion"):
+            suggestion = suggestion[:1000]  # Clip suggestion length
             if "```" not in suggestion:
                 # Extract original line indentation and apply to suggestion
                 if original_line := review_data.get("diff_files", {}).get(file_path, {}).get(line):
@@ -309,6 +291,10 @@ def main(*args, **kwargs):
         print(f"Skipping: PR state is {event.pr.get('state') if event.pr else 'None'}")
         return
 
+    # Skip self-authored or bot PRs unless manually review_requested
+    if event.event_data.get("action") != "review_requested" and event.should_skip_pr_author():
+        return
+
     print(f"Starting PR review for #{event.pr['number']}")
     review_number = dismiss_previous_reviews(event)
 

{ultralytics_actions-0.1.1 → ultralytics_actions-0.1.3}/actions/summarize_pr.py

@@ -96,10 +96,11 @@ def main(*args, **kwargs):
     """Summarize a pull request and update its description with a summary."""
     event = Action(*args, **kwargs)
     action = event.event_data.get("action")
-
     if action == "opened":
         print("Skipping PR open - handled by first_interaction.py with unified API call")
         return
+    if event.should_skip_openai():
+        return
 
     print(f"Retrieving diff for PR {event.pr['number']}")
     diff = event.get_pr_diff()

{ultralytics_actions-0.1.1 → ultralytics_actions-0.1.3}/actions/summarize_release.py

@@ -19,7 +19,7 @@ def get_release_diff(event, previous_tag: str, latest_tag: str) -> str:
     """Retrieves the differences between two specified Git tags in a GitHub repository."""
     url = f"{GITHUB_API_URL}/repos/{event.repository}/compare/{previous_tag}...{latest_tag}"
     r = event.get(url, headers=event.headers_diff)
-    return r.text if r.status_code == 200 else f"Failed to get diff: {r.content}"
+    return r.text if r.status_code == 200 else f"Failed to get diff: {r.text}"
 
 
 def get_prs_between_tags(event, previous_tag: str, latest_tag: str) -> list:
@@ -42,8 +42,8 @@ def get_prs_between_tags(event, previous_tag: str, latest_tag: str) -> list:
         pr_numbers.update(pr_matches)
 
     prs = []
-    time.sleep(10)  # Allow final PR summary to update on merge
     for pr_number in sorted(pr_numbers):  # earliest to latest
+        time.sleep(1)  # Rate limit: GitHub search API has strict limits
         pr_url = f"{GITHUB_API_URL}/repos/{event.repository}/pulls/{pr_number}"
         pr_response = event.get(pr_url)
         if pr_response.status_code == 200:
@@ -52,7 +52,7 @@ def get_prs_between_tags(event, previous_tag: str, latest_tag: str) -> list:
                 {
                     "number": pr_data["number"],
                     "title": pr_data["title"],
-                    "body": remove_html_comments(pr_data["body"]),
+                    "body": remove_html_comments(pr_data.get("body", "")),
                     "author": pr_data["user"]["login"],
                     "html_url": pr_data["html_url"],
                     "merged_at": pr_data["merged_at"],
@@ -140,7 +140,7 @@ def generate_release_summary(
         },
         {
             "role": "user",
-            "content": f"Summarize the updates made in the '{latest_tag}' tag, focusing on major model or features changes, their purpose, and potential impact. Keep the summary clear and suitable for a broad audience. Add emojis to enliven the summary. Prioritize changes from the current PR (the first in the list), which is usually the most important in the release. Reply directly with a summary along these example guidelines, though feel free to adjust as appropriate:\n\n"
+            "content": f"Summarize the updates made in the '{latest_tag}' tag, focusing on major model or features changes, their purpose, and potential impact. Keep the summary clear and suitable for a broad audience. Add emojis to enliven the summary. Prioritize changes from the current PR (the last in the list), which is usually the most important in the release. Reply directly with a summary along these example guidelines, though feel free to adjust as appropriate:\n\n"
             f"## 🌟 Summary (single-line synopsis)\n"
             f"## 📊 Key Changes (bullet points highlighting any major changes)\n"
             f"## 🎯 Purpose & Impact (bullet points explaining any benefits and potential impact to users)\n\n\n"

{ultralytics_actions-0.1.1 → ultralytics_actions-0.1.3}/actions/utils/github_utils.py

@@ -114,6 +114,7 @@ class Action:
         self.verbose = verbose
         self.eyes_reaction_id = None
         self._pr_diff_cache = None
+        self._pr_summary_cache = None
         self._username_cache = None
         self._default_status = {
             "get": [200],
@@ -187,6 +188,36 @@ class Action:
         """Checks if a user is a member of the organization."""
         return self.get(f"{GITHUB_API_URL}/orgs/{self.repository.split('/')[0]}/members/{username}").status_code == 204
 
+    def should_skip_pr_author(self) -> bool:
+        """Checks if PR should be skipped based on author (self-authored or bot PRs)."""
+        if not self.pr:
+            return False
+        if pr_author := self.pr.get("user", {}).get("login"):
+            if pr_author == self.get_username():
+                print(f"Skipping: PR author ({pr_author}) is the same as bot")
+                return True
+            # Check both user.type and [bot] suffix for robust bot detection
+            if self.pr.get("user", {}).get("type") == "Bot" or pr_author.endswith("[bot]"):
+                print(f"Skipping: PR author ({pr_author}) is a bot")
+                return True
+        return False
+
+    def is_fork_pr(self) -> bool:
+        """Checks if PR is from a fork (different repo than base)."""
+        if not self.pr:
+            return False
+        head_repo = self.pr.get("head", {}).get("repo", {}).get("full_name")
+        return bool(head_repo) and head_repo != self.repository
+
+    def should_skip_openai(self) -> bool:
+        """Check if OpenAI operations should be skipped."""
+        from actions.utils.openai_utils import OPENAI_API_KEY
+
+        if not OPENAI_API_KEY:
+            print("⚠️ Skipping LLM operations (OPENAI_API_KEY not found)")
+            return True
+        return False
+
     def get_pr_diff(self) -> str:
         """Retrieves the diff content for a specified pull request with caching."""
         if self._pr_diff_cache:
@@ -256,6 +287,7 @@ class Action:
             updated_description = description + "\n\n" + new_summary
 
         self.patch(url, json={"body": updated_description})
+        self._pr_summary_cache = new_summary
 
     def get_label_ids(self, labels: list[str]) -> list[str]:
         """Retrieves GitHub label IDs for a list of label names using the GraphQL API."""
@@ -368,7 +400,7 @@ Thank you 🙏
         try:
             data = response.json()["data"]["repository"]["pullRequest"]
             comments = data["reviews"]["nodes"] + data["comments"]["nodes"]
-            token_username = self.get_username()
+            username = self.get_username()
             author = data["author"]["login"] if data["author"]["__typename"] != "Bot" else None
 
             contributors = {x["author"]["login"] for x in comments if x["author"]["__typename"] != "Bot"}
@@ -381,10 +413,10 @@ Thank you 🙏
                             contributors.add(login)
 
             contributors.discard(author)
-            contributors.discard(token_username)
+            contributors.discard(username)
 
             pr_credit = ""
-            if author and author != token_username:
+            if author and author != username:
                 pr_credit += f"@{author}"
             if contributors:
                 pr_credit += (" with contributions from " if pr_credit else "") + ", ".join(

{ultralytics_actions-0.1.1 → ultralytics_actions-0.1.3}/ultralytics_actions.egg-info/PKG-INFO

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