ultralytics-actions 0.1.4__tar.gz → 0.1.5__tar.gz

{ultralytics_actions-0.1.4 → ultralytics_actions-0.1.5}/PKG-INFO

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

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

{ultralytics_actions-0.1.4 → ultralytics_actions-0.1.5}/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

{ultralytics_actions-0.1.4 → ultralytics_actions-0.1.5}/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":

{ultralytics_actions-0.1.4 → ultralytics_actions-0.1.5}/actions/review_pr.py

@@ -5,7 +5,7 @@ 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"
 EMOJI_MAP = {"CRITICAL": "❗", "HIGH": "⚠️", "MEDIUM": "💡", "LOW": "📝", "SUGGESTION": "💭"}
@@ -75,8 +75,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 +117,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 +160,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 +173,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"⚠️ Review generation encountered an error: `{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:
@@ -239,12 +239,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 +299,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")

{ultralytics_actions-0.1.4 → ultralytics_actions-0.1.5}/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"]

{ultralytics_actions-0.1.4 → ultralytics_actions-0.1.5}/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"],

{ultralytics_actions-0.1.4 → ultralytics_actions-0.1.5}/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",

{ultralytics_actions-0.1.4 → ultralytics_actions-0.1.5}/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()
 
 

{ultralytics_actions-0.1.4 → ultralytics_actions-0.1.5}/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 } } }

{ultralytics_actions-0.1.4 → ultralytics_actions-0.1.5}/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,66 @@ 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)
+            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 +178,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 → ultralytics_actions-0.1.5}/ultralytics_actions.egg-info/PKG-INFO

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