ultralytics-actions 0.0.54__py3-none-any.whl → 0.0.56__py3-none-any.whl

actions/__init__.py

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

actions/summarize_pr.py

@@ -29,7 +29,7 @@ def generate_merge_message(pr_summary=None, pr_credit=None, pr_url=None):
                 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."
+                f"figure, and connect it to the PR's impact. Keep it concise yet meaningful, ensuring contributors feel valued."
             ),
         },
     ]
@@ -44,8 +44,13 @@ def post_merge_message(pr_number, pr_url, repository, summary, pr_credit, header
     return response.status_code == 201
 
 
-def generate_issue_comment(pr_url, pr_summary, pr_credit):
-    """Generates a personalized issue comment using based on the PR context."""
+def generate_issue_comment(pr_url, pr_summary, pr_credit, pr_title=""):
+    """Generates personalized issue comment based on PR context."""
+    # Extract repo info from PR URL (format: api.github.com/repos/owner/repo/pulls/number)
+    repo_parts = pr_url.split("/repos/")[1].split("/pulls/")[0] if "/repos/" in pr_url else ""
+    owner_repo = repo_parts.split("/")
+    repo_name = owner_repo[-1] if len(owner_repo) > 1 else "package"
+
     messages = [
         {
             "role": "system",
@@ -54,13 +59,15 @@ def generate_issue_comment(pr_url, pr_summary, pr_credit):
         {
             "role": "user",
             "content": f"Write a GitHub issue comment announcing a potential fix for this issue is now merged in linked PR {pr_url} by {pr_credit}\n\n"
+            f"PR Title: {pr_title}\n\n"
             f"Context from PR:\n{pr_summary}\n\n"
             f"Include:\n"
             f"1. An explanation of key changes from the PR that may resolve this issue\n"
             f"2. Credit to the PR author and contributors\n"
             f"3. Options for testing if PR changes have resolved this issue:\n"
-            f"   - pip install git+https://github.com/ultralytics/ultralytics.git@main # test latest changes\n"
-            f"   - or await next official PyPI release\n"
+            f"   - If the PR mentions a specific version number (like v8.0.0 or 3.1.0), include: pip install -U {repo_name}>=VERSION\n"
+            f"   - Also suggest: pip install git+https://github.com/{repo_parts}.git@main\n"
+            f"   - If appropriate, mention they can also wait for the next official PyPI release\n"
             f"4. Request feedback on whether the PR changes resolve the issue\n"
             f"5. Thank 🙏 for reporting the issue and welcome any further feedback if the issue persists\n\n",
         },
@@ -120,13 +127,14 @@ def update_pr_description(pr_url, new_summary, headers, max_retries=2):
 
 
 def label_fixed_issues(repository, pr_number, pr_summary, headers, action):
-    """Labels issues closed by PR when merged, notifies users, returns PR contributors."""
+    """Labels issues closed by PR when merged, notifies users, and returns PR contributors."""
     query = """
 query($owner: String!, $repo: String!, $pr_number: Int!) {
     repository(owner: $owner, name: $repo) {
         pullRequest(number: $pr_number) {
             closingIssuesReferences(first: 50) { nodes { number } }
             url
+            title
             body
             author { login, __typename }
             reviews(first: 50) { nodes { author { login, __typename } } }
@@ -143,13 +151,14 @@ query($owner: String!, $repo: String!, $pr_number: Int!) {
 
     if response.status_code != 200:
         print(f"Failed to fetch linked issues. Status code: {response.status_code}")
-        return [], None
+        return None
 
     try:
         data = response.json()["data"]["repository"]["pullRequest"]
         comments = data["reviews"]["nodes"] + data["comments"]["nodes"]
         token_username = action.get_username()  # get GITHUB_TOKEN username
         author = data["author"]["login"] if data["author"]["__typename"] != "Bot" else None
+        pr_title = data.get("title", "")
 
         # Get unique contributors from reviews and comments
         contributors = {x["author"]["login"] for x in comments if x["author"]["__typename"] != "Bot"}
@@ -173,7 +182,9 @@ query($owner: String!, $repo: String!, $pr_number: Int!) {
             pr_credit += (" with contributions from " if pr_credit else "") + ", ".join(f"@{c}" for c in contributors)
 
         # Generate personalized comment
-        comment = generate_issue_comment(pr_url=data["url"], pr_summary=pr_summary, pr_credit=pr_credit)
+        comment = generate_issue_comment(
+            pr_url=data["url"], pr_summary=pr_summary, pr_credit=pr_credit, pr_title=pr_title
+        )
 
         # Update linked issues
         for issue in data["closingIssuesReferences"]["nodes"]:
@@ -197,7 +208,7 @@ query($owner: String!, $repo: String!, $pr_number: Int!) {
         return pr_credit
     except KeyError as e:
         print(f"Error parsing GraphQL response: {e}")
-        return [], None
+        return None
 
 
 def remove_todos_on_merge(pr_number, repository, headers):

actions/update_markdown_code_blocks.py

@@ -158,7 +158,7 @@ def main(root_dir=Path.cwd(), verbose=False):
     for markdown_file in markdown_files:
         if verbose:
             print(f"Processing {markdown_file}")
-        markdown_content, temp_files = process_markdown_file(markdown_file, temp_dir)
+        markdown_content, temp_files = process_markdown_file(markdown_file, temp_dir, verbose)
         if markdown_content and temp_files:
             all_temp_files.append((markdown_file, markdown_content, temp_files))
 

actions/utils/github_utils.py

@@ -32,7 +32,7 @@ class Action:
 
     @staticmethod
     def _load_event_data(event_path: str) -> dict:
-        """Loads GitHub event data from path if it exists."""
+        """Load GitHub event data from path if it exists."""
         if event_path and Path(event_path).exists():
             return json.loads(Path(event_path).read_text())
         return {}
@@ -79,7 +79,7 @@ class Action:
         return result
 
     def print_info(self):
-        """Print GitHub Actions information."""
+        """Print GitHub Actions information including event details and repository information."""
         info = {
             "github.event_name": self.event_name,
             "github.event.action": self.event_data.get("action"),
@@ -110,7 +110,7 @@ class Action:
 
 
 def ultralytics_actions_info():
-    """Returns GitHub Actions environment information and configuration details for Ultralytics workflows."""
+    """Return GitHub Actions environment information and configuration details for Ultralytics workflows."""
     Action().print_info()
 
 

actions/utils/openai_utils.py

@@ -14,11 +14,19 @@ 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". 
   - Avoid Equations: Do not include equations or mathematical notations.
-  - Markdown: Always respond in Markdown.
   - Tone: Adopt a professional, friendly, and concise tone.
 """
 
 
+def remove_outer_codeblocks(string):
+    """Removes outer code block markers and language identifiers from a string while preserving inner content."""
+    string = string.strip()
+    if string.startswith("```") and string.endswith("```"):
+        # Get everything after first ``` and newline, up to the last ```
+        string = string[string.find("\n") + 1 : string.rfind("```")].strip()
+    return string
+
+
 def get_completion(
     messages: List[Dict[str, str]],
     check_links: bool = True,
@@ -45,6 +53,7 @@ def get_completion(
         r = requests.post(url, headers=headers, json=data)
         r.raise_for_status()
         content = r.json()["choices"][0]["message"]["content"].strip()
+        content = remove_outer_codeblocks(content)
         for x in remove:
             content = content.replace(x, "")
         if not check_links or check_links_in_string(content):  # if no checks or checks are passing return response

{ultralytics_actions-0.0.54.dist-info → ultralytics_actions-0.0.56.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ultralytics-actions
-Version: 0.0.54
+Version: 0.0.56
 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.56.dist-info/RECORD

@@ -0,0 +1,15 @@
+actions/__init__.py,sha256=OsA1f_zloRyT6FbnRCGPxd76Cu-N-Vf5WReu9UDlbqY,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=tUChNBIZN-B_unGMG9yQk-dohi7bit02Yl3xc4UtycQ,6610
+actions/utils/__init__.py,sha256=WStdEAYROVnF0nubEOmrFLrejkRiMXIefA5O1ckfcFs,476
+actions/utils/common_utils.py,sha256=PZkK9Wc3od34J9VSw4ICWHhQQC033o3DCrCy0VIyZE4,6024
+actions/utils/github_utils.py,sha256=-F--JgxtXE0fSPMFEzakz7iZilp-vonzLiyXfg0b17Y,7117
+actions/utils/openai_utils.py,sha256=qQbmrJpOUANxSMf7inDSgPIwgf0JHD1fWZuab-y2W6g,2942
+ultralytics_actions-0.0.56.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+ultralytics_actions-0.0.56.dist-info/METADATA,sha256=lvqelxqPqg0UiVc-ocGZ6Dy884vQtnjxwlqHqW34AcQ,10561
+ultralytics_actions-0.0.56.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+ultralytics_actions-0.0.56.dist-info/entry_points.txt,sha256=GowvOFplj0C7JmsjbKcbpgLpdf2r921pcaOQkAHWZRA,378
+ultralytics_actions-0.0.56.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
+ultralytics_actions-0.0.56.dist-info/RECORD,,

ultralytics_actions-0.0.54.dist-info/RECORD

@@ -1,15 +0,0 @@
-actions/__init__.py,sha256=jdvLrgMZ2Flg6xeVVfi6ayAk9bKCZWgtopdmkyIv8CI,742
-actions/first_interaction.py,sha256=1_WvQHCi5RWaSfyi49ClF2Zk_3CKGjFnZqz6FlxPRAc,17868
-actions/summarize_pr.py,sha256=WHte6PNJbk_-T8fRDJERs6x3KL4Ah54oH0OH-tsqMUA,11092
-actions/summarize_release.py,sha256=tov6qsYGC68lfobvkwVyoWZBGtJ598G0m097n4Ydzvo,8472
-actions/update_markdown_code_blocks.py,sha256=ip-KRIfnTMDrs9IOqdSY7YXR4abNLsylGnXtrvoXjwY,6601
-actions/utils/__init__.py,sha256=WStdEAYROVnF0nubEOmrFLrejkRiMXIefA5O1ckfcFs,476
-actions/utils/common_utils.py,sha256=PZkK9Wc3od34J9VSw4ICWHhQQC033o3DCrCy0VIyZE4,6024
-actions/utils/github_utils.py,sha256=0h0Hz2tgUta61Ymn9YggRXBZ7aZdF5krKnX7Tj9jqRU,7068
-actions/utils/openai_utils.py,sha256=EQAs1cd-YXXF8xHKLrY3Tsl_AVzgdlXcYOZMYh88_gM,2524
-ultralytics_actions-0.0.54.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-ultralytics_actions-0.0.54.dist-info/METADATA,sha256=4RFFr5-vHZihzjgS_ZklhfTvhO9-TXjmbIDjMxyeYWY,10561
-ultralytics_actions-0.0.54.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-ultralytics_actions-0.0.54.dist-info/entry_points.txt,sha256=GowvOFplj0C7JmsjbKcbpgLpdf2r921pcaOQkAHWZRA,378
-ultralytics_actions-0.0.54.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
-ultralytics_actions-0.0.54.dist-info/RECORD,,