ultralytics-actions 0.1.8__py3-none-any.whl → 0.2.0__py3-none-any.whl

actions/__init__.py

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

actions/dispatch_actions.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import os
 import time
 from datetime import datetime
 
@@ -12,47 +13,99 @@ RUN_CI_KEYWORD = "@ultralytics/run-ci"  # and then to merge "@ultralytics/run-ci
 WORKFLOW_FILES = ["ci.yml", "docker.yml"]
 
 
-def get_pr_branch(event) -> str:
-    """Gets the PR branch name."""
+def get_pr_branch(event) -> tuple[str, str | None]:
+    """Gets the PR branch name, creating temp branch for forks, returning (branch, temp_branch_to_delete)."""
+    import subprocess
+    import tempfile
+
     pr_number = event.event_data["issue"]["number"]
     pr_data = event.get_repo_data(f"pulls/{pr_number}")
-    return pr_data.get("head", {}).get("ref", "main")
-
-
-def trigger_and_get_workflow_info(event, branch: str) -> list[dict]:
-    """Triggers workflows and returns their information."""
+    head = pr_data.get("head", {})
+
+    # Check if PR is from a fork
+    is_fork = head.get("repo") and head["repo"]["id"] != pr_data["base"]["repo"]["id"]
+
+    if is_fork:
+        # Create temp branch in base repo by pushing fork code
+        temp_branch = f"temp-ci-{pr_number}-{int(time.time() * 1000)}"
+        fork_repo = head["repo"]["full_name"]
+        fork_branch = head["ref"]
+        base_repo = event.repository
+        token = os.environ.get("GITHUB_TOKEN")
+        if not token:
+            raise ValueError("GITHUB_TOKEN environment variable is not set")
+
+        with tempfile.TemporaryDirectory() as tmp_dir:
+            repo_dir = os.path.join(tmp_dir, "repo")
+            base_url = f"https://x-access-token:{token}@github.com/{base_repo}.git"
+            fork_url = f"https://github.com/{fork_repo}.git"
+
+            try:
+                # Clone base repo (minimal)
+                subprocess.run(["git", "clone", "--depth", "1", base_url, repo_dir], check=True, capture_output=True)
+
+                # Add fork as remote and fetch the PR branch
+                subprocess.run(
+                    ["git", "remote", "add", "fork", fork_url], cwd=repo_dir, check=True, capture_output=True
+                )
+                subprocess.run(
+                    ["git", "fetch", "fork", f"{fork_branch}:{temp_branch}"],
+                    cwd=repo_dir,
+                    check=True,
+                    capture_output=True,
+                )
+
+                # Push temp branch to base repo
+                subprocess.run(["git", "push", "origin", temp_branch], cwd=repo_dir, check=True, capture_output=True)
+            except subprocess.CalledProcessError as e:
+                # Sanitize error output to prevent token leakage
+                stderr = e.stderr.decode() if e.stderr else "No stderr output"
+                stderr = stderr.replace(token, "***TOKEN***")
+                raise RuntimeError(f"Failed to create tmp branch from fork (exit code {e.returncode}): {stderr}") from e
+
+        return temp_branch, temp_branch
+
+    return head.get("ref", "main"), None
+
+
+def trigger_and_get_workflow_info(event, branch: str, temp_branch: str | None = None) -> list[dict]:
+    """Triggers workflows and returns their information, deleting temp branch if provided."""
     repo = event.repository
     results = []
 
-    # Trigger all workflows
-    for file in WORKFLOW_FILES:
-        event.post(f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}/dispatches", json={"ref": branch})
+    try:
+        # Trigger all workflows
+        for file in WORKFLOW_FILES:
+            event.post(f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}/dispatches", json={"ref": branch})
 
-    # Wait for workflows to be created
-    time.sleep(10)
+        # Wait for workflows to be created and start
+        time.sleep(60)
 
-    # Collect information about all workflows
-    for file in WORKFLOW_FILES:
-        # Get workflow name
-        response = event.get(f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}")
-        name = file.replace(".yml", "").title()
-        if response.status_code == 200:
-            name = response.json().get("name", name)
+        # Collect information about all workflows
+        for file in WORKFLOW_FILES:
+            # Get workflow name
+            response = event.get(f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}")
+            name = file.replace(".yml", "").title()
+            if response.status_code == 200:
+                name = response.json().get("name", name)
 
-        # Get run information
-        run_url = f"https://github.com/{repo}/actions/workflows/{file}"
-        run_number = None
+            # Get run information
+            run_url = f"https://github.com/{repo}/actions/workflows/{file}"
+            run_number = None
 
-        runs_response = event.get(
-            f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}/runs?branch={branch}&event=workflow_dispatch&per_page=1"
-        )
+            runs_response = event.get(
+                f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}/runs?branch={branch}&event=workflow_dispatch&per_page=1"
+            )
 
-        if runs_response.status_code == 200:
-            if runs := runs_response.json().get("workflow_runs", []):
+            if runs_response.status_code == 200 and (runs := runs_response.json().get("workflow_runs", [])):
                 run_url = runs[0].get("html_url", run_url)
                 run_number = runs[0].get("run_number")
 
-        results.append({"name": name, "file": file, "url": run_url, "run_number": run_number})
+            results.append({"name": name, "file": file, "url": run_url, "run_number": run_number})
+    finally:
+        # Always delete temp branch even if workflow collection fails
+        if temp_branch:
+            event.delete(f"{GITHUB_API_URL}/repos/{repo}/git/refs/heads/{temp_branch}")
 
     return results
 
@@ -63,12 +116,15 @@ def update_comment(event, comment_body: str, triggered_actions: list[dict], bran
         return
 
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S UTC")
-    summary = (
-        f"\n\n## ⚡ Actions Trigger\n\n"
-        f"<sub>Made with ❤️ by [Ultralytics Actions](https://www.ultralytics.com/actions)<sub>\n\n"
-        f"GitHub Actions below triggered via workflow dispatch on this "
-        f"PR branch `{branch}` at {timestamp} with `{RUN_CI_KEYWORD}` command:\n\n"
-    )
+    summary = f"""
+
+## ⚡ Actions Trigger
+
+<sub>Made with ❤️ by [Ultralytics Actions](https://www.ultralytics.com/actions)<sub>
+
+GitHub Actions below triggered via workflow dispatch for this PR at {timestamp} with `{RUN_CI_KEYWORD}` command:
+
+"""
 
     for action in triggered_actions:
         run_info = f" run {action['run_number']}" if action["run_number"] else ""
@@ -104,10 +160,10 @@ def main(*args, **kwargs):
 
     # Get branch, trigger workflows, and update comment
     event.toggle_eyes_reaction(True)
-    branch = get_pr_branch(event)
-    print(f"Triggering workflows on branch: {branch}")
+    branch, temp_branch = get_pr_branch(event)
+    print(f"Triggering workflows on branch: {branch}" + (" (temp)" if temp_branch else ""))
 
-    triggered_actions = trigger_and_get_workflow_info(event, branch)
+    triggered_actions = trigger_and_get_workflow_info(event, branch, temp_branch)
     update_comment(event, comment_body, triggered_actions, branch)
     event.toggle_eyes_reaction(False)
 

actions/first_interaction.py

@@ -6,11 +6,9 @@ import os
 import time
 
 from . import review_pr
-from .utils import Action, filter_labels, get_completion, get_pr_open_response, remove_html_comments
+from .summarize_pr import SUMMARY_MARKER
+from .utils import ACTIONS_CREDIT, 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"
 
@@ -196,7 +194,7 @@ def main(*args, **kwargs):
 
         if summary := response.get("summary"):
             print("Updating PR description with summary...")
-            event.update_pr_description(number, SUMMARY_START + summary)
+            event.update_pr_description(number, f"{SUMMARY_MARKER}\n\n{ACTIONS_CREDIT}\n\n{summary}")
         else:
             summary = body
 

actions/review_pr.py

@@ -5,9 +5,9 @@ from __future__ import annotations
 import json
 import re
 
-from .utils import GITHUB_API_URL, MAX_PROMPT_CHARS, Action, get_completion, remove_html_comments
+from .utils import ACTIONS_CREDIT, GITHUB_API_URL, MAX_PROMPT_CHARS, Action, get_completion, remove_html_comments
 
-REVIEW_MARKER = "🔍 PR Review"
+REVIEW_MARKER = "## 🔍 PR Review"
 ERROR_MARKER = "⚠️ Review generation encountered an error"
 EMOJI_MAP = {"CRITICAL": "❗", "HIGH": "⚠️", "MEDIUM": "💡", "LOW": "📝", "SUGGESTION": "💭"}
 SKIP_PATTERNS = [
@@ -30,9 +30,10 @@ SKIP_PATTERNS = [
 ]
 
 
-def parse_diff_files(diff_text: str) -> dict:
-    """Parse diff to extract file paths, valid line numbers, and line content for comments (both sides)."""
+def parse_diff_files(diff_text: str) -> tuple[dict, str]:
+    """Parse diff and return file mapping with line numbers AND augmented diff with explicit line numbers."""
     files, current_file, new_line, old_line = {}, None, 0, 0
+    augmented_lines = []
 
     for line in diff_text.split("\n"):
         if line.startswith("diff --git"):
@@ -41,23 +42,31 @@ def parse_diff_files(diff_text: str) -> dict:
             new_line, old_line = 0, 0
             if current_file:
                 files[current_file] = {"RIGHT": {}, "LEFT": {}}
+            augmented_lines.append(line)
         elif line.startswith("@@") and current_file:
-            # Extract both old and new line numbers
             match = re.search(r"@@ -(\d+)(?:,\d+)? \+(\d+)(?:,\d+)?", line)
             if match:
                 old_line, new_line = int(match.group(1)), int(match.group(2))
+            augmented_lines.append(line)
         elif current_file and (new_line > 0 or old_line > 0):
             if line.startswith("+") and not line.startswith("+++"):
-                files[current_file]["RIGHT"][new_line] = line[1:]  # Added line (right/new side)
+                files[current_file]["RIGHT"][new_line] = line[1:]
+                augmented_lines.append(f"R{new_line:>5} {line}")  # Prefix with RIGHT line number
                 new_line += 1
             elif line.startswith("-") and not line.startswith("---"):
-                files[current_file]["LEFT"][old_line] = line[1:]  # Removed line (left/old side)
+                files[current_file]["LEFT"][old_line] = line[1:]
+                augmented_lines.append(f"L{old_line:>5} {line}")  # Prefix with LEFT line number
                 old_line += 1
-            elif not line.startswith("\\"):  # Context line (ignore "No newline" markers)
+            elif not line.startswith("\\"):
+                augmented_lines.append(f"       {line}")  # Context line, no number
                 new_line += 1
                 old_line += 1
+            else:
+                augmented_lines.append(line)
+        else:
+            augmented_lines.append(line)
 
-    return files
+    return files, "\n".join(augmented_lines)
 
 
 def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_description: str) -> dict:
@@ -65,7 +74,7 @@ def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_descri
     if not diff_text:
         return {"comments": [], "summary": "No changes detected in diff"}
 
-    diff_files = parse_diff_files(diff_text)
+    diff_files, augmented_diff = parse_diff_files(diff_text)
     if not diff_files:
         return {"comments": [], "summary": "No files with changes detected in diff"}
 
@@ -82,45 +91,45 @@ 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())
-    diff_truncated = len(diff_text) > MAX_PROMPT_CHARS
+    diff_truncated = len(augmented_diff) > MAX_PROMPT_CHARS
     lines_changed = sum(len(sides["RIGHT"]) + len(sides["LEFT"]) for sides in diff_files.values())
 
     content = (
-        "You are an expert code reviewer for Ultralytics. Provide detailed inline comments on specific code changes.\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"
+        "You are an expert code reviewer for Ultralytics. Review the code changes and provide inline comments where you identify issues or opportunities for improvement.\n\n"
+        "Focus on: bugs, security vulnerabilities, performance issues, best practices, edge cases, error handling, and code clarity.\n\n"
         "CRITICAL RULES:\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"
+        "1. Provide balanced, constructive feedback - flag bugs, improvements, and best practice issues\n"
+        "2. For issues spanning multiple adjacent lines, use 'start_line' to create ONE multi-line comment, never separate comments\n"
+        "3. Combine related issues into a single comment when they stem from the same root cause\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"
+        "6. Use backticks for code: `x=3`, `file.py`, `function()`\n"
+        "7. 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"
-        "- 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"
+        "- Provide 'suggestion' field with ready-to-merge code when you can confidently fix the issue\n"
+        "- Suggestions must be complete, working code with NO comments, placeholders, or explanations\n"
+        "- For single-line fixes: provide 'suggestion' without 'start_line' to replace the line at 'line'\n"
+        "- Do not provide multi-line fixes: suggestions should only be single line\n"
+        "- Match the exact indentation of the original code\n"
+        "- Avoid triple backticks (```) in suggestions as they break markdown formatting\n\n"
         "LINE NUMBERS:\n"
-        "- You MUST extract line numbers directly from the @@ hunk headers in the diff below\n"
-        "- RIGHT (added +): Find @@ lines, use numbers after +N (e.g., @@ -10,5 +20,7 @@ means RIGHT starts at line 20)\n"
-        "- LEFT (removed -): Find @@ lines, use numbers after -N (e.g., @@ -10,5 +20,7 @@ means LEFT starts at line 10)\n"
-        "- Count forward from hunk start: + lines increment RIGHT, - lines increment LEFT, context lines increment both\n"
-        "- CRITICAL: Using line numbers not in the diff will cause your comment to be rejected\n"
-        "- Suggestions only work on RIGHT (added) lines, never on LEFT (removed) lines\n\n"
+        "- Each line in the diff is prefixed with its line number for clarity:\n"
+        "  R  123 +added code     <- RIGHT side (new file), line 123\n"
+        "  L   45 -removed code   <- LEFT side (old file), line 45\n"
+        "         context line    <- context (no number needed)\n"
+        "- Extract the number after R or L prefix to get the exact line number\n"
+        "- Use 'side': 'RIGHT' for R-prefixed lines, 'side': 'LEFT' for L-prefixed lines\n"
+        "- Suggestions only work on RIGHT lines, never on LEFT lines\n"
+        "- CRITICAL: Only use line numbers that you see explicitly prefixed in the diff\n\n"
         "Return JSON: "
         '{"comments": [{"file": "exact/path", "line": N, "side": "RIGHT", "severity": "HIGH", "message": "..."}], "summary": "..."}\n\n'
         "Rules:\n"
-        "- Verify line numbers from @@ hunks: +N for RIGHT (added), -N for LEFT (removed)\n"
-        "- Exact paths (no ./), 'side' field defaults to RIGHT if omitted\n"
+        "- Extract line numbers from R#### or L#### prefixes in the diff\n"
+        "- Exact paths (no ./), 'side' field must match R (RIGHT) or L (LEFT) prefix\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"- Files changed: {len(file_list)} ({', '.join(file_list[:30])}{'...' if len(file_list) > 30 else ''})\n"
         f"- Lines changed: {lines_changed}\n"
     )
 
@@ -129,17 +138,21 @@ def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_descri
         {
             "role": "user",
             "content": (
-                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[:MAX_PROMPT_CHARS]}\n\n"
+                f"Review this PR in https://github.com/{repository}:\n\n"
+                f"TITLE:\n{pr_title}\n\n"
+                f"BODY:\n{remove_html_comments(pr_description or '')[:1000]}\n\n"
+                f"DIFF:\n{augmented_diff[:MAX_PROMPT_CHARS]}\n\n"
                 "Now review this diff according to the rules above. Return JSON with comments array and summary."
             ),
         },
     ]
 
+    # Debug output
+    # print(f"\nSystem prompt (first 3000 chars):\n{messages[0]['content'][:3000]}...\n")
+    # print(f"\nUser prompt (first 3000 chars):\n{messages[1]['content'][:3000]}...\n")
+
     try:
-        response = get_completion(messages, reasoning_effort="medium", model="gpt-5-codex")
+        response = get_completion(messages, reasoning_effort="low", model="gpt-5-codex")
 
         json_str = re.search(r"```(?:json)?\s*(\{.*?\})\s*```", response, re.DOTALL)
         review_data = json.loads(json_str.group(1) if json_str else response)
@@ -161,20 +174,14 @@ def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_descri
                 print(f"Filtered out {file_path}:{line_num} (file not in diff)")
                 continue
             if line_num not in diff_files[file_path].get(side, {}):
-                # Try other side if not found
-                other_side = "LEFT" if side == "RIGHT" else "RIGHT"
-                if line_num in diff_files[file_path].get(other_side, {}):
-                    print(f"Switching {file_path}:{line_num} from {side} to {other_side}")
-                    c["side"] = other_side
-                    side = other_side
-                    # GitHub rejects suggestions on removed lines
-                    if side == "LEFT" and c.get("suggestion"):
-                        print(f"Dropping suggestion for {file_path}:{line_num} - LEFT side doesn't support suggestions")
-                        c.pop("suggestion", None)
-                else:
-                    available = {s: list(diff_files[file_path][s].keys())[:10] for s in ["RIGHT", "LEFT"]}
-                    print(f"Filtered out {file_path}:{line_num} (available: {available})")
-                    continue
+                available = {s: list(diff_files[file_path][s].keys())[:10] for s in ["RIGHT", "LEFT"]}
+                print(f"Filtered out {file_path}:{line_num} (side={side}, available: {available})")
+                continue
+
+            # GitHub rejects suggestions on removed lines
+            if side == "LEFT" and c.get("suggestion"):
+                print(f"Dropping suggestion for {file_path}:{line_num} - LEFT side doesn't support suggestions")
+                c.pop("suggestion", None)
 
             # Validate start_line if provided - drop start_line for suggestions (single-line only)
             if start_line:
@@ -262,9 +269,9 @@ def post_review_summary(event: Action, review_data: dict, review_number: int) ->
     event_type = "COMMENT" if (has_error or has_inline_comments or has_issues) 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')[:1000]}\n\n"  # Clip summary length
+        f"{review_title}\n\n"
+        f"{ACTIONS_CREDIT}\n\n"
+        f"{review_data.get('summary', 'Review completed')[:3000]}\n\n"  # Clip summary length
     )
 
     if comments:
@@ -284,13 +291,11 @@ def post_review_summary(event: Action, review_data: dict, review_number: int) ->
             continue
 
         severity = comment.get("severity") or "SUGGESTION"
-        comment_body = f"{EMOJI_MAP.get(severity, '💭')} **{severity}**: {(comment.get('message') or '')[:1000]}"
-
-        # Get side (LEFT for removed lines, RIGHT for added lines)
         side = comment.get("side", "RIGHT")
+        comment_body = f"{EMOJI_MAP.get(severity, '💭')} **{severity}**: {(comment.get('message') or '')[:3000]}"
 
         if suggestion := comment.get("suggestion"):
-            suggestion = suggestion[:1000]  # Clip suggestion length
+            suggestion = suggestion[:3000]  # 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(side, {}).get(line):
@@ -300,19 +305,16 @@ def post_review_summary(event: Action, review_data: dict, review_number: int) ->
 
         # Build comment with optional start_line for multi-line context
         review_comment = {"path": file_path, "line": line, "body": comment_body, "side": side}
-        if start_line := comment.get("start_line"):
-            if start_line < line:
-                review_comment["start_line"] = start_line
-                review_comment["start_side"] = side
-                print(f"Multi-line comment: {file_path}:{start_line}-{line} ({side})")
+        if (start_line := comment.get("start_line")) and start_line < line:
+            review_comment["start_line"] = start_line
+            review_comment["start_side"] = side
 
         review_comments.append(review_comment)
 
     # Submit review with inline comments
-    payload = {"commit_id": commit_sha, "body": body, "event": event_type}
+    payload = {"commit_id": commit_sha, "body": body.strip(), "event": event_type}
     if review_comments:
         payload["comments"] = review_comments
-        print(f"Posting review with {len(review_comments)} inline comments")
 
     event.post(
         f"{GITHUB_API_URL}/repos/{event.repository}/pulls/{pr_number}/reviews",

actions/summarize_pr.py

@@ -2,11 +2,9 @@
 
 from __future__ import annotations
 
-from .utils import GITHUB_API_URL, Action, get_completion, get_pr_summary_prompt
+from .utils import ACTIONS_CREDIT, GITHUB_API_URL, Action, get_completion, get_pr_summary_prompt
 
-SUMMARY_START = (
-    "## 🛠️ PR Summary\n\n<sub>Made with ❤️ by [Ultralytics Actions](https://github.com/ultralytics/actions)<sub>\n\n"
-)
+SUMMARY_MARKER = "## 🛠️ PR Summary"
 
 
 def generate_merge_message(pr_summary, pr_credit, pr_url):
@@ -73,7 +71,8 @@ def generate_pr_summary(repository, diff_text):
     reply = get_completion(messages, temperature=1.0)
     if is_large:
         reply = "**WARNING ⚠️** this PR is very large, summary may not cover all changes.\n\n" + reply
-    return SUMMARY_START + reply
+
+    return f"{SUMMARY_MARKER}\n\n{ACTIONS_CREDIT}\n\n{reply}"
 
 
 def label_fixed_issues(event, pr_summary):

actions/summarize_release.py

@@ -196,7 +196,7 @@ def main(*args, **kwargs):
     try:
         summary = generate_release_summary(event, diff, prs, CURRENT_TAG, previous_tag)
     except Exception as e:
-        print(f"Failed to generate summary: {str(e)}")
+        print(f"Failed to generate summary: {e}")
         summary = "Failed to generate summary."
 
     # Get the latest commit message

actions/utils/__init__.py

@@ -1,6 +1,7 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
 from .common_utils import (
+    ACTIONS_CREDIT,
     REDIRECT_END_IGNORE_LIST,
     REDIRECT_START_IGNORE_LIST,
     REQUESTS_HEADERS,
@@ -20,13 +21,14 @@ from .openai_utils import (
 from .version_utils import check_pubdev_version, check_pypi_version
 
 __all__ = (
+    "ACTIONS_CREDIT",
     "GITHUB_API_URL",
     "GITHUB_GRAPHQL_URL",
     "MAX_PROMPT_CHARS",
+    "REDIRECT_END_IGNORE_LIST",
+    "REDIRECT_START_IGNORE_LIST",
     "REQUESTS_HEADERS",
     "URL_IGNORE_LIST",
-    "REDIRECT_START_IGNORE_LIST",
-    "REDIRECT_END_IGNORE_LIST",
     "Action",
     "allow_redirect",
     "check_pubdev_version",

actions/utils/common_utils.py

@@ -23,6 +23,7 @@ REQUESTS_HEADERS = {
     "Sec-Fetch-User": "?1",
     "Sec-Fetch-Dest": "document",
 }
+ACTIONS_CREDIT = "<sub>Made with ❤️ by [Ultralytics Actions](https://www.ultralytics.com/actions)</sub>"
 BAD_HTTP_CODES = frozenset(
     {
         204,  # No content

actions/utils/github_utils.py

@@ -101,7 +101,13 @@ mutation($labelableId: ID!, $labelIds: [ID!]!) {
 class Action:
     """Handles GitHub Actions API interactions and event processing."""
 
-    def __init__(self, token: str = None, event_name: str = None, event_data: dict = None, verbose: bool = True):
+    def __init__(
+        self,
+        token: str | None = None,
+        event_name: str | None = None,
+        event_data: dict | None = None,
+        verbose: bool = True,
+    ):
         """Initializes a GitHub Actions API handler with token and event data for processing events."""
         self.token = token or os.getenv("GITHUB_TOKEN")
         self.event_name = event_name or os.getenv("GITHUB_EVENT_NAME")
@@ -116,8 +122,8 @@ class Action:
         self._pr_summary_cache = None
         self._username_cache = None
         self._default_status = {
-            "get": [200],
-            "post": [200, 201],
+            "get": [200, 204],
+            "post": [200, 201, 204],
             "put": [200, 201, 204],
             "patch": [200],
             "delete": [200, 204],
@@ -131,12 +137,15 @@ class Action:
 
         if self.verbose:
             elapsed = r.elapsed.total_seconds()
-            print(f"{'✓' if success else '✗'} {method.upper()} {url} → {r.status_code} ({elapsed:.1f}s)")
+            print(f"{'✓' if success else '✗'} {method.upper()} {url} → {r.status_code} ({elapsed:.1f}s)", flush=True)
             if not success:
                 try:
-                    print(f"  ❌ Error: {r.json().get('message', 'Unknown error')}")
+                    error_data = r.json()
+                    print(f"  ❌ Error: {error_data.get('message', 'Unknown error')}")
+                    if errors := error_data.get("errors"):
+                        print(f"  Details: {errors}")
                 except Exception:
-                    print(f"  ❌ Error: {r.text[:200]}")
+                    print(f"  ❌ Error: {r.text[:1000]}")
 
         if not success and hard:
             r.raise_for_status()
@@ -257,7 +266,7 @@ class Action:
             self.delete(f"{url}/{self.eyes_reaction_id}")
             self.eyes_reaction_id = None
 
-    def graphql_request(self, query: str, variables: dict = None) -> dict:
+    def graphql_request(self, query: str, variables: dict | None = None) -> dict:
         """Executes a GraphQL query against the GitHub API."""
         result = self.post(GITHUB_GRAPHQL_URL, json={"query": query, "variables": variables}).json()
         if "data" not in result or result.get("errors"):
@@ -331,7 +340,9 @@ class Action:
         else:
             self.post(f"{GITHUB_API_URL}/repos/{self.repository}/issues/{number}/comments", json={"body": comment})
 
-    def update_content(self, number: int, node_id: str, issue_type: str, title: str = None, body: str = None):
+    def update_content(
+        self, number: int, node_id: str, issue_type: str, title: str | None = None, body: str | None = None
+    ):
         """Updates the title and/or body of an issue, pull request, or discussion."""
         if issue_type == "discussion":
             variables = {"discussionId": node_id}
@@ -373,7 +384,7 @@ class Action:
     def handle_alert(self, number: int, node_id: str, issue_type: str, username: str, block: bool = False):
         """Handles content flagged as alert: updates content, locks, optionally closes and blocks user."""
         new_title = "Content Under Review"
-        new_body = """This post has been flagged for review by [Ultralytics Actions](https://ultralytics.com/actions) due to possible spam, abuse, or off-topic content. For more information please see our:
+        new_body = """This post has been flagged for review by [Ultralytics Actions](https://www.ultralytics.com/actions) due to possible spam, abuse, or off-topic content. For more information please see our:
 
 - [Code of Conduct](https://docs.ultralytics.com/help/code-of-conduct/)
 - [Security Policy](https://docs.ultralytics.com/help/security/)

actions/utils/openai_utils.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import json
 import os
 import time
 
@@ -12,6 +13,12 @@ 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)
+MODEL_COSTS = {
+    "gpt-5-codex": (1.25, 10.00),
+    "gpt-5-2025-08-07": (1.25, 10.00),
+    "gpt-5-nano-2025-08-07": (0.05, 0.40),
+    "gpt-5-mini-2025-08-07": (0.25, 2.00),
+}
 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.
@@ -34,7 +41,7 @@ def remove_outer_codeblocks(string):
     return string
 
 
-def filter_labels(available_labels: dict, current_labels: list = None, is_pr: bool = False) -> dict:
+def filter_labels(available_labels: dict, current_labels: list | None = None, is_pr: bool = False) -> dict:
     """Filters labels by removing manually-assigned and mutually exclusive labels."""
     current_labels = current_labels or []
     filtered = available_labels.copy()
@@ -107,8 +114,8 @@ def get_completion(
     check_links: bool = True,
     remove: list[str] = (" @giscus[bot]",),
     temperature: float = 1.0,
-    reasoning_effort: str = None,
-    response_format: dict = None,
+    reasoning_effort: str | None = None,
+    response_format: dict | None = None,
     model: str = OPENAI_MODEL,
 ) -> str | dict:
     """Generates a completion using OpenAI's Responses API with retry logic."""
@@ -124,23 +131,46 @@ def get_completion(
             data["reasoning"] = {"effort": reasoning_effort or "low"}
 
         try:
-            r = requests.post(url, json=data, headers=headers, timeout=600)
+            r = requests.post(url, json=data, headers=headers, timeout=(30, 900))
+            elapsed = r.elapsed.total_seconds()
             success = r.status_code == 200
-            print(f"{'✓' if success else '✗'} POST {url} → {r.status_code} ({r.elapsed.total_seconds():.1f}s)")
+            print(f"{'✓' if success else '✗'} POST {url} → {r.status_code} ({elapsed:.1f}s)")
+
+            # Retry server errors
+            if attempt < 2 and r.status_code >= 500:
+                print(f"Retrying {r.status_code} in {2**attempt}s (attempt {attempt + 1}/3)...")
+                time.sleep(2**attempt)
+                continue
+
             r.raise_for_status()
 
             # Parse response
+            response_json = r.json()
             content = ""
-            for item in r.json().get("output", []):
+            for item in response_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()
 
-            if response_format and response_format.get("type") == "json_object":
-                import json
+            # Extract and print token usage
+            if usage := response_json.get("usage"):
+                input_tokens = usage.get("input_tokens", 0)
+                output_tokens = usage.get("output_tokens", 0)
+                thinking_tokens = (usage.get("output_tokens_details") or {}).get("reasoning_tokens", 0)
 
+                # Calculate cost
+                costs = MODEL_COSTS.get(model, (0.0, 0.0))
+                cost = (input_tokens * costs[0] + output_tokens * costs[1]) / 1e6
+
+                # Format summary
+                token_str = f"{input_tokens}→{output_tokens - thinking_tokens}"
+                if thinking_tokens:
+                    token_str += f" (+{thinking_tokens} thinking)"
+                print(f"{model} ({token_str} = {input_tokens + output_tokens} tokens, ${cost:.5f}, {elapsed:.1f}s)")
+
+            if response_format and response_format.get("type") == "json_object":
                 return json.loads(content)
 
             content = remove_outer_codeblocks(content)
@@ -154,17 +184,13 @@ def get_completion(
 
             return content
 
-        except (requests.exceptions.ConnectionError, requests.exceptions.Timeout):
+        except (requests.exceptions.ConnectionError, requests.exceptions.Timeout, json.JSONDecodeError) as e:
             if attempt < 2:
-                print(f"Connection error, retrying in {2**attempt}s")
+                print(f"Retrying {e.__class__.__name__} in {2**attempt}s (attempt {attempt + 1}/3)...")
                 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
+        except requests.exceptions.HTTPError:  # 4xx errors
             raise
 
     return content

{ultralytics_actions-0.1.8.dist-info → ultralytics_actions-0.2.0.dist-info}/METADATA

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

@@ -0,0 +1,19 @@
+actions/__init__.py,sha256=Ms2e2jGfy0slxc1dw9maEddDHLCkempqURchc3G3TRw,772
+actions/dispatch_actions.py,sha256=ljlFR1o8m1qTHbStsJJVMVDdJv7iVqMfdPzKlZyKXl8,6743
+actions/first_interaction.py,sha256=c8I6trXvsgii3B3k_HFWOYmLqbyz_oimHD2BeTGPoUM,9795
+actions/review_pr.py,sha256=JSAHyqoVty6Ob8_08zfllCynI6cY_3rcZT6ZLz5OHFU,17041
+actions/summarize_pr.py,sha256=0y4Cl4_ZMMtDWVhxwWasn3mHo_4GCnegJrf29yujUYM,5715
+actions/summarize_release.py,sha256=8D5EOQ36mho1HKtWD2J-IDH_xJJb3q0shgXZSdemmDM,9078
+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=Uf7S5qYHS59zoAP9uKVIZwhpUbgyI947dD9jAWu50Lg,1115
+actions/utils/common_utils.py,sha256=InBc-bsXcwzQYjuDxtrrm3bj7J-70U54G0s2nQKgCg8,12052
+actions/utils/github_utils.py,sha256=5yzNIiu7-WBmH1-gSi4O31m1Fwd4k8pfbwM2BPVGf88,19989
+actions/utils/openai_utils.py,sha256=gblrRoWND0AMlN9AYmMxIPAnppK9ZWrsYR5547_5zUU,11839
+actions/utils/version_utils.py,sha256=EIbm3iZVNyNl3dh8aNz_9ITeTC93ZxfyUzIRkO3tSXw,3242
+ultralytics_actions-0.2.0.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+ultralytics_actions-0.2.0.dist-info/METADATA,sha256=CbtP6vJwbnRoElLzOPgYDRYMSZsEEZGUjHyZvkWWlb4,12368
+ultralytics_actions-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ultralytics_actions-0.2.0.dist-info/entry_points.txt,sha256=n_VbDs3Xj33daaeN_2D72UTEuyeH8hVc6-CPH55ymkY,496
+ultralytics_actions-0.2.0.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
+ultralytics_actions-0.2.0.dist-info/RECORD,,

ultralytics_actions-0.1.8.dist-info/RECORD

@@ -1,19 +0,0 @@
-actions/__init__.py,sha256=iaUhZH1t2gpoOoqwPRBkP7bO6z4Xd3R_JPDt8VxcO2U,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=6svsUPJTH4FSVfq4yQ4Y_PuwcEy522hmYSTOtDZbBKc,17371
-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.8.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-ultralytics_actions-0.1.8.dist-info/METADATA,sha256=dSciUEZ62eiZNHZS60RdWLla4u7Qt_81RSBqDR7xSNs,12368
-ultralytics_actions-0.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ultralytics_actions-0.1.8.dist-info/entry_points.txt,sha256=n_VbDs3Xj33daaeN_2D72UTEuyeH8hVc6-CPH55ymkY,496
-ultralytics_actions-0.1.8.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
-ultralytics_actions-0.1.8.dist-info/RECORD,,