ultralytics-actions 0.0.10__tar.gz → 0.0.15__tar.gz

{ultralytics_actions-0.0.10/ultralytics_actions.egg-info → ultralytics_actions-0.0.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics-actions
-Version: 0.0.10
+Version: 0.0.15
 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.0.10 → ultralytics_actions-0.0.15}/actions/__init__.py

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

{ultralytics_actions-0.0.10 → ultralytics_actions-0.0.15}/actions/summarize_pr.py

@@ -1,5 +1,7 @@
 # Ultralytics Actions 🚀, AGPL-3.0 license https://ultralytics.com/license
 
+import time
+
 import requests
 
 from .utils import (
@@ -39,24 +41,30 @@ def generate_pr_summary(repo_name, diff_text):
     ]
     reply = get_completion(messages)
     if len(diff_text) > limit:
-        return SUMMARY_START + "**WARNING ⚠️** this PR is very large, summary may not cover all changes.\n\n" + reply
-    else:
-        return SUMMARY_START + reply
+        reply = "**WARNING ⚠️** this PR is very large, summary may not cover all changes.\n\n" + reply
+    return reply
 
 
-def update_pr_description(repo_name, pr_number, new_summary):
-    """Updates the PR description with a new summary, replacing existing summary if present."""
-    # Fetch the current PR description
+def update_pr_description(repo_name, pr_number, new_summary, max_retries=2):
+    """Updates PR description with new summary, retrying if description is None."""
     pr_url = f"{GITHUB_API_URL}/repos/{repo_name}/pulls/{pr_number}"
-    pr_response = requests.get(pr_url, headers=GITHUB_HEADERS)
-    pr_data = pr_response.json()
-    current_description = pr_data.get("body") or ""  # warning, can be None
+    description = ""
+    for i in range(max_retries + 1):
+        description = requests.get(pr_url, headers=GITHUB_HEADERS).json().get("body") or ""
+        if description:
+            break
+        if i < max_retries:
+            print("No current PR description found, retrying...")
+            time.sleep(1)
 
     # Check if existing summary is present and update accordingly
-    if SUMMARY_START in current_description:
-        updated_description = current_description.split(SUMMARY_START)[0] + new_summary
+    start = "## 🛠️ PR Summary"
+    if start in description:
+        print("Existing PR Summary found, replacing.")
+        updated_description = description.split(start)[0] + new_summary
     else:
-        updated_description = current_description + "\n\n" + new_summary
+        print("PR Summary not found, appending.")
+        updated_description = description + "\n\n" + new_summary
 
     # Update the PR description
     update_response = requests.patch(pr_url, json={"body": updated_description}, headers=GITHUB_HEADERS)

{ultralytics_actions-0.0.10 → ultralytics_actions-0.0.15}/actions/utils/common_utils.py

@@ -36,6 +36,7 @@ def is_url(url, check=True, max_attempts=3, timeout=2):
         "mailto:",
         "github.com",  # ignore GitHub links that may be private repos
         "kaggle.com",  # blocks automated header requests
+        "reddit.com",  # blocks automated header requests
     )
     try:
         # Check allow list

{ultralytics_actions-0.0.10 → ultralytics_actions-0.0.15}/actions/utils/github_utils.py

@@ -5,6 +5,8 @@ from pathlib import Path
 
 import requests
 
+from actions import __version__
+
 GITHUB_TOKEN = os.getenv("GITHUB_TOKEN")
 GITHUB_REPOSITORY = os.getenv("GITHUB_REPOSITORY")
 GITHUB_EVENT_NAME = os.getenv("GITHUB_EVENT_NAME")
@@ -121,7 +123,8 @@ def ultralytics_actions_info():
 
     # Print information
     max_key_length = max(len(key) for key in info.keys())
-    print("Ultralytics Actions Information " + "-" * 40)  # header (72 long)
+    header = f"Ultralytics Actions {__version__} Information " + "-" * 40
+    print(header)
     for key, value in info.items():
         print(f"{key:<{max_key_length + 5}}{value}")
-    print("-" * 72)  # footer
+    print("-" * len(header))  # footer

{ultralytics_actions-0.0.10 → ultralytics_actions-0.0.15/ultralytics_actions.egg-info}/PKG-INFO

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