ultralytics-actions 0.0.68__py3-none-any.whl → 0.0.70__py3-none-any.whl

actions/__init__.py

@@ -22,4 +22,4 @@
 #     ├── test_summarize_pr.py
 #     └── ...
 
-__version__ = "0.0.68"
+__version__ = "0.0.70"

actions/dispatch_actions.py

@@ -0,0 +1,127 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+import time
+from datetime import datetime
+from typing import Dict, List
+
+import requests
+
+from .utils import GITHUB_API_URL, Action, remove_html_comments
+
+# Configuration
+RUN_CI_KEYWORD = "@ultralytics/run-ci"  # and then to merge "@ultralytics/run-ci-and-merge"
+WORKFLOW_FILES = ["format.yml", "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:
+        requests.post(
+            f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}/dispatches",
+            json={"ref": branch},
+            headers=event.headers,
+        )
+
+    # Wait for workflows to be created
+    time.sleep(10)
+
+    # Collect information about all workflows
+    for file in WORKFLOW_FILES:
+        # Get workflow name
+        response = requests.get(f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}", headers=event.headers)
+        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 = requests.get(
+            f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}/runs?branch={branch}&event=workflow_dispatch&per_page=1",
+            headers=event.headers,
+        )
+
+        if runs_response.status_code == 200:
+            runs = runs_response.json().get("workflow_runs", [])
+            if 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) -> bool:
+    """Updates the comment with workflow information."""
+    if not triggered_actions:
+        return False
+
+    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 `@ultralytics/dispatch-actions 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"]
+
+    response = requests.patch(
+        f"{GITHUB_API_URL}/repos/{event.repository}/issues/comments/{comment_id}",
+        json={"body": new_body},
+        headers=event.headers,
+    )
+
+    return response.status_code == 200
+
+
+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 = remove_html_comments(event.event_data["comment"].get("body", ""))
+    username = event.event_data["comment"]["user"]["login"]
+
+    # Check for trigger keyword and permissions
+    if RUN_CI_KEYWORD not in comment_body or not event.is_org_member(username):
+        return
+
+    # Get branch, trigger workflows, and update comment
+    branch = get_pr_branch(event)
+    print(f"Triggering workflows on branch: {branch}")
+
+    triggered_actions = trigger_and_get_workflow_info(event, branch)
+    success = update_comment(event, comment_body, triggered_actions, branch)
+
+    print(f"Comment update {'succeeded' if success else 'failed'}.")
+
+
+if __name__ == "__main__":
+    main()

actions/utils/common_utils.py

@@ -36,6 +36,15 @@ BAD_HTTP_CODES = frozenset(
         525,  # Cloudfare handshake error
     }
 )
+
+URL_ERROR_LIST = {  # automatically reject these URLs (important: replace spaces with '%20')
+    "https://blog.research.google/search/label/Spam%20and%20Abuse",
+    "https://blog.research.google/search/label/Adversarial%20Attacks",
+    "https://www.microsoft.com/en-us/security/business/ai-machine-learning-security",
+    "https://about.netflix.com/en/news/netflix-recommendations-beyond-the-5-stars-part-1",
+    "https://about.netflix.com/en/news/netflix-research-recommendations",
+}
+
 URL_IGNORE_LIST = {  # use a set (not frozenset) to update with possible private GitHub repos
     "localhost",
     "127.0.0",
@@ -51,10 +60,11 @@ URL_IGNORE_LIST = {  # use a set (not frozenset) to update with possible private
     "mailto:",
     "linkedin.com",
     "twitter.com",
-    "x.com",
+    "https://x.com",  # do not use just 'x' as this will catch other domains like netflix.com
     "storage.googleapis.com",  # private GCS buckets
     "{",  # possible Python fstring
     "(",  # breaks pattern matches
+    ")",
     "api.",  # ignore api endpoints
 }
 REDIRECT_START_IGNORE_LIST = frozenset(
@@ -69,9 +79,14 @@ REDIRECT_START_IGNORE_LIST = frozenset(
         "ultralytics.com/actions",
         "ultralytics.com/bilibili",
         "ultralytics.com/images",
+        "ultralytics.com/license",
+        "ultralytics.com/assets",
         "app.gong.io/call?",
         "docs.openvino.ai",
+        ".git",
+        "/raw/",  # GitHub images
     }
+    | URL_IGNORE_LIST
 )
 REDIRECT_END_IGNORE_LIST = frozenset(
     {
@@ -161,7 +176,7 @@ def is_url(url, session=None, check=True, max_attempts=3, timeout=3, return_url=
         # Check structure
         result = parse.urlparse(url)
         partition = result.netloc.partition(".")  # i.e. netloc = "github.com" -> ("github", ".", "com")
-        if not result.scheme or not partition[0] or not partition[2]:
+        if not result.scheme or not partition[0] or not partition[2] or (url in URL_ERROR_LIST):
             return (False, url) if return_url else False
 
         if check:

actions/utils/github_utils.py

@@ -50,6 +50,13 @@ class Action:
             print(f"Error parsing authenticated user response: {e}")
             return None
 
+    def is_org_member(self, username: str) -> bool:
+        """Checks if a user is a member of the organization using the GitHub API."""
+        org_name = self.repository.split("/")[0]
+        url = f"{GITHUB_API_URL}/orgs/{org_name}/members/{username}"
+        r = requests.get(url, headers=self.headers)
+        return r.status_code == 204  # 204 means the user is a member
+
     def get_pr_diff(self) -> str:
         """Retrieves the diff content for a specified pull request."""
         url = f"{GITHUB_API_URL}/repos/{self.repository}/pulls/{self.pr.get('number')}"

actions/utils/openai_utils.py

@@ -9,7 +9,7 @@ import requests
 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-4o-2024-11-20")
+OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-4.1-2025-04-14")
 SYSTEM_PROMPT_ADDITION = """
 Guidance:
   - Ultralytics Branding: Use YOLO11, YOLO12, etc., not YOLOv11, YOLOv12 (only older versions like YOLOv10 have a v). Always capitalize "HUB" in "Ultralytics HUB"; use "Ultralytics HUB", not "The Ultralytics HUB". 

{ultralytics_actions-0.0.68.dist-info → ultralytics_actions-0.0.70.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-actions
-Version: 0.0.68
+Version: 0.0.70
 Summary: Ultralytics Actions for GitHub automation and PR management.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>
@@ -59,8 +59,8 @@ Ultralytics Actions automatically applies formats, updates, and enhancements usi
 - **Swift Code:** Formatted with [`swift-format`](https://github.com/swiftlang/swift-format) to maintain a uniform coding style across Swift projects. _(Note: Requires the `macos-latest` runner.)_
 - **Spell Check:** Common misspellings are caught using [codespell](https://github.com/codespell-project/codespell).
 - **Broken Links Check:** Broken links in documentation and Markdown files are identified using [Lychee](https://github.com/lycheeverse/lychee).
-- **PR Summary:** Concise Pull Request summaries are generated using [OpenAI](https://openai.com/) GPT-4o, improving clarity and review efficiency.
-- **Auto-labeling:** Relevant labels are applied to issues and pull requests via [OpenAI](https://openai.com/) GPT-4o for intelligent categorization.
+- **PR Summary:** Concise Pull Request summaries are generated using [OpenAI](https://openai.com/) GPT-4.1, improving clarity and review efficiency.
+- **Auto-labeling:** Relevant labels are applied to issues and pull requests via [OpenAI](https://openai.com/) GPT-4.1 for intelligent categorization.
 
 ## 🛠️ How It Works
 
@@ -69,9 +69,9 @@ Ultralytics Actions triggers on various GitHub events to streamline workflows:
 - **Push Events:** Automatically formats code when changes are pushed to the `main` branch.
 - **Pull Requests:**
   - Ensures contributions meet formatting standards before merging.
-  - Generates a concise summary of changes using GPT-4o.
-  - Applies relevant labels using GPT-4o for intelligent categorization.
-- **Issues:** Automatically applies relevant labels using GPT-4o when new issues are created.
+  - Generates a concise summary of changes using GPT-4.1.
+  - Applies relevant labels using GPT-4.1 for intelligent categorization.
+- **Issues:** Automatically applies relevant labels using GPT-4.1 when new issues are created.
 
 These automated actions help maintain high code quality, improve documentation clarity, and streamline the review process by providing consistent formatting, informative summaries, and appropriate categorization.
 
@@ -101,13 +101,13 @@ To integrate this action into your Ultralytics repository:
             uses: ultralytics/actions@main
             with:
               token: ${{ secrets.GITHUB_TOKEN }} # Automatically generated, do not modify
-              labels: true # Autolabel issues and PRs using GPT-4o (requires 'openai_api_key')
+              labels: true # Autolabel issues and PRs using GPT-4.1 (requires 'openai_api_key')
               python: true # Format Python code and docstrings with Ruff and docformatter
               prettier: true # Format YAML, JSON, Markdown, and CSS with Prettier
               swift: false # Format Swift code with swift-format (requires 'runs-on: macos-latest')
               spelling: true # Check spelling with codespell
               links: true # Check for broken links with Lychee
-              summary: true # Generate PR summary with GPT-4o (requires 'openai_api_key')
+              summary: true # Generate PR summary with GPT-4.1 (requires 'openai_api_key')
               openai_api_key: ${{ secrets.OPENAI_API_KEY }} # Add your OpenAI API key as a repository secret
     ```
 

ultralytics_actions-0.0.70.dist-info/RECORD

@@ -0,0 +1,16 @@
+actions/__init__.py,sha256=jkNfCMkrM58RMv6ijJXC4tASCnP7q1VvKsh1yKtLQn4,742
+actions/dispatch_actions.py,sha256=plfDaFdfBPldqI1uUQPfgSeIDaRTKbkzkXnsYbGkmwI,4415
+actions/first_interaction.py,sha256=1_WvQHCi5RWaSfyi49ClF2Zk_3CKGjFnZqz6FlxPRAc,17868
+actions/summarize_pr.py,sha256=BKttOq-MGaanVaChLU5B1ewKUA8K6S05Cy3FQtyRmxU,11681
+actions/summarize_release.py,sha256=tov6qsYGC68lfobvkwVyoWZBGtJ598G0m097n4Ydzvo,8472
+actions/update_markdown_code_blocks.py,sha256=9PL7YIQfApRNAa0que2hYHv7umGZTZoHlblesB0xFj4,8587
+actions/utils/__init__.py,sha256=ZE0RmC9qOCt9TUhvORd6uVhbxOKVFWJDobR454v55_M,682
+actions/utils/common_utils.py,sha256=BCqB2noJRXxXLAH8rmZ7c4ss1WjFXDC9OPPPFX5OuRU,11758
+actions/utils/github_utils.py,sha256=EZIATz5OSrNgdBrLFxL-9KRxn0i_Ph5-ZXHhhcIrtPE,7495
+actions/utils/openai_utils.py,sha256=txbsEPQnIOieejatBuE6Yk7xR1fQ0erWOEs6cYgUQX4,2943
+ultralytics_actions-0.0.70.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+ultralytics_actions-0.0.70.dist-info/METADATA,sha256=c9aZUeDQntfNp4gptnUlEVAxDVoEieK1h2qcGQ88jDs,10930
+ultralytics_actions-0.0.70.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+ultralytics_actions-0.0.70.dist-info/entry_points.txt,sha256=GowvOFplj0C7JmsjbKcbpgLpdf2r921pcaOQkAHWZRA,378
+ultralytics_actions-0.0.70.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
+ultralytics_actions-0.0.70.dist-info/RECORD,,

{ultralytics_actions-0.0.68.dist-info → ultralytics_actions-0.0.70.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

ultralytics_actions-0.0.68.dist-info/RECORD

@@ -1,15 +0,0 @@
-actions/__init__.py,sha256=JRXgdKfSoTN4UJEHO2P9prHZYIHPLBQo9hbiRD1WC20,742
-actions/first_interaction.py,sha256=1_WvQHCi5RWaSfyi49ClF2Zk_3CKGjFnZqz6FlxPRAc,17868
-actions/summarize_pr.py,sha256=BKttOq-MGaanVaChLU5B1ewKUA8K6S05Cy3FQtyRmxU,11681
-actions/summarize_release.py,sha256=tov6qsYGC68lfobvkwVyoWZBGtJ598G0m097n4Ydzvo,8472
-actions/update_markdown_code_blocks.py,sha256=9PL7YIQfApRNAa0que2hYHv7umGZTZoHlblesB0xFj4,8587
-actions/utils/__init__.py,sha256=ZE0RmC9qOCt9TUhvORd6uVhbxOKVFWJDobR454v55_M,682
-actions/utils/common_utils.py,sha256=YRdEz8qluwzCZfWgqXNmyhKqNhdxNMpoHhGaHUD4AaM,11013
-actions/utils/github_utils.py,sha256=-F--JgxtXE0fSPMFEzakz7iZilp-vonzLiyXfg0b17Y,7117
-actions/utils/openai_utils.py,sha256=qQbmrJpOUANxSMf7inDSgPIwgf0JHD1fWZuab-y2W6g,2942
-ultralytics_actions-0.0.68.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-ultralytics_actions-0.0.68.dist-info/METADATA,sha256=CpcyV3LeuvtCpvOAzycM_EwA3Gsnq190EPQQuTZ3vUw,10923
-ultralytics_actions-0.0.68.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-ultralytics_actions-0.0.68.dist-info/entry_points.txt,sha256=GowvOFplj0C7JmsjbKcbpgLpdf2r921pcaOQkAHWZRA,378
-ultralytics_actions-0.0.68.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
-ultralytics_actions-0.0.68.dist-info/RECORD,,