ultralytics-actions 0.1.9__tar.gz → 0.2.0__tar.gz

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/PKG-INFO

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

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

ultralytics_actions-0.2.0/actions/dispatch_actions.py

@@ -0,0 +1,172 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+from __future__ import annotations
+
+import os
+import time
+from datetime import datetime
+
+from .utils import GITHUB_API_URL, Action
+
+# Configuration
+RUN_CI_KEYWORD = "@ultralytics/run-ci"  # and then to merge "@ultralytics/run-ci-and-merge"
+WORKFLOW_FILES = ["ci.yml", "docker.yml"]
+
+
+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}")
+    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 = []
+
+    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 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)
+
+            # 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"
+            )
+
+            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})
+    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
+
+
+def update_comment(event, comment_body: str, triggered_actions: list[dict], branch: str):
+    """Updates the comment with workflow information."""
+    if not triggered_actions:
+        return
+
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S UTC")
+    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 ""
+        summary += f"* ✅ [{action['name']}]({action['url']}): `{action['file']}`{run_info}\n"
+
+    new_body = comment_body.replace(RUN_CI_KEYWORD, summary).strip()
+    comment_id = event.event_data["comment"]["id"]
+    event.patch(f"{GITHUB_API_URL}/repos/{event.repository}/issues/comments/{comment_id}", json={"body": new_body})
+
+
+def main(*args, **kwargs):
+    """Handles triggering workflows from PR comments."""
+    event = Action(*args, **kwargs)
+
+    # Only process new comments on PRs
+    if (
+        event.event_name != "issue_comment"
+        or "pull_request" not in event.event_data.get("issue", {})
+        or event.event_data.get("action") != "created"
+    ):
+        return
+
+    # Get comment info
+    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
+    has_keyword = RUN_CI_KEYWORD in comment_body and comment_body.count(RUN_CI_KEYWORD) > comment_body.count(
+        f"`{RUN_CI_KEYWORD}`"
+    )
+    if not has_keyword or not event.is_org_member(username):
+        return
+
+    # Get branch, trigger workflows, and update comment
+    event.toggle_eyes_reaction(True)
+    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, temp_branch)
+    update_comment(event, comment_body, triggered_actions, branch)
+    event.toggle_eyes_reaction(False)
+
+
+if __name__ == "__main__":
+    main()

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/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
 

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/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 = [
@@ -95,27 +95,25 @@ def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_descri
     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"
         "- 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"
@@ -131,7 +129,7 @@ def generate_pr_review(repository: str, diff_text: str, pr_title: str, pr_descri
         "- 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"
     )
 
@@ -140,18 +138,18 @@ 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{augmented_diff[: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: print prompts sent to AI
-    # print(f"\nSystem prompt (first 1000 chars):\n{messages[0]['content'][:2000]}...\n")
-    # print(f"\nUser prompt (first 1000 chars):\n{messages[1]['content'][:2000]}...\n")
+    # 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="low", model="gpt-5-codex")
@@ -271,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:
@@ -294,10 +292,10 @@ def post_review_summary(event: Action, review_data: dict, review_number: int) ->
 
         severity = comment.get("severity") or "SUGGESTION"
         side = comment.get("side", "RIGHT")
-        comment_body = f"{EMOJI_MAP.get(severity, '💭')} **{severity}**: {(comment.get('message') or '')[:1000]}"
+        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):
@@ -307,10 +305,9 @@ 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
+        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)
 

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/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):

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/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

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/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",

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/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

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/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],
@@ -134,9 +140,12 @@ class Action:
             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/)

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/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,18 +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:
-            status_code = getattr(e.response, "status_code", 0) if e.response else 0
-            if attempt < 2 and status_code >= 500:
-                print(f"Server error {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.9 → ultralytics_actions-0.2.0}/tests/test_dispatch_actions.py

@@ -16,14 +16,39 @@ def test_get_pr_branch():
     """Test getting PR branch name."""
     mock_event = MagicMock()
     mock_event.event_data = {"issue": {"number": 123}}
-    mock_event.get_repo_data.return_value = {"head": {"ref": "feature-branch"}}
+    mock_event.get_repo_data.return_value = {
+        "head": {"ref": "feature-branch", "repo": {"id": 1}},
+        "base": {"repo": {"id": 1}},
+    }
 
-    branch = get_pr_branch(mock_event)
+    branch, temp_branch = get_pr_branch(mock_event)
 
     assert branch == "feature-branch"
+    assert temp_branch is None
     mock_event.get_repo_data.assert_called_once_with("pulls/123")
 
 
+def test_get_pr_branch_fork():
+    """Test getting PR branch name for fork PRs."""
+    mock_event = MagicMock()
+    mock_event.event_data = {"issue": {"number": 456}}
+    mock_event.repository = "base/repo"
+    mock_event.get_repo_data.return_value = {
+        "head": {"ref": "fork-branch", "sha": "abc123", "repo": {"id": 2, "full_name": "fork/repo"}},
+        "base": {"repo": {"id": 1}},
+    }
+
+    with patch("time.time", return_value=1234567.890):
+        with patch("subprocess.run") as mock_run:
+            with patch("os.environ.get", return_value="test-token"):
+                branch, temp_branch = get_pr_branch(mock_event)
+
+    assert branch == "temp-ci-456-1234567890"
+    assert temp_branch == "temp-ci-456-1234567890"
+    # Verify git commands were called
+    assert mock_run.call_count == 4  # clone, remote add, fetch, push
+
+
 def test_trigger_and_get_workflow_info():
     """Test triggering workflows and getting info."""
     mock_event = MagicMock()
@@ -54,9 +79,8 @@ def test_trigger_and_get_workflow_info():
     mock_event.get.side_effect = get_side_effect
 
     # Use patch to skip time.sleep and limit to one workflow
-    with patch("time.sleep"):
-        with patch("actions.dispatch_actions.WORKFLOW_FILES", ["ci.yml"]):
-            results = trigger_and_get_workflow_info(mock_event, "feature-branch")
+    with patch("time.sleep"), patch("actions.dispatch_actions.WORKFLOW_FILES", ["ci.yml"]):
+        results = trigger_and_get_workflow_info(mock_event, "feature-branch")
 
     # Check results
     assert len(results) == 1
@@ -64,6 +88,38 @@ def test_trigger_and_get_workflow_info():
     assert results[0]["run_number"] == 42
 
 
+def test_trigger_and_get_workflow_info_with_temp_branch():
+    """Test that temp branches are deleted after triggering workflows."""
+    mock_event = MagicMock()
+    mock_event.repository = "test/repo"
+
+    workflow_response = MagicMock()
+    workflow_response.status_code = 200
+    workflow_response.json.return_value = {"name": "CI Workflow"}
+
+    runs_response = MagicMock()
+    runs_response.status_code = 200
+    runs_response.json.return_value = {"workflow_runs": [{"html_url": "https://example.com", "run_number": 1}]}
+
+    def get_side_effect(url):
+        if "workflows/ci.yml" in url and "runs" not in url:
+            return workflow_response
+        elif "runs" in url:
+            return runs_response
+        default = MagicMock()
+        default.status_code = 404
+        return default
+
+    mock_event.get.side_effect = get_side_effect
+
+    with patch("time.sleep"):
+        with patch("actions.dispatch_actions.WORKFLOW_FILES", ["ci.yml"]):
+            trigger_and_get_workflow_info(mock_event, "temp-ci-123-456", temp_branch="temp-ci-123-456")
+
+    # Verify temp branch was deleted
+    mock_event.delete.assert_called_once_with("https://api.github.com/repos/test/repo/git/refs/heads/temp-ci-123-456")
+
+
 def test_update_comment_function():
     """Test updating comment with workflow info."""
     mock_event = MagicMock()
@@ -116,7 +172,7 @@ def test_main_triggers_workflows():
             with patch("actions.dispatch_actions.trigger_and_get_workflow_info") as mock_trigger:
                 with patch("actions.dispatch_actions.update_comment"):
                     # Set return values
-                    mock_get_branch.return_value = "feature-branch"
+                    mock_get_branch.return_value = ("feature-branch", None)
                     mock_trigger.return_value = [{"name": "CI", "file": "ci.yml", "url": "url", "run_number": 1}]
 
                     # Call the function

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/tests/test_github_utils.py

@@ -28,15 +28,14 @@ def test_check_pypi_version():
 
 def test_action_init():
     """Test Action class initialization with default values."""
-    with patch.dict("os.environ", {"GITHUB_TOKEN": "test-token", "GITHUB_EVENT_NAME": "push"}):
-        with patch(
-            "actions.utils.github_utils.Action._load_event_data",
-            return_value={"repository": {"full_name": "test/repo"}},
-        ):
-            action = Action()
-            assert action.token == "test-token"
-            assert action.event_name == "push"
-            assert action.repository == "test/repo"
+    with patch.dict("os.environ", {"GITHUB_TOKEN": "test-token", "GITHUB_EVENT_NAME": "push"}), patch(
+        "actions.utils.github_utils.Action._load_event_data",
+        return_value={"repository": {"full_name": "test/repo"}},
+    ):
+        action = Action()
+        assert action.token == "test-token"
+        assert action.event_name == "push"
+        assert action.repository == "test/repo"
 
 
 def test_action_request_methods():

{ultralytics_actions-0.1.9 → ultralytics_actions-0.2.0}/ultralytics_actions.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-actions
-Version: 0.1.9
+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.1.9/actions/dispatch_actions.py

@@ -1,116 +0,0 @@
-# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
-
-from __future__ import annotations
-
-import time
-from datetime import datetime
-
-from .utils import GITHUB_API_URL, Action
-
-# Configuration
-RUN_CI_KEYWORD = "@ultralytics/run-ci"  # and then to merge "@ultralytics/run-ci-and-merge"
-WORKFLOW_FILES = ["ci.yml", "docker.yml"]
-
-
-def get_pr_branch(event) -> str:
-    """Gets the PR branch name."""
-    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."""
-    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})
-
-    # Wait for workflows to be created
-    time.sleep(10)
-
-    # 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
-
-        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", []):
-                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})
-
-    return results
-
-
-def update_comment(event, comment_body: str, triggered_actions: list[dict], branch: str):
-    """Updates the comment with workflow information."""
-    if not triggered_actions:
-        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"
-    )
-
-    for action in triggered_actions:
-        run_info = f" run {action['run_number']}" if action["run_number"] else ""
-        summary += f"* ✅ [{action['name']}]({action['url']}): `{action['file']}`{run_info}\n"
-
-    new_body = comment_body.replace(RUN_CI_KEYWORD, summary).strip()
-    comment_id = event.event_data["comment"]["id"]
-    event.patch(f"{GITHUB_API_URL}/repos/{event.repository}/issues/comments/{comment_id}", json={"body": new_body})
-
-
-def main(*args, **kwargs):
-    """Handles triggering workflows from PR comments."""
-    event = Action(*args, **kwargs)
-
-    # Only process new comments on PRs
-    if (
-        event.event_name != "issue_comment"
-        or "pull_request" not in event.event_data.get("issue", {})
-        or event.event_data.get("action") != "created"
-    ):
-        return
-
-    # Get comment info
-    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
-    has_keyword = RUN_CI_KEYWORD in comment_body and comment_body.count(RUN_CI_KEYWORD) > comment_body.count(
-        f"`{RUN_CI_KEYWORD}`"
-    )
-    if not has_keyword or not event.is_org_member(username):
-        return
-
-    # Get branch, trigger workflows, and update comment
-    event.toggle_eyes_reaction(True)
-    branch = get_pr_branch(event)
-    print(f"Triggering workflows on branch: {branch}")
-
-    triggered_actions = trigger_and_get_workflow_info(event, branch)
-    update_comment(event, comment_body, triggered_actions, branch)
-    event.toggle_eyes_reaction(False)
-
-
-if __name__ == "__main__":
-    main()