ultralytics-actions 0.0.71__tar.gz → 0.0.73__tar.gz

{ultralytics_actions-0.0.71/ultralytics_actions.egg-info → ultralytics_actions-0.0.73}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-actions
-Version: 0.0.71
+Version: 0.0.73
 Summary: Ultralytics Actions for GitHub automation and PR management.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>
@@ -33,6 +33,7 @@ Requires-Dist: ruff>=0.9.1
 Requires-Dist: docformatter>=1.7.5
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
 Dynamic: license-file
 
 <a href="https://www.ultralytics.com/"><img src="https://raw.githubusercontent.com/ultralytics/assets/main/logo/Ultralytics_Logotype_Original.svg" width="320" alt="Ultralytics logo"></a>
@@ -42,12 +43,16 @@ Dynamic: license-file
 Welcome to the [Ultralytics Actions](https://github.com/ultralytics/actions) repository, your go-to solution for maintaining consistent code quality across Ultralytics Python and Swift projects. This GitHub Action is designed to automate the formatting of Python, Markdown, and Swift files, ensuring adherence to our coding standards and enhancing project maintainability.
 
 [![GitHub Actions Marketplace](https://img.shields.io/badge/Marketplace-Ultralytics_Actions-blue?style=flat&logo=github)](https://github.com/marketplace/actions/ultralytics-actions)
+
+[![Actions CI](https://github.com/ultralytics/actions/actions/workflows/ci.yml/badge.svg)](https://github.com/ultralytics/actions/actions/workflows/ci.yml)
 [![Ultralytics Actions](https://github.com/ultralytics/actions/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/actions/actions/workflows/format.yml)
+[![codecov](https://codecov.io/github/ultralytics/actions/graph/badge.svg?token=DoizJ1WS6j)](https://codecov.io/github/ultralytics/actions)
+[![PyPI version](https://badge.fury.io/py/ultralytics-actions.svg)](https://badge.fury.io/py/ultralytics-actions)
+[![Downloads](https://static.pepy.tech/badge/ultralytics-actions)](https://www.pepy.tech/projects/ultralytics-actions)
+
 [![Ultralytics Discord](https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue)](https://discord.com/invite/ultralytics)
 [![Ultralytics Forums](https://img.shields.io/discourse/users?server=https%3A%2F%2Fcommunity.ultralytics.com&logo=discourse&label=Forums&color=blue)](https://community.ultralytics.com/)
 [![Ultralytics Reddit](https://img.shields.io/reddit/subreddit-subscribers/ultralytics?style=flat&logo=reddit&logoColor=white&label=Reddit&color=blue)](https://reddit.com/r/ultralytics)
-[![PyPI version](https://badge.fury.io/py/ultralytics-actions.svg)](https://badge.fury.io/py/ultralytics-actions)
-[![Downloads](https://static.pepy.tech/badge/ultralytics-actions)](https://www.pepy.tech/projects/ultralytics-actions)
 
 ## 📄 Actions Description
 
@@ -84,6 +89,11 @@ To integrate this action into your Ultralytics repository:
 2.  **Add the Action:** Configure the Ultralytics Actions in your workflow file as shown below:
 
     ```yaml
+    # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+    # Ultralytics Actions https://github.com/ultralytics/actions
+    # This workflow formats code and documentation in PRs to Ultralytics standards
+
     name: Ultralytics Actions
 
     on:
@@ -91,24 +101,31 @@ To integrate this action into your Ultralytics repository:
         types: [opened]
       pull_request:
         branches: [main]
-        types: [opened, closed]
+        types: [opened, closed, synchronize, review_requested]
+
+    permissions:
+      contents: write # Modify code in PRs
+      pull-requests: write # Add comments and labels to PRs
+      issues: write # Add comments and labels to issues
 
     jobs:
-      format:
-        runs-on: ubuntu-latest # Use 'macos-latest' if 'swift: true'
+      actions:
+        runs-on: ubuntu-latest
         steps:
-          - name: Run Ultralytics Formatting
+          - name: Run Ultralytics Actions
             uses: ultralytics/actions@main
             with:
-              token: ${{ secrets.GITHUB_TOKEN }} # Automatically generated, do not modify
-              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')
+              token: ${{ secrets.GITHUB_TOKEN }} # Auto-generated token
+              labels: true # Auto-label issues/PRs using AI
+              python: true # Format Python with Ruff and docformatter
+              prettier: true # Format YAML, JSON, Markdown, CSS
+              swift: false # Format Swift (requires macos-latest)
+              dart: false # Format Dart/Flutter
               spelling: true # Check spelling with codespell
-              links: true # Check for broken links with Lychee
-              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
+              links: true # Check broken links with Lychee
+              summary: true # Generate AI-powered PR summaries
+              openai_api_key: ${{ secrets.OPENAI_API_KEY }} # Powers PR summaries, labels and comments
+              brave_api_key: ${{ secrets.BRAVE_API_KEY }} # Used for broken link resolution
     ```
 
 3.  **Customize:** Adjust the `runs-on` runner and the boolean flags (`labels`, `python`, `prettier`, `swift`, `spelling`, `links`, `summary`) based on your project's needs. Remember to add your `OPENAI_API_KEY` as a secret in your repository settings if you enable `labels` or `summary`.

{ultralytics_actions-0.0.71 → ultralytics_actions-0.0.73}/README.md

@@ -5,12 +5,16 @@
 Welcome to the [Ultralytics Actions](https://github.com/ultralytics/actions) repository, your go-to solution for maintaining consistent code quality across Ultralytics Python and Swift projects. This GitHub Action is designed to automate the formatting of Python, Markdown, and Swift files, ensuring adherence to our coding standards and enhancing project maintainability.
 
 [![GitHub Actions Marketplace](https://img.shields.io/badge/Marketplace-Ultralytics_Actions-blue?style=flat&logo=github)](https://github.com/marketplace/actions/ultralytics-actions)
+
+[![Actions CI](https://github.com/ultralytics/actions/actions/workflows/ci.yml/badge.svg)](https://github.com/ultralytics/actions/actions/workflows/ci.yml)
 [![Ultralytics Actions](https://github.com/ultralytics/actions/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/actions/actions/workflows/format.yml)
+[![codecov](https://codecov.io/github/ultralytics/actions/graph/badge.svg?token=DoizJ1WS6j)](https://codecov.io/github/ultralytics/actions)
+[![PyPI version](https://badge.fury.io/py/ultralytics-actions.svg)](https://badge.fury.io/py/ultralytics-actions)
+[![Downloads](https://static.pepy.tech/badge/ultralytics-actions)](https://www.pepy.tech/projects/ultralytics-actions)
+
 [![Ultralytics Discord](https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue)](https://discord.com/invite/ultralytics)
 [![Ultralytics Forums](https://img.shields.io/discourse/users?server=https%3A%2F%2Fcommunity.ultralytics.com&logo=discourse&label=Forums&color=blue)](https://community.ultralytics.com/)
 [![Ultralytics Reddit](https://img.shields.io/reddit/subreddit-subscribers/ultralytics?style=flat&logo=reddit&logoColor=white&label=Reddit&color=blue)](https://reddit.com/r/ultralytics)
-[![PyPI version](https://badge.fury.io/py/ultralytics-actions.svg)](https://badge.fury.io/py/ultralytics-actions)
-[![Downloads](https://static.pepy.tech/badge/ultralytics-actions)](https://www.pepy.tech/projects/ultralytics-actions)
 
 ## 📄 Actions Description
 
@@ -47,6 +51,11 @@ To integrate this action into your Ultralytics repository:
 2.  **Add the Action:** Configure the Ultralytics Actions in your workflow file as shown below:
 
     ```yaml
+    # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+    # Ultralytics Actions https://github.com/ultralytics/actions
+    # This workflow formats code and documentation in PRs to Ultralytics standards
+
     name: Ultralytics Actions
 
     on:
@@ -54,24 +63,31 @@ To integrate this action into your Ultralytics repository:
         types: [opened]
       pull_request:
         branches: [main]
-        types: [opened, closed]
+        types: [opened, closed, synchronize, review_requested]
+
+    permissions:
+      contents: write # Modify code in PRs
+      pull-requests: write # Add comments and labels to PRs
+      issues: write # Add comments and labels to issues
 
     jobs:
-      format:
-        runs-on: ubuntu-latest # Use 'macos-latest' if 'swift: true'
+      actions:
+        runs-on: ubuntu-latest
         steps:
-          - name: Run Ultralytics Formatting
+          - name: Run Ultralytics Actions
             uses: ultralytics/actions@main
             with:
-              token: ${{ secrets.GITHUB_TOKEN }} # Automatically generated, do not modify
-              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')
+              token: ${{ secrets.GITHUB_TOKEN }} # Auto-generated token
+              labels: true # Auto-label issues/PRs using AI
+              python: true # Format Python with Ruff and docformatter
+              prettier: true # Format YAML, JSON, Markdown, CSS
+              swift: false # Format Swift (requires macos-latest)
+              dart: false # Format Dart/Flutter
               spelling: true # Check spelling with codespell
-              links: true # Check for broken links with Lychee
-              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
+              links: true # Check broken links with Lychee
+              summary: true # Generate AI-powered PR summaries
+              openai_api_key: ${{ secrets.OPENAI_API_KEY }} # Powers PR summaries, labels and comments
+              brave_api_key: ${{ secrets.BRAVE_API_KEY }} # Used for broken link resolution
     ```
 
 3.  **Customize:** Adjust the `runs-on` runner and the boolean flags (`labels`, `python`, `prettier`, `swift`, `spelling`, `links`, `summary`) based on your project's needs. Remember to add your `OPENAI_API_KEY` as a secret in your repository settings if you enable `labels` or `summary`.

{ultralytics_actions-0.0.71 → ultralytics_actions-0.0.73}/actions/__init__.py

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

{ultralytics_actions-0.0.71 → ultralytics_actions-0.0.73}/actions/first_interaction.py

@@ -249,8 +249,7 @@ mutation($discussionId: ID!, $body: String!) {
 """
         event.graphql_request(mutation, variables={"discussionId": node_id, "body": comment})
     else:
-        url = f"{GITHUB_API_URL}/repos/{event.repository}/issues/{number}/comments"
-        event.post(url, json={"body": comment}, headers=event.headers)
+        event.post(f"{GITHUB_API_URL}/repos/{event.repository}/issues/{number}/comments", json={"body": comment})
 
 
 def get_first_interaction_response(event, issue_type: str, title: str, body: str, username: str) -> str:

ultralytics_actions-0.0.73/actions/update_file_headers.py

@@ -0,0 +1,191 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+import os
+from pathlib import Path
+
+from actions.utils import Action
+
+# Base header text
+HEADER = os.getenv("HEADER")
+
+# Map file extensions to comment styles
+COMMENT_MAP = {
+    # Python style
+    ".py": ("# ", None, None),
+    ".yml": ("# ", None, None),
+    ".yaml": ("# ", None, None),
+    ".toml": ("# ", None, None),
+    ".sh": ("# ", None, None),  # Bash scripts
+    ".bash": ("# ", None, None),  # Bash scripts
+    # C/C++/Java/JS style
+    ".c": ("// ", "/* ", " */"),  # C files
+    ".cpp": ("// ", "/* ", " */"),  # C++ files
+    ".h": ("// ", "/* ", " */"),  # C/C++ header files
+    ".hpp": ("// ", "/* ", " */"),  # C++ header files
+    ".swift": ("// ", "/* ", " */"),
+    ".js": ("// ", "/* ", " */"),
+    ".ts": ("// ", "/* ", " */"),  # TypeScript files
+    ".dart": ("// ", "/* ", " */"),  # Dart/Flutter files
+    ".rs": ("// ", "/* ", " */"),  # Rust files
+    ".java": ("// ", "/* ", " */"),  # Android Java
+    ".kt": ("// ", "/* ", " */"),  # Android Kotlin
+    # CSS style
+    ".css": (None, "/* ", " */"),
+    # HTML/XML style
+    ".html": (None, "<!-- ", " -->"),
+    ".xml": (None, "<!-- ", " -->"),  # Android XML
+    # MATLAB style
+    ".m": ("% ", None, None),
+}
+
+# Ignore these Paths (do not update their headers)
+IGNORE_PATHS = [
+    ".idea",
+    ".venv",
+    "env",
+    "node_modules",
+    ".git",
+    "__pycache__",
+    "mkdocs_github_authors.yaml",
+    # Build and distribution directories
+    "dist",
+    "build",
+    ".eggs",
+    "site",  # mkdocs build directory
+    # Generated code
+    "generated",
+    "auto_gen",
+    # Lock files
+    "lock",
+    # Minified files
+    ".min.js",
+    ".min.css",
+]
+
+
+def update_file(file_path, prefix, block_start, block_end, base_header):
+    """Update file with the correct header and proper spacing."""
+    try:
+        with open(file_path, encoding="utf-8") as f:
+            lines = f.readlines()
+    except Exception as e:
+        print(f"Error reading {file_path}: {e}")
+        return False
+
+    if not lines:
+        return False
+
+    # Format the header based on comment style
+    if prefix:
+        formatted_header = f"{prefix}{base_header}\n"
+    elif block_start and block_end:
+        formatted_header = f"{block_start}{base_header}{block_end}\n"
+    else:
+        formatted_header = f"# {base_header}\n"
+
+    # Keep shebang line if it exists
+    start_idx = 0
+    if lines and lines[0].startswith("#!"):
+        start_idx = 1
+
+    modified = False
+    new_lines = lines[:start_idx]
+    remaining_lines = lines[start_idx:]
+
+    # If first line is already the exact header we want
+    if remaining_lines and remaining_lines[0] == formatted_header:
+        # Check if spacing is correct
+        new_lines.append(remaining_lines[0])
+        if len(remaining_lines) > 1:
+            second_line = remaining_lines[1].strip()
+            if second_line == "" or second_line in ["#", "//", "/*", "*", "<!--", "%"]:
+                # Spacing is correct, append the rest
+                new_lines.extend(remaining_lines[1:])
+            else:
+                # Add blank line
+                new_lines.append("\n")
+                new_lines.extend(remaining_lines[1:])
+                modified = True
+        else:
+            # Only header exists, no need for blank line
+            pass
+    # Check if first line has AGPL but is not the exact header
+    elif remaining_lines and "AGPL" in remaining_lines[0] and remaining_lines[0] != formatted_header:
+        # Replace with proper header
+        new_lines.append(formatted_header)
+        modified = True
+
+        # Check if second line is blank or commented
+        if len(remaining_lines) > 1:
+            second_line = remaining_lines[1].strip()
+            if second_line == "" or second_line in ["#", "//", "/*", "*", "<!--", "%"]:
+                # Keep existing blank/comment line
+                new_lines.append(remaining_lines[1])
+                new_lines.extend(remaining_lines[2:])
+            else:
+                # Add blank line
+                new_lines.append("\n")
+                new_lines.extend(remaining_lines[1:])
+        else:
+            # Only header line, no need for blank line after
+            pass
+    # No header found, add it
+    else:
+        # Add header at the beginning
+        new_lines.append(formatted_header)
+        # Add blank line if content follows
+        if remaining_lines and remaining_lines[0].strip():
+            new_lines.append("\n")
+        new_lines.extend(remaining_lines)
+        modified = True
+
+    if modified:
+        try:
+            with open(file_path, "w", encoding="utf-8") as f:
+                f.writelines(new_lines)
+            return True
+        except Exception as e:
+            print(f"Error writing {file_path}: {e}")
+            return False
+
+    return False
+
+
+def main(*args, **kwargs):
+    """Automates file header updates for all files in the specified directory."""
+    event = Action(*args, **kwargs)
+
+    if "ultralytics" in event.repository.lower():
+        if event.is_repo_private() and event.repository.startswith("ultralytics/"):
+            from datetime import datetime
+
+            notice = f"Copyright © 2014-{datetime.now().year}"
+            header = f"Ultralytics Inc. 🚀 {notice} - CONFIDENTIAL - https://ultralytics.com - All Rights Reserved"
+        else:
+            header = "Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license"
+    elif HEADER and HEADER.lower() not in {"true", "false", "none"}:
+        header = HEADER
+    else:
+        return
+
+    directory = Path.cwd()
+    total = changed = unchanged = 0
+    for ext, comment_style in COMMENT_MAP.items():
+        prefix, block_start, block_end = comment_style
+
+        for file_path in directory.rglob(f"*{ext}"):
+            if any(part in str(file_path) for part in IGNORE_PATHS):
+                continue
+
+            total += 1
+            if update_file(file_path, prefix, block_start, block_end, header):
+                print(f"Updated: {file_path.relative_to(directory)}")
+                changed += 1
+            else:
+                unchanged += 1
+
+    print(f"Headers: {total}, Updated: {changed}, Unchanged: {unchanged}")
+
+
+if __name__ == "__main__":
+    main()

{ultralytics_actions-0.0.71 → ultralytics_actions-0.0.73}/actions/utils/github_utils.py

@@ -41,7 +41,7 @@ class Action:
         self.eyes_reaction_id = None
         self.verbose = verbose
 
-    def _request(self, method: str, url: str, headers=None, expected_status=None, **kwargs):
+    def _request(self, method: str, url: str, headers=None, expected_status=None, hard=False, **kwargs):
         """Unified request handler with error checking."""
         headers = headers or self.headers
         expected_status = expected_status or self._default_status[method.lower()]
@@ -55,11 +55,11 @@ class Action:
             if not success:
                 try:
                     error_detail = response.json()
-                    print(f"  Error: {error_detail.get('message', 'Unknown error')}")
-                except:
-                    print(f"  Error: {response.text[:100]}...")
+                    print(f"  ❌ Error: {error_detail.get('message', 'Unknown error')}")
+                except Exception as e:
+                    print(f"  ❌ Error: {response.text[:100]}... {e}")
 
-        if not success:
+        if not success and hard:
             response.raise_for_status()
 
         return response
@@ -91,6 +91,10 @@ class Action:
             return json.loads(Path(event_path).read_text())
         return {}
 
+    def is_repo_private(self) -> bool:
+        """Checks if the repository is public using event data or GitHub API if needed."""
+        return self.event_data.get("repository", {}).get("private")
+
     def get_username(self) -> str | None:
         """Gets username associated with the GitHub token."""
         response = self.post(GITHUB_GRAPHQL_URL, json={"query": "query { viewer { login } }"})
@@ -152,6 +156,7 @@ class Action:
             "github.event_name": self.event_name,
             "github.event.action": self.event_data.get("action"),
             "github.repository": self.repository,
+            "github.repository.private": self.is_repo_private(),
             "github.event.pull_request.number": self.pr.get("number"),
             "github.event.pull_request.head.repo.full_name": self.pr.get("head", {}).get("repo", {}).get("full_name"),
             "github.actor": os.environ.get("GITHUB_ACTOR"),

{ultralytics_actions-0.0.71 → ultralytics_actions-0.0.73}/actions/utils/openai_utils.py

@@ -50,7 +50,7 @@ def get_completion(
             "temperature": temperature,
         }
 
-        r = requests.post(url, headers=headers, json=data)
+        r = requests.post(url, json=data, headers=headers)
         r.raise_for_status()
         content = r.json()["choices"][0]["message"]["content"].strip()
         content = remove_outer_codeblocks(content)

{ultralytics_actions-0.0.71 → ultralytics_actions-0.0.73}/pyproject.toml

@@ -73,6 +73,7 @@ dependencies = [
 [project.optional-dependencies]
 dev = [
     "pytest",
+    "pytest-cov",
 ]
 
 [project.urls]
@@ -87,6 +88,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-header = "actions.update_file_headers:main"
 ultralytics-actions-info = "actions.utils:ultralytics_actions_info"
 
 [tool.setuptools]

ultralytics_actions-0.0.73/tests/test_cli_commands.py

@@ -0,0 +1,19 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+
+# Import CLI command modules
+from actions import (
+    first_interaction,
+    summarize_pr,
+    summarize_release,
+    update_markdown_code_blocks,
+)
+
+
+def test_importable_modules():
+    """Test that all modules can be imported without errors."""
+    # This is a simple test to ensure modules can be imported successfully
+    assert hasattr(first_interaction, "main")
+    assert hasattr(summarize_pr, "main")
+    assert hasattr(summarize_release, "main")
+    assert hasattr(update_markdown_code_blocks, "main")

ultralytics_actions-0.0.73/tests/test_common_utils.py

@@ -0,0 +1,63 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+from unittest.mock import MagicMock, patch
+
+from actions.utils.common_utils import (
+    allow_redirect,
+    brave_search,
+    clean_url,
+    remove_html_comments,
+)
+
+
+def test_remove_html_comments():
+    """Test removing HTML comments from strings."""
+    test_str = "Before <!-- Comment --> After"
+    assert remove_html_comments(test_str) == "Before  After"
+
+    # Multiline comment
+    test_str = "Before\n<!-- Comment\nline 2\nline 3 -->\nAfter"
+    assert remove_html_comments(test_str) == "Before\n\nAfter"
+
+    # No comments
+    test_str = "No comments here"
+    assert remove_html_comments(test_str) == "No comments here"
+
+
+def test_clean_url():
+    """Test cleaning URL strings."""
+    # Test removing quotes and trailing characters
+    assert clean_url('"https://example.com"') == "https://example.com"
+    assert clean_url("'https://example.com'") == "https://example.com"
+    assert clean_url("https://example.com.") == "https://example.com"
+    assert clean_url("https://example.com,") == "https://example.com"
+
+    # Test git URLs
+    assert clean_url("git+https://github.com/user/repo.git@main") == "https://github.com/user/repo"
+
+
+def test_allow_redirect():
+    """Test allowing URL redirects based on rules."""
+    # Should not allow - start ignores
+    assert not allow_redirect("https://youtu.be/xyz", "https://youtube.com")
+
+    # Should not allow - end ignores
+    assert not allow_redirect("https://example.com", "https://example.com/404")
+
+    # Empty end URL
+    assert not allow_redirect("https://example.com", "")
+
+
+@patch("requests.get")
+def test_brave_search(mock_get):
+    """Test Brave search API integration."""
+    mock_response = MagicMock()
+    mock_response.status_code = 200
+    mock_response.json.return_value = {
+        "web": {"results": [{"url": "https://example.com"}, {"url": "https://example.org"}]}
+    }
+    mock_get.return_value = mock_response
+
+    results = brave_search("test query", "test-api-key", count=2)
+    assert results == ["https://example.com", "https://example.org"]
+    mock_get.assert_called_once()