ultralytics-actions 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

actions/__init__.py

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

actions/dispatch_actions.py

@@ -92,7 +92,7 @@ def main(*args, **kwargs):
         return
 
     # Get comment info
-    comment_body = event.event_data["comment"].get("body", "")
+    comment_body = event.event_data["comment"].get("body") or ""
     username = event.event_data["comment"]["user"]["login"]
 
     # Check for keyword without surrounding backticks to avoid triggering on replies

actions/first_interaction.py

@@ -52,7 +52,7 @@ def get_event_content(event) -> tuple[int, str, str, str, str, str, str]:
     number = item["number"]
     node_id = item.get("node_id") or item.get("id")
     title = item["title"]
-    body = remove_html_comments(item.get("body", ""))
+    body = remove_html_comments(item.get("body") or "")
     username = item["user"]["login"]
     return number, node_id, title, body, username, issue_type, action
 
@@ -183,7 +183,7 @@ def main(*args, **kwargs):
 
     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}
+    label_descriptions = {label["name"]: label.get("description") or "" 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":

actions/review_pr.py

@@ -5,9 +5,10 @@ from __future__ import annotations
 import json
 import re
 
-from .utils import GITHUB_API_URL, Action, get_completion, remove_html_comments
+from .utils import GITHUB_API_URL, MAX_PROMPT_CHARS, Action, get_completion, remove_html_comments
 
 REVIEW_MARKER = "🔍 PR Review"
+ERROR_MARKER = "⚠️ Review generation encountered an error"
 EMOJI_MAP = {"CRITICAL": "❗", "HIGH": "⚠️", "MEDIUM": "💡", "LOW": "📝", "SUGGESTION": "💭"}
 SKIP_PATTERNS = [
     r"\.lock$",  # Lock files
@@ -75,8 +76,7 @@ def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_descri
         return {"comments": [], "summary": f"All {skipped_count} changed files are generated/vendored (skipped review)"}
 
     file_list = list(diff_files.keys())
-    limit = round(128000 * 3.3 * 0.5)  # 3.3 characters per token for half a 256k context window
-    diff_truncated = len(diff_text) > limit
+    diff_truncated = len(diff_text) > MAX_PROMPT_CHARS
     lines_changed = sum(len(lines) for lines in diff_files.values())
 
     content = (
@@ -118,7 +118,7 @@ def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_descri
                 f"Review this PR in https://github.com/{repository}:\n"
                 f"Title: {pr_title}\n"
                 f"Description: {remove_html_comments(pr_description or '')[:1000]}\n\n"
-                f"Diff:\n{diff_text[:limit]}\n\n"
+                f"Diff:\n{diff_text[:MAX_PROMPT_CHARS]}\n\n"
                 "Now review this diff according to the rules above. Return JSON with comments array and summary."
             ),
         },
@@ -161,7 +161,7 @@ def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_descri
             if key not in unique_comments:
                 unique_comments[key] = c
             else:
-                print(f"⚠️  AI duplicate for {key}: {c.get('severity')} - {c.get('message')[:60]}...")
+                print(f"⚠️  AI duplicate for {key}: {c.get('severity')} - {(c.get('message') or '')[:60]}...")
 
         review_data.update(
             {
@@ -174,15 +174,16 @@ def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_descri
         print(f"Valid comments after filtering: {len(review_data['comments'])}")
         return review_data
 
-    except json.JSONDecodeError as e:
-        print(f"JSON parsing failed... {e}")
-        return {"comments": [], "summary": "Review generation encountered a JSON parsing error"}
     except Exception as e:
-        print(f"Review generation failed: {e}")
         import traceback
 
-        traceback.print_exc()
-        return {"comments": [], "summary": "Review generation encountered an error"}
+        error_details = traceback.format_exc()
+        print(f"Review generation failed: {e}\n{error_details}")
+        summary = (
+            f"{ERROR_MARKER}: `{type(e).__name__}`\n\n"
+            f"<details><summary>Debug Info</summary>\n\n```\n{error_details}\n```\n</details>"
+        )
+        return {"comments": [], "summary": summary}
 
 
 def dismiss_previous_reviews(event: Action) -> int:
@@ -219,7 +220,29 @@ 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 = "COMMENT" if any(c.get("severity") not in ["LOW", "SUGGESTION", None] for c in comments) else "APPROVE"
+    summary = review_data.get("summary") or ""
+
+    # Don't approve if error occurred or if there are critical/high severity issues
+    has_error = not summary or ERROR_MARKER in summary
+    has_issues = any(c.get("severity") not in ["LOW", "SUGGESTION", None] for c in comments)
+    requests_changes = any(
+        phrase in summary.lower()
+        for phrase in [
+            "please",
+            "should",
+            "must",
+            "need to",
+            "needs to",
+            "before merging",
+            "fix",
+            "error",
+            "issue",
+            "problem",
+            "warning",
+            "concern",
+        ]
+    )
+    event_type = "COMMENT" if (has_error or has_issues or requests_changes) else "APPROVE"
 
     body = (
         f"## {review_title}\n\n"
@@ -239,12 +262,12 @@ 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]:  # Limit to 10 comments
+    for comment in comments[:10]:  # Limit inline 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', '')[:1000]}"
+        severity = comment.get("severity") or "SUGGESTION"
+        comment_body = f"{EMOJI_MAP.get(severity, '💭')} **{severity}**: {(comment.get('message') or '')[:1000]}"
 
         if suggestion := comment.get("suggestion"):
             suggestion = suggestion[:1000]  # Clip suggestion length
@@ -299,7 +322,7 @@ def main(*args, **kwargs):
     review_number = dismiss_previous_reviews(event)
 
     diff = event.get_pr_diff()
-    review = generate_pr_review(event.repository, diff, event.pr.get("title", ""), event.pr.get("body", ""))
+    review = generate_pr_review(event.repository, diff, event.pr.get("title") or "", event.pr.get("body") or "")
 
     post_review_summary(event, review, review_number)
     print("PR review completed")

actions/summarize_pr.py

@@ -82,7 +82,7 @@ def label_fixed_issues(event, pr_summary):
     if not pr_credit:
         return None
 
-    comment = generate_issue_comment(data["url"], pr_summary, pr_credit, data.get("title", ""))
+    comment = generate_issue_comment(data["url"], pr_summary, pr_credit, data.get("title") or "")
 
     for issue in data["closingIssuesReferences"]["nodes"]:
         number = issue["number"]

actions/summarize_release.py

@@ -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.get("body", "")),
+                    "body": remove_html_comments(pr_data.get("body") or ""),
                     "author": pr_data["user"]["login"],
                     "html_url": pr_data["html_url"],
                     "merged_at": pr_data["merged_at"],

actions/utils/__init__.py

@@ -10,6 +10,7 @@ from .common_utils import (
 )
 from .github_utils import GITHUB_API_URL, GITHUB_GRAPHQL_URL, Action, ultralytics_actions_info
 from .openai_utils import (
+    MAX_PROMPT_CHARS,
     filter_labels,
     get_completion,
     get_pr_open_response,
@@ -21,6 +22,7 @@ from .version_utils import check_pubdev_version, check_pypi_version
 __all__ = (
     "GITHUB_API_URL",
     "GITHUB_GRAPHQL_URL",
+    "MAX_PROMPT_CHARS",
     "REQUESTS_HEADERS",
     "URL_IGNORE_LIST",
     "REDIRECT_START_IGNORE_LIST",

actions/utils/common_utils.py

@@ -133,6 +133,8 @@ URL_PATTERN = re.compile(
 
 def remove_html_comments(body: str) -> str:
     """Removes HTML comments from a string using regex pattern matching."""
+    if not body:
+        return ""
     return re.sub(r"<!--.*?-->", "", body, flags=re.DOTALL).strip()
 
 

actions/utils/github_utils.py

@@ -34,7 +34,6 @@ query($owner: String!, $repo: String!, $pr_number: Int!) {
             closingIssuesReferences(first: 50) { nodes { number } }
             url
             title
-            body
             author { login, __typename }
             reviews(first: 50) { nodes { author { login, __typename } } }
             comments(first: 50) { nodes { author { login, __typename } } }
@@ -126,21 +125,22 @@ class Action:
 
     def _request(self, method: str, url: str, headers=None, expected_status=None, hard=False, **kwargs):
         """Unified request handler with error checking."""
-        response = getattr(requests, method)(url, headers=headers or self.headers, **kwargs)
+        r = getattr(requests, method)(url, headers=headers or self.headers, **kwargs)
         expected = expected_status or self._default_status[method]
-        success = response.status_code in expected
+        success = r.status_code in expected
 
         if self.verbose:
-            print(f"{'✓' if success else '✗'} {method.upper()} {url} → {response.status_code}")
+            elapsed = r.elapsed.total_seconds()
+            print(f"{'✓' if success else '✗'} {method.upper()} {url} → {r.status_code} ({elapsed:.1f}s)")
             if not success:
                 try:
-                    print(f"  ❌ Error: {response.json().get('message', 'Unknown error')}")
+                    print(f"  ❌ Error: {r.json().get('message', 'Unknown error')}")
                 except Exception:
-                    print(f"  ❌ Error: {response.text[:200]}")
+                    print(f"  ❌ Error: {r.text[:200]}")
 
         if not success and hard:
-            response.raise_for_status()
-        return response
+            r.raise_for_status()
+        return r
 
     def get(self, url, **kwargs):
         """Performs GET request with error handling."""

actions/utils/openai_utils.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import os
+import time
 
 import requests
 
@@ -10,6 +11,7 @@ from actions.utils.common_utils import check_links_in_string
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-5-2025-08-07")
+MAX_PROMPT_CHARS = round(128000 * 3.3 * 0.5)  # Max characters for prompt (50% of 128k context)
 SYSTEM_PROMPT_ADDITION = """Guidance:
   - Ultralytics Branding: Use YOLO11, YOLO26, etc., not YOLOv11, YOLOv26 (only older versions like YOLOv10 have a v). Always capitalize "HUB" in "Ultralytics HUB"; use "Ultralytics HUB", not "The Ultralytics HUB". 
   - Avoid Equations: Do not include equations or mathematical notations.
@@ -81,13 +83,8 @@ def get_pr_summary_guidelines() -> str:
 
 def get_pr_summary_prompt(repository: str, diff_text: str) -> tuple[str, bool]:
     """Returns the complete PR summary generation prompt with diff (used by PR update/merge)."""
-    ratio = 3.3  # about 3.3 characters per token
-    limit = round(128000 * ratio * 0.5)  # use up to 50% of the 128k context window for prompt
-
-    prompt = (
-        f"{get_pr_summary_guidelines()}\n\nRepository: '{repository}'\n\nHere's the PR diff:\n\n{diff_text[:limit]}"
-    )
-    return prompt, len(diff_text) > limit
+    prompt = f"{get_pr_summary_guidelines()}\n\nRepository: '{repository}'\n\nHere's the PR diff:\n\n{diff_text[:MAX_PROMPT_CHARS]}"
+    return prompt, len(diff_text) > MAX_PROMPT_CHARS
 
 
 def get_pr_first_comment_template(repository: str) -> str:
@@ -114,61 +111,68 @@ def get_completion(
     response_format: dict = None,
     model: str = OPENAI_MODEL,
 ) -> str | dict:
-    """Generates a completion using OpenAI's Responses API based on input messages."""
+    """Generates a completion using OpenAI's Responses API with retry logic."""
     assert OPENAI_API_KEY, "OpenAI API key is required."
     url = "https://api.openai.com/v1/responses"
     headers = {"Authorization": f"Bearer {OPENAI_API_KEY}", "Content-Type": "application/json"}
     if messages and messages[0].get("role") == "system":
         messages[0]["content"] += "\n\n" + SYSTEM_PROMPT_ADDITION
 
-    max_retries = 2
-    for attempt in range(max_retries + 2):
+    for attempt in range(3):
         data = {"model": model, "input": messages, "store": False, "temperature": temperature}
         if "gpt-5" in model:
             data["reasoning"] = {"effort": reasoning_effort or "low"}
-            # GPT-5 Responses API handles JSON via prompting, not format parameter
 
-        r = requests.post(url, json=data, headers=headers)
-        if r.status_code != 200:
-            print(f"❌ OpenAI error {r.status_code}:\n{r.text}\n")
-        r.raise_for_status()
-        response_data = r.json()
+        try:
+            r = requests.post(url, json=data, headers=headers, timeout=600)
+            success = r.status_code == 200
+            print(f"{'✓' if success else '✗'} POST {url} → {r.status_code} ({r.elapsed.total_seconds():.1f}s)")
+            r.raise_for_status()
+
+            # Parse response
+            content = ""
+            for item in r.json().get("output", []):
+                if item.get("type") == "message":
+                    for c in item.get("content", []):
+                        if c.get("type") == "output_text":
+                            content += c.get("text") or ""
+            content = content.strip()
 
-        content = ""
-        for item in response_data.get("output", []):
-            if item.get("type") == "message":
-                for content_item in item.get("content", []):
-                    if content_item.get("type") == "output_text":
-                        content += content_item.get("text", "")
+            if response_format and response_format.get("type") == "json_object":
+                import json
 
-        content = content.strip()
-        if response_format and response_format.get("type") == "json_object":
-            import json
+                return json.loads(content)
 
-            return json.loads(content)
+            content = remove_outer_codeblocks(content)
+            for x in remove:
+                content = content.replace(x, "")
 
-        content = remove_outer_codeblocks(content)
-        for x in remove:
-            content = content.replace(x, "")
+            # Retry on bad links
+            if attempt < 2 and check_links and not check_links_in_string(content):
+                print("Bad URLs detected, retrying")
+                continue
 
-        if not check_links or check_links_in_string(content):
             return content
 
-        if attempt < max_retries:
-            print(f"Attempt {attempt + 1}: Found bad URLs. Retrying with a new random seed.")
-        else:
-            print("Max retries reached. Updating prompt to exclude links.")
-            messages.append({"role": "user", "content": "Please provide a response without any URLs or links in it."})
-            check_links = False
+        except (requests.exceptions.ConnectionError, requests.exceptions.Timeout):
+            if attempt < 2:
+                print(f"Connection error, retrying in {2**attempt}s")
+                time.sleep(2**attempt)
+                continue
+            raise
+        except requests.exceptions.HTTPError as e:
+            if attempt < 2 and e.response and e.response.status_code >= 500:
+                print(f"Server error {e.response.status_code}, retrying in {2**attempt}s")
+                time.sleep(2**attempt)
+                continue
+            raise
 
     return content
 
 
 def get_pr_open_response(repository: str, diff_text: str, title: str, body: str, available_labels: dict) -> dict:
     """Generates unified PR response with summary, labels, and first comment in a single API call."""
-    ratio = 3.3  # about 3.3 characters per token
-    limit = round(128000 * ratio * 0.5)  # use up to 50% of the 128k context window for prompt
-    is_large = len(diff_text) > limit
+    is_large = len(diff_text) > MAX_PROMPT_CHARS
 
     filtered_labels = filter_labels(available_labels, is_pr=True)
     labels_str = "\n".join(f"- {name}: {description}" for name, description in filtered_labels.items())
@@ -176,7 +180,7 @@ def get_pr_open_response(repository: str, diff_text: str, title: str, body: str,
     prompt = f"""You are processing a new GitHub pull request for the {repository} repository.
 
 Generate 3 outputs in a single JSON response for the PR titled {title} with the following diff:
-{diff_text[:limit]}
+{diff_text[:MAX_PROMPT_CHARS]}
 
 
 --- FIRST JSON OUTPUT (PR SUMMARY) ---

{ultralytics_actions-0.1.4.dist-info → ultralytics_actions-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-actions
-Version: 0.1.4
+Version: 0.1.6
 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.6.dist-info/RECORD

@@ -0,0 +1,19 @@
+actions/__init__.py,sha256=owzYNIk3pmwSP7At9iZuEgZyKtiaKSazh22WkbEyfFo,772
+actions/dispatch_actions.py,sha256=i81UeHrYudAsOUFUfN71u6X-1cmZaZaiiTj6p2rvz8A,4217
+actions/first_interaction.py,sha256=QxPsLjd-m2G-QYOcQb2hQfIB_alupzeZzSHTk-jw0bg,9856
+actions/review_pr.py,sha256=_x-HwX2iqR1xhQkJBDW5CPVq-4Cba7pZZBz63hN9e1Y,15060
+actions/summarize_pr.py,sha256=3nFotiZX42dz-mzDQ9wcoUILJKkcaxrC5EeyxvuvY60,5775
+actions/summarize_release.py,sha256=iCXa9a1DcOrDVe8pMWEsYKgDxuIOhIgMsYymElOLK6o,9083
+actions/update_file_headers.py,sha256=E5fKYLdeW16-BHCcuqxohGpGZqgEh-WX4ZmCQJw2R90,6684
+actions/update_markdown_code_blocks.py,sha256=w3DTRltg2Rmr4-qrNawv_S2vJbheKE0tne1iz79FzXg,8692
+actions/utils/__init__.py,sha256=unjXYIFNFeHrdC8LooDFVWlj6fAdGhssUgASo5229zY,1073
+actions/utils/common_utils.py,sha256=2DRvcyCgmn507w3T4FJcQSZNI9KC1gVUb8CnJqPapD0,11943
+actions/utils/github_utils.py,sha256=cBgEDJBpImTJbGBoZTteVSmCqXPuzEb51np7gRhqPeM,19702
+actions/utils/openai_utils.py,sha256=xI_DZpsEBzXyqQDozMLEtmjwuNlOpNL9n2b-gA6xL5Y,10658
+actions/utils/version_utils.py,sha256=EIbm3iZVNyNl3dh8aNz_9ITeTC93ZxfyUzIRkO3tSXw,3242
+ultralytics_actions-0.1.6.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+ultralytics_actions-0.1.6.dist-info/METADATA,sha256=7Zj67iXEIw61XyAKY8Myu0vZYqsO7F_08XvcDstqhWI,12368
+ultralytics_actions-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ultralytics_actions-0.1.6.dist-info/entry_points.txt,sha256=n_VbDs3Xj33daaeN_2D72UTEuyeH8hVc6-CPH55ymkY,496
+ultralytics_actions-0.1.6.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
+ultralytics_actions-0.1.6.dist-info/RECORD,,

ultralytics_actions-0.1.4.dist-info/RECORD

@@ -1,19 +0,0 @@
-actions/__init__.py,sha256=UDIRZMqICZj8MM9O1637KKavoMhRWu2lgE1IltT0m0s,772
-actions/dispatch_actions.py,sha256=8jaaVkA_LSlpUQ4tuzmQtf2kw3G09uVRD_LmJyXYKNE,4215
-actions/first_interaction.py,sha256=8LxJ0RI4ddm-DjdD5tU1jAljCemP__tAc47nmSP8EdA,9852
-actions/review_pr.py,sha256=hbk2y04tjtsizhgWrYi69btUy8yBdfnIUOe2bODTr28,14410
-actions/summarize_pr.py,sha256=XLYsNTf4J6VPyyecZcuiJaSBDgjDSWFj37v5vb1ATCA,5773
-actions/summarize_release.py,sha256=_067Q5AP-Zdnt_qzhHaCuGCr7T4MXSB5_N-M5GX6qgQ,9081
-actions/update_file_headers.py,sha256=E5fKYLdeW16-BHCcuqxohGpGZqgEh-WX4ZmCQJw2R90,6684
-actions/update_markdown_code_blocks.py,sha256=w3DTRltg2Rmr4-qrNawv_S2vJbheKE0tne1iz79FzXg,8692
-actions/utils/__init__.py,sha256=sKNx6o5jcAraEdGFph0o-YC7dMMY-dg_FprIBa6Jydw,1027
-actions/utils/common_utils.py,sha256=8ZmgaXZU3J2sg-HSaldp3hHYq7bI3akcJHdIXPmcNAo,11908
-actions/utils/github_utils.py,sha256=mexywBlj4_eVuRFnh6-sXpnHTrAffoCUfeKidAqZwqA,19703
-actions/utils/openai_utils.py,sha256=EIu7UQEYUatuU_sYQv-UxsIVveArMGHlpy4nXrI5Kl8,10503
-actions/utils/version_utils.py,sha256=EIbm3iZVNyNl3dh8aNz_9ITeTC93ZxfyUzIRkO3tSXw,3242
-ultralytics_actions-0.1.4.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-ultralytics_actions-0.1.4.dist-info/METADATA,sha256=nsiw0RdHqqVa0-jbY4eI-cvFhOQlxzKkTyTOtxLYcrc,12368
-ultralytics_actions-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ultralytics_actions-0.1.4.dist-info/entry_points.txt,sha256=n_VbDs3Xj33daaeN_2D72UTEuyeH8hVc6-CPH55ymkY,496
-ultralytics_actions-0.1.4.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
-ultralytics_actions-0.1.4.dist-info/RECORD,,