ultralytics-actions 0.0.6__tar.gz → 0.0.8__tar.gz

{ultralytics_actions-0.0.6/ultralytics_actions.egg-info → ultralytics_actions-0.0.8}/PKG-INFO

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

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

{ultralytics_actions-0.0.6 → ultralytics_actions-0.0.8}/actions/utils/__init__.py

@@ -14,6 +14,7 @@ from .github_utils import (
     get_github_data,
     get_pr_diff,
     graphql_request,
+    ultralytics_actions_info,
 )
 from .openai_utils import OPENAI_API_KEY, OPENAI_MODEL, get_completion
 
@@ -34,4 +35,5 @@ __all__ = (
     "OPENAI_MODEL",
     "get_completion",
     "check_pypi_version",
+    "ultralytics_actions_info",
 )

ultralytics_actions-0.0.8/actions/utils/common_utils.py

@@ -0,0 +1,92 @@
+# Ultralytics Actions 🚀, AGPL-3.0 license https://ultralytics.com/license
+
+import re
+import socket
+import time
+import urllib
+from concurrent.futures import ThreadPoolExecutor
+
+
+def remove_html_comments(body: str) -> str:
+    """Removes HTML comments from a string using regex pattern matching."""
+    return re.sub(r"<!--.*?-->", "", body, flags=re.DOTALL).strip()
+
+
+def clean_url(url):
+    """Remove extra characters from URL strings."""
+    for _ in range(3):
+        url = str(url).strip('"').strip("'").rstrip(".,:;!?`\\").replace(".git@main", "").replace("git+", "")
+    return url
+
+
+def is_url(url, check=True, max_attempts=3, timeout=2):
+    """Check if string is URL and check if URL exists."""
+    allow_list = (
+        "localhost",
+        "127.0.0",
+        ":5000",
+        ":3000",
+        ":8000",
+        ":8080",
+        ":6006",
+        "MODEL_ID",
+        "API_KEY",
+        "url",
+        "example",
+        "mailto:",
+    )
+    try:
+        # Check allow list
+        if any(x in url for x in allow_list):
+            return True
+
+        # Check structure
+        result = urllib.parse.urlparse(url)
+        if not all([result.scheme, result.netloc]):
+            return False
+
+        # Check response
+        if check:
+            for attempt in range(max_attempts):
+                try:
+                    req = urllib.request.Request(url, method="HEAD", headers={"User-Agent": "Chrome/120.0.0.0"})
+                    with urllib.request.urlopen(req, timeout=timeout) as response:
+                        return response.getcode() < 400
+                except (urllib.error.URLError, socket.timeout):
+                    if attempt == max_attempts - 1:  # last attempt
+                        return False
+                    time.sleep(2**attempt)  # exponential backoff
+            return False
+        return True
+    except Exception:
+        return False
+
+
+def check_links_in_string(text, verbose=True, return_bad=False):
+    """Process a given text, find unique URLs within it, and check for any 404 errors."""
+    pattern = (
+        r"\[([^\]]+)\]\(([^)]+)\)"  # Matches Markdown links [text](url)
+        r"|"
+        r"("  # Start capturing group for plaintext URLs
+        r"(?:https?://)?"  # Optional http:// or https://
+        r"(?:www\.)?"  # Optional www.
+        r"[\w.-]+"  # Domain name and subdomains
+        r"\.[a-zA-Z]{2,}"  # TLD
+        r"(?:/[^\s\"')\]]*)?"  # Optional path
+        r")"
+    )
+    all_urls = []
+    for md_text, md_url, plain_url in re.findall(pattern, text):
+        url = md_url or plain_url
+        if url and urllib.parse.urlparse(url).scheme:
+            all_urls.append(url)
+
+    urls = set(map(clean_url, all_urls))  # remove extra characters and make unique
+    with ThreadPoolExecutor(max_workers=16) as executor:  # multi-thread
+        bad_urls = [url for url, valid in zip(urls, executor.map(lambda x: not is_url(x, check=True), urls)) if valid]
+
+    passing = not bad_urls
+    if verbose and not passing:
+        print(f"WARNING ⚠️ errors found in URLs {bad_urls}")
+
+    return (passing, bad_urls) if return_bad else passing

{ultralytics_actions-0.0.6 → ultralytics_actions-0.0.8}/actions/utils/github_utils.py

@@ -1,6 +1,7 @@
 # Ultralytics Actions 🚀, AGPL-3.0 license https://ultralytics.com/license
-
+import json
 import os
+from pathlib import Path
 
 import requests
 
@@ -85,3 +86,44 @@ def check_pypi_version(pyproject_toml="pyproject.toml"):
         publish = True  # publish as this is likely a first release
 
     return local_version, online_version, publish
+
+
+def ultralytics_actions_info():
+    """Print Ultralytics Actions information."""
+    event_data = {}
+    if GITHUB_EVENT_PATH:
+        event_path = Path(GITHUB_EVENT_PATH)
+        if event_path.exists():
+            event_data = json.loads(event_path.read_text())
+
+    pr = event_data.get("pull_request", {})
+    pr_head_ref = pr.get("head", {}).get("ref")
+
+    info = {
+        "github.event_name": GITHUB_EVENT_NAME,
+        "github.event.action": event_data.get("action"),
+        "github.repository": REPO_NAME,
+        "github.event.pull_request.number": pr.get("number"),
+        "github.event.pull_request.head.repo.full_name": pr.get("head", {}).get("repo", {}).get("full_name"),
+        "github.actor": os.environ.get("GITHUB_ACTOR"),
+        "github.event.pull_request.head.ref": pr_head_ref,
+        "github.ref": os.environ.get("GITHUB_REF"),
+        "github.head_ref": os.environ.get("GITHUB_HEAD_REF"),
+        "github.base_ref": os.environ.get("GITHUB_BASE_REF"),
+    }
+
+    if GITHUB_EVENT_NAME == "discussion":
+        discussion = event_data.get("discussion", {})
+        info.update(
+            {
+                "github.event.discussion.node_id": discussion.get("node_id"),
+                "github.event.discussion.number": discussion.get("number"),
+            }
+        )
+
+    # Print information
+    max_key_length = max(len(key) for key in info.keys())
+    print("Ultralytics Actions Information " + "-" * 40)  # header (72 long)
+    for key, value in info.items():
+        print(f"{key:<{max_key_length + 5}}{value}")
+    print("-" * 72)  # footer

ultralytics_actions-0.0.8/actions/utils/openai_utils.py

@@ -0,0 +1,45 @@
+# Ultralytics Actions 🚀, AGPL-3.0 license https://ultralytics.com/license
+
+import os
+import random
+from typing import Dict, List
+
+import requests
+
+from actions.utils.common_utils import check_links_in_string
+
+OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-4o")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
+
+def get_completion(
+    messages: List[Dict[str, str]],
+    check_links: bool = True,
+    remove: List[str] = (" @giscus[bot]",),  # strings to remove from response
+) -> str:
+    """Generates a completion using OpenAI's API based on input messages."""
+    assert OPENAI_API_KEY, "OpenAI API key is required."
+    url = "https://api.openai.com/v1/chat/completions"
+    headers = {"Authorization": f"Bearer {OPENAI_API_KEY}", "Content-Type": "application/json"}
+
+    content = ""
+    max_retries = 2
+    for attempt in range(max_retries + 2):  # attempt = [0, 1, 2, 3], 2 random retries before asking for no links
+        data = {"model": OPENAI_MODEL, "messages": messages, "seed": random.randint(1, 1000000)}
+
+        r = requests.post(url, headers=headers, json=data)
+        r.raise_for_status()
+        content = r.json()["choices"][0]["message"]["content"].strip()
+        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
+            return content
+
+        if attempt < max_retries:
+            print(f"Attempt {attempt + 1}: Found bad URLs. Retrying with a new random seed.")
+        else:
+            print("Max retries reached. Updating prompt to exclude links.")
+            messages.append({"role": "user", "content": "Please provide a response without any URLs or links in it."})
+            check_links = False  # automatically accept the last message
+
+    return content

{ultralytics_actions-0.0.6 → ultralytics_actions-0.0.8}/pyproject.toml

@@ -87,6 +87,7 @@ ultralytics-actions-first-interaction = "actions.first_interaction:main"
 ultralytics-actions-summarize-pr = "actions.summarize_pr:main"
 ultralytics-actions-summarize-release = "actions.summarize_release:main"
 ultralytics-actions-update-markdown-code-blocks = "actions.update_markdown_code_blocks:main"
+ultralytics-actions-info = "actions.utils:ultralytics_actions_info"
 
 [tool.setuptools]
 packages = { find = { where = ["."], include = ["actions", "actions.*"] } }

{ultralytics_actions-0.0.6 → ultralytics_actions-0.0.8/ultralytics_actions.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics-actions
-Version: 0.0.6
+Version: 0.0.8
 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.6 → ultralytics_actions-0.0.8}/ultralytics_actions.egg-info/entry_points.txt

@@ -1,5 +1,6 @@
 [console_scripts]
 ultralytics-actions-first-interaction = actions.first_interaction:main
+ultralytics-actions-info = actions.utils:ultralytics_actions_info
 ultralytics-actions-summarize-pr = actions.summarize_pr:main
 ultralytics-actions-summarize-release = actions.summarize_release:main
 ultralytics-actions-update-markdown-code-blocks = actions.update_markdown_code_blocks:main

ultralytics_actions-0.0.6/actions/utils/common_utils.py

@@ -1,8 +0,0 @@
-# Ultralytics Actions 🚀, AGPL-3.0 license https://ultralytics.com/license
-
-import re
-
-
-def remove_html_comments(body: str) -> str:
-    """Removes HTML comments from a string using regex pattern matching."""
-    return re.sub(r"<!--.*?-->", "", body, flags=re.DOTALL).strip()

ultralytics_actions-0.0.6/actions/utils/openai_utils.py

@@ -1,24 +0,0 @@
-# Ultralytics Actions 🚀, AGPL-3.0 license https://ultralytics.com/license
-
-import os
-
-import requests
-
-OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-4o")
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-
-
-def get_completion(messages: list) -> str:
-    """Generates a completion using OpenAI's API based on input messages."""
-    assert OPENAI_API_KEY, "OpenAI API key is required."
-    url = "https://api.openai.com/v1/chat/completions"
-    headers = {"Authorization": f"Bearer {OPENAI_API_KEY}", "Content-Type": "application/json"}
-    data = {"model": OPENAI_MODEL, "messages": messages}
-
-    r = requests.post(url, headers=headers, json=data)
-    r.raise_for_status()
-    content = r.json()["choices"][0]["message"]["content"].strip()
-    remove = [" @giscus[bot]"]
-    for x in remove:
-        content = content.replace(x, "")
-    return content