ultralytics-actions 0.0.48__py3-none-any.whl → 0.0.50__py3-none-any.whl

actions/__init__.py

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

actions/summarize_pr.py

@@ -16,29 +16,29 @@ SUMMARY_START = (
 )
 
 
-def generate_merge_message(pr_summary=None, pr_credit=None):
-    """Generates a thank-you message for merged PR contributors."""
+def generate_merge_message(pr_summary=None, pr_credit=None, pr_url=None):
+    """Generates a motivating thank-you message for merged PR contributors."""
     messages = [
         {
             "role": "system",
-            "content": "You are an Ultralytics AI assistant. Generate meaningful, inspiring messages to GitHub users.",
+            "content": "You are an Ultralytics AI assistant. Generate inspiring, appreciative messages for GitHub contributors.",
         },
         {
             "role": "user",
-            "content": f"Write a friendly thank you for a merged GitHub PR by {pr_credit}. "
-            f"Context from PR:\n{pr_summary}\n\n"
-            f"Start with the exciting message that this PR is now merged, and weave in an inspiring but obscure quote "
-            f"from a historical figure in science, art, stoicism and philosophy. "
-            f"Keep the message concise yet relevant to the specific contributions in this PR. "
-            f"We want the contributors to feel their effort is appreciated and will make a difference in the world.",
+            "content": (
+                f"Write a warm thank-you comment for the merged PR {pr_url} by {pr_credit}. "
+                f"Context:\n{pr_summary}\n\n"
+                f"Start with an enthusiastic note about the merge, incorporate a relevant inspirational quote from a historical "
+                f"figure, and connect it to the PR’s impact. Keep it concise yet meaningful, ensuring contributors feel valued."
+            ),
         },
     ]
     return get_completion(messages)
 
 
-def post_merge_message(pr_number, repository, summary, pr_credit, headers):
+def post_merge_message(pr_number, pr_url, repository, summary, pr_credit, headers):
     """Posts thank you message on PR after merge."""
-    message = generate_merge_message(summary, pr_credit)
+    message = generate_merge_message(summary, pr_credit, pr_url)
     comment_url = f"{GITHUB_API_URL}/repos/{repository}/issues/{pr_number}/comments"
     response = requests.post(comment_url, json={"body": message}, headers=headers)
     return response.status_code == 201
@@ -94,9 +94,8 @@ def generate_pr_summary(repository, diff_text):
     return SUMMARY_START + reply
 
 
-def update_pr_description(repository, pr_number, new_summary, headers, max_retries=2):
+def update_pr_description(pr_url, new_summary, headers, max_retries=2):
     """Updates PR description with new summary, retrying if description is None."""
-    pr_url = f"{GITHUB_API_URL}/repos/{repository}/pulls/{pr_number}"
     description = ""
     for i in range(max_retries + 1):
         description = requests.get(pr_url, headers=headers).json().get("body") or ""
@@ -210,9 +209,10 @@ def remove_todos_on_merge(pr_number, repository, headers):
 def main(*args, **kwargs):
     """Summarize a pull request and update its description with a summary."""
     action = Action(*args, **kwargs)
-    pr_number = action.pr["number"]
     headers = action.headers
     repository = action.repository
+    pr_number = action.pr["number"]
+    pr_url = f"{GITHUB_API_URL}/repos/{repository}/pulls/{pr_number}"
 
     print(f"Retrieving diff for PR {pr_number}")
     diff = action.get_pr_diff()
@@ -223,7 +223,7 @@ def main(*args, **kwargs):
 
     # Update PR description
     print("Updating PR description...")
-    status_code = update_pr_description(repository, pr_number, summary, headers)
+    status_code = update_pr_description(pr_url, summary, headers)
     if status_code == 200:
         print("PR description updated successfully.")
     else:
@@ -237,7 +237,7 @@ def main(*args, **kwargs):
         remove_todos_on_merge(pr_number, repository, headers)
         if pr_credit:
             print("Posting PR author thank you message...")
-            post_merge_message(pr_number, repository, summary, pr_credit, headers)
+            post_merge_message(pr_number, pr_url, repository, summary, pr_credit, headers)
 
 
 if __name__ == "__main__":

actions/utils/common_utils.py

@@ -34,27 +34,24 @@ BAD_HTTP_CODES = frozenset(
         504,  # Gateway Timeout - upstream server didn't respond in time
     }
 )
-URL_IGNORE_LIST = frozenset(
-    {
-        "localhost",
-        "127.0.0",
-        ":5000",
-        ":3000",
-        ":8000",
-        ":8080",
-        ":6006",
-        "MODEL_ID",
-        "API_KEY",
-        "url",
-        "example",
-        "mailto:",
-        "github.com",  # ignore GitHub links that may be private repos
-        "linkedin.com",
-        "twitter.com",
-        "x.com",
-        "storage.googleapis.com",  # private GCS buckets
-    }
-)
+URL_IGNORE_LIST = {  # use a set (not frozenset) to update with possible private GitHub repos
+    "localhost",
+    "127.0.0",
+    ":5000",
+    ":3000",
+    ":8000",
+    ":8080",
+    ":6006",
+    "MODEL_ID",
+    "API_KEY",
+    "url",
+    "example",
+    "mailto:",
+    "linkedin.com",
+    "twitter.com",
+    "x.com",
+    "storage.googleapis.com",  # private GCS buckets
+}
 URL_PATTERN = re.compile(
     r"\[([^]]+)]\(([^)]+)\)"  # Matches Markdown links [text](url)
     r"|"
@@ -81,7 +78,7 @@ def clean_url(url):
 
 
 def is_url(url, session=None, check=True, max_attempts=3, timeout=2):
-    """Check if string is URL and optionally verify it exists."""
+    """Check if string is URL and optionally verify it exists, with fallback for GitHub repos."""
     try:
         # Check allow list
         if any(x in url for x in URL_IGNORE_LIST):
@@ -105,6 +102,16 @@ def is_url(url, session=None, check=True, max_attempts=3, timeout=2):
                     for method in (requester.head, requester.get):
                         if method(url, stream=method == requester.get, **kwargs).status_code not in BAD_HTTP_CODES:
                             return True
+
+                        # If GitHub and check fails (repo might be private), add the base GitHub URL to ignore list
+                        if result.hostname == "github.com":
+                            parts = result.path.strip("/").split("/")
+                            if len(parts) >= 2:
+                                base_url = f"https://github.com/{parts[0]}/{parts[1]}"  # https://github.com/org/repo
+                                if requester.head(base_url, **kwargs).status_code == 404:
+                                    URL_IGNORE_LIST.add(base_url)
+                                    return True
+
                     return False
                 except Exception:
                     if attempt == max_attempts - 1:  # last attempt

{ultralytics_actions-0.0.48.dist-info → ultralytics_actions-0.0.50.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ultralytics-actions
-Version: 0.0.48
+Version: 0.0.50
 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.48.dist-info → ultralytics_actions-0.0.50.dist-info}/RECORD

@@ -1,15 +1,15 @@
-actions/__init__.py,sha256=OqYKEDJfchmeCRCnZmLhs3knsOhRPO9LKEQzuaWOgq0,742
+actions/__init__.py,sha256=4U8iPnscnkKIwXcYrHT-0T4HHmy37vzw_sbFoMwGSyw,742
 actions/first_interaction.py,sha256=Oosw3GlXObF4RxYJG8haIjKXSV4MAx_uPXfMdRKY5qA,17851
-actions/summarize_pr.py,sha256=Pfcm6ArfZfcMme48FcCGNmMwrFFlsG0czifhXHDKAlY,11166
+actions/summarize_pr.py,sha256=gLtKlE_iKzP92vBfA-KA9bXUwdrKIe79pH2f4Oyq_gg,11075
 actions/summarize_release.py,sha256=2D1IIeS4xrQNEJER3HItm5u9XvTL8hwGX_jWD2S8q1Y,8455
 actions/update_markdown_code_blocks.py,sha256=DN6rrDw2VHXUTetrrg1SHlYDco9iaGOfQBKFORqbCBI,6362
 actions/utils/__init__.py,sha256=WStdEAYROVnF0nubEOmrFLrejkRiMXIefA5O1ckfcFs,476
-actions/utils/common_utils.py,sha256=o2dMSS7TxpwJLU58ybcZFgtaV-PM5lzQRP0B9cW-bBc,5420
+actions/utils/common_utils.py,sha256=LJ79U5sQuTnVyhc22Bryb5YUBAWy13rGoHdfbaBQl3s,5988
 actions/utils/github_utils.py,sha256=0h0Hz2tgUta61Ymn9YggRXBZ7aZdF5krKnX7Tj9jqRU,7068
 actions/utils/openai_utils.py,sha256=U7DjxTdFGdhmWSE4_KIg1yPQdtrfG4GkbNZCgMw4_1Q,1822
-ultralytics_actions-0.0.48.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-ultralytics_actions-0.0.48.dist-info/METADATA,sha256=bQqAndEfjcyjmYcRB73ybeISLirURmF2ZLxRK_uYv5s,10561
-ultralytics_actions-0.0.48.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ultralytics_actions-0.0.48.dist-info/entry_points.txt,sha256=GowvOFplj0C7JmsjbKcbpgLpdf2r921pcaOQkAHWZRA,378
-ultralytics_actions-0.0.48.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
-ultralytics_actions-0.0.48.dist-info/RECORD,,
+ultralytics_actions-0.0.50.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+ultralytics_actions-0.0.50.dist-info/METADATA,sha256=qPiz8EXLadPuzCesY3Sojb5Hs-D-B2j5znK16qmYR9I,10561
+ultralytics_actions-0.0.50.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ultralytics_actions-0.0.50.dist-info/entry_points.txt,sha256=GowvOFplj0C7JmsjbKcbpgLpdf2r921pcaOQkAHWZRA,378
+ultralytics_actions-0.0.50.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
+ultralytics_actions-0.0.50.dist-info/RECORD,,