codeflash 0.7.2__tar.gz → 0.7.4__tar.gz

{codeflash-0.7.2 → codeflash-0.7.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codeflash
-Version: 0.7.2
+Version: 0.7.4
 Summary: Client for codeflash.ai - automatic code performance optimization, powered by AI
 Home-page: https://codeflash.ai
 License: BSL-1.1
@@ -24,6 +24,7 @@ Requires-Dist: isort (>=5.11.0)
 Requires-Dist: jedi (>=0.19.1)
 Requires-Dist: junitparser (>=3.1.0)
 Requires-Dist: libcst (>=1.0.1)
+Requires-Dist: lxml (>=5.3.0,<6.0.0)
 Requires-Dist: parameterized (>=0.9.0)
 Requires-Dist: posthog (>=3.0.0)
 Requires-Dist: pydantic (>=1.10.1)

{codeflash-0.7.2 → codeflash-0.7.4}/codeflash/api/aiservice.py

@@ -9,7 +9,7 @@ import requests
 from pydantic.dataclasses import dataclass
 from pydantic.json import pydantic_encoder
 
-from codeflash.cli_cmds.console import logger
+from codeflash.cli_cmds.console import console, logger
 from codeflash.code_utils.env_utils import get_codeflash_api_key
 from codeflash.discovery.functions_to_optimize import FunctionToOptimize
 from codeflash.models.ExperimentMetadata import ExperimentMetadata
@@ -33,10 +33,7 @@ class OptimizedCandidate:
 class AiServiceClient:
     def __init__(self) -> None:
         self.base_url = self.get_aiservice_base_url()
-        self.headers = {
-            "Authorization": f"Bearer {get_codeflash_api_key()}",
-            "Connection": "close",
-        }
+        self.headers = {"Authorization": f"Bearer {get_codeflash_api_key()}", "Connection": "close"}
 
     def get_aiservice_base_url(self) -> str:
         if os.environ.get("CODEFLASH_AIS_SERVER", default="prod").lower() == "local":
@@ -45,11 +42,7 @@ class AiServiceClient:
         return "https://app.codeflash.ai"
 
     def make_ai_service_request(
-        self,
-        endpoint: str,
-        method: str = "POST",
-        payload: dict[str, Any] | None = None,
-        timeout: float | None = None,
+        self, endpoint: str, method: str = "POST", payload: dict[str, Any] | None = None, timeout: float | None = None
     ) -> requests.Response:
         """Make an API request to the given endpoint on the AI service.
 
@@ -96,13 +89,11 @@ class AiServiceClient:
             "experiment_metadata": experiment_metadata,
             "codeflash_version": codeflash_version,
         }
+
         logger.info("Generating optimized candidates ...")
+        console.rule()
         try:
-            response = self.make_ai_service_request(
-                "/optimize",
-                payload=payload,
-                timeout=600,
-            )
+            response = self.make_ai_service_request("/optimize", payload=payload, timeout=600)
         except requests.exceptions.RequestException as e:
             logger.exception(f"Error generating optimized candidates: {e}")
             ph("cli-optimize-error-caught", {"error": str(e)})
@@ -111,6 +102,7 @@ class AiServiceClient:
         if response.status_code == 200:
             optimizations_json = response.json()["optimizations"]
             logger.info(f"Generated {len(optimizations_json)} candidates.")
+            console.rule()
             return [
                 OptimizedCandidate(
                     source_code=opt["source_code"],
@@ -124,10 +116,8 @@ class AiServiceClient:
         except Exception:
             error = response.text
         logger.error(f"Error generating optimized candidates: {response.status_code} - {error}")
-        ph(
-            "cli-optimize-error-response",
-            {"response_status_code": response.status_code, "error": error},
-        )
+        ph("cli-optimize-error-response", {"response_status_code": response.status_code, "error": error})
+        console.rule()
         return []
 
     def log_results(
@@ -225,20 +215,17 @@ class AiServiceClient:
         try:
             error = response.json()["error"]
             logger.error(f"Error generating tests: {response.status_code} - {error}")
-            ph(
-                "cli-testgen-error-response",
-                {"response_status_code": response.status_code, "error": error},
-            )
+            ph("cli-testgen-error-response", {"response_status_code": response.status_code, "error": error})
             return None
         except Exception:
             logger.error(f"Error generating tests: {response.status_code} - {response.text}")
-            ph(
-                "cli-testgen-error-response",
-                {"response_status_code": response.status_code, "error": response.text},
-            )
+            ph("cli-testgen-error-response", {"response_status_code": response.status_code, "error": response.text})
             return None
 
 
 class LocalAiServiceClient(AiServiceClient):
+    """Client for interacting with the local AI service."""
+
     def get_aiservice_base_url(self) -> str:
+        """Get the base URL for the local AI service."""
         return "http://localhost:8000"

{codeflash-0.7.2 → codeflash-0.7.4}/codeflash/api/cfapi.py

@@ -4,17 +4,19 @@ import json
 import os
 from functools import lru_cache
 from pathlib import Path
-from typing import Any, Dict, Optional
+from typing import TYPE_CHECKING, Any, Dict, Optional
 
 import requests
 from pydantic.json import pydantic_encoder
-from requests import Response
 
 from codeflash.cli_cmds.console import logger
 from codeflash.code_utils.env_utils import ensure_codeflash_api_key, get_codeflash_api_key, get_pr_number
 from codeflash.code_utils.git_utils import get_repo_owner_and_name
 from codeflash.github.PrComment import FileDiffContent, PrComment
 
+if TYPE_CHECKING:
+    from requests import Response
+
 if os.environ.get("CODEFLASH_CFAPI_SERVER", default="prod").lower() == "local":
     CFAPI_BASE_URL = "http://localhost:3001"
     logger.info(f"Using local CF API at {CFAPI_BASE_URL}.")
@@ -22,12 +24,9 @@ else:
     CFAPI_BASE_URL = "https://app.codeflash.ai"
 
 
-def make_cfapi_request(
-    endpoint: str,
-    method: str,
-    payload: Optional[Dict[str, Any]] = None,
-) -> requests.Response:
+def make_cfapi_request(endpoint: str, method: str, payload: dict[str, Any] | None = None) -> Response:
     """Make an HTTP request using the specified method, URL, headers, and JSON payload.
+
     :param endpoint: The endpoint URL to send the request to.
     :param method: The HTTP method to use ('GET', 'POST', etc.).
     :param payload: Optional JSON payload to include in the POST request body.
@@ -38,15 +37,16 @@ def make_cfapi_request(
     if method.upper() == "POST":
         json_payload = json.dumps(payload, indent=None, default=pydantic_encoder)
         cfapi_headers["Content-Type"] = "application/json"
-        response = requests.post(url, data=json_payload, headers=cfapi_headers)
+        response = requests.post(url, data=json_payload, headers=cfapi_headers, timeout=60)
     else:
-        response = requests.get(url, headers=cfapi_headers)
+        response = requests.get(url, headers=cfapi_headers, timeout=60)
     return response
 
 
 @lru_cache(maxsize=1)
 def get_user_id() -> Optional[str]:
     """Retrieve the user's userid by making a request to the /cfapi/cli-get-user endpoint.
+
     :return: The userid or None if the request fails.
     """
     if not ensure_codeflash_api_key():
@@ -55,11 +55,8 @@ def get_user_id() -> Optional[str]:
     response = make_cfapi_request(endpoint="/cli-get-user", method="GET")
     if response.status_code == 200:
         return response.text
-    else:
-        logger.error(
-            f"Failed to look up your userid; is your CF API key valid? ({response.reason})",
-        )
-        return None
+    logger.error(f"Failed to look up your userid; is your CF API key valid? ({response.reason})")
+    return None
 
 
 def suggest_changes(
@@ -73,6 +70,7 @@ def suggest_changes(
     trace_id: str,
 ) -> Response:
     """Suggest changes to a pull request.
+
     Will make a review suggestion when possible;
     or create a new dependent pull request with the suggested changes.
     :param owner: The owner of the repository.
@@ -93,8 +91,7 @@ def suggest_changes(
         "generatedTests": generated_tests,
         "traceId": trace_id,
     }
-    response = make_cfapi_request(endpoint="/suggest-pr-changes", method="POST", payload=payload)
-    return response
+    return make_cfapi_request(endpoint="/suggest-pr-changes", method="POST", payload=payload)
 
 
 def create_pr(
@@ -107,7 +104,8 @@ def create_pr(
     generated_tests: str,
     trace_id: str,
 ) -> Response:
-    """Create a pull request, targeting the specified branch. (usually 'main')
+    """Create a pull request, targeting the specified branch. (usually 'main').
+
     :param owner: The owner of the repository.
     :param repo: The name of the repository.
     :param base_branch: The base branch to target.
@@ -133,37 +131,41 @@ def create_pr(
 
 def is_github_app_installed_on_repo(owner: str, repo: str) -> bool:
     """Check if the Codeflash GitHub App is installed on the specified repository.
+
     :param owner: The owner of the repository.
     :param repo: The name of the repository.
     :return: The response object.
     """
-    response = make_cfapi_request(
-        endpoint=f"/is-github-app-installed?repo={repo}&owner={owner}",
-        method="GET",
-    )
+    response = make_cfapi_request(endpoint=f"/is-github-app-installed?repo={repo}&owner={owner}", method="GET")
     if not response.ok or response.text != "true":
         logger.error(f"Error: {response.text}")
         return False
     return True
 
 
-def get_blocklisted_functions() -> dict[str, str]:
+def get_blocklisted_functions() -> dict[str, set[str]]:
+    """Retrieve blocklisted functions for the current pull request.
+
+    Returns A dictionary mapping filenames to sets of blocklisted function names.
+    """
     pr_number = get_pr_number()
     if pr_number is None:
         return {}
 
+    not_found = 404
+    internal_server_error = 500
+
     owner, repo = get_repo_owner_and_name()
-    information = {
-        "pr_number": pr_number,
-        "repo_owner": owner,
-        "repo_name": repo,
-    }
+    information = {"pr_number": pr_number, "repo_owner": owner, "repo_name": repo}
     try:
-        req = make_cfapi_request(
-            endpoint="/verify-existing-optimizations",
-            method="POST",
-            payload=information,
-        )
+        req = make_cfapi_request(endpoint="/verify-existing-optimizations", method="POST", payload=information)
+        if req.status_code == not_found:
+            logger.debug(req.json()["message"])
+            return {}
+        if req.status_code == internal_server_error:
+            logger.error(req.json()["message"])
+            return {}
+        req.raise_for_status()
         content: dict[str, list[str]] = req.json()
     except Exception as e:
         logger.error(f"Error getting blocklisted functions: {e}")

{codeflash-0.7.2 → codeflash-0.7.4}/codeflash/cli_cmds/cli.py

@@ -31,10 +31,7 @@ def parse_args() -> Namespace:
     init_actions_parser = subparsers.add_parser("init-actions", help="Initialize GitHub Actions workflow")
     init_actions_parser.set_defaults(func=install_github_actions)
     parser.add_argument("--file", help="Try to optimize only this file")
-    parser.add_argument(
-        "--function",
-        help="Try to optimize only this function within the given file path",
-    )
+    parser.add_argument("--function", help="Try to optimize only this function within the given file path")
     parser.add_argument(
         "--all",
         help="Try to optimize all functions. Can take a really long time. Can pass an optional starting directory to"
@@ -50,30 +47,16 @@ def parse_args() -> Namespace:
         " This is the top-level root directory where all the Python source code is located.",
     )
     parser.add_argument(
-        "--tests-root",
-        type=str,
-        help="Path to the test directory of the project, where all the tests are located.",
+        "--tests-root", type=str, help="Path to the test directory of the project, where all the tests are located."
     )
     parser.add_argument("--test-framework", choices=["pytest", "unittest"], default="pytest")
+    parser.add_argument("--config-file", type=str, help="Path to the pyproject.toml with codeflash configs.")
     parser.add_argument(
-        "--config-file",
-        type=str,
-        help="Path to the pyproject.toml with codeflash configs.",
+        "--use-cached-tests", action="store_true", help="Use cached tests from a specified file for debugging."
     )
+    parser.add_argument("--replay-test", type=str, help="Path to replay test to optimize functions from")
     parser.add_argument(
-        "--use-cached-tests",
-        action="store_true",
-        help="Use cached tests from a specified file for debugging.",
-    )
-    parser.add_argument(
-        "--replay-test",
-        type=str,
-        help="Path to replay test to optimize functions from",
-    )
-    parser.add_argument(
-        "--no-pr",
-        action="store_true",
-        help="Do not create a PR for the optimization, only update the code locally.",
+        "--no-pr", action="store_true", help="Do not create a PR for the optimization, only update the code locally."
     )
     parser.add_argument(
         "--verify-setup",
@@ -170,6 +153,7 @@ def process_pyproject_config(args: Namespace) -> Namespace:
     # in this case, the ".." becomes outside project scope, causing issues with un-importable paths
     args.project_root = project_root_from_module_root(args.module_root, pyproject_file_path)
     args.tests_root = Path(args.tests_root).resolve()
+    args.test_project_root = project_root_from_module_root(args.tests_root, pyproject_file_path)
     return handle_optimize_all_arg_parsing(args)
 
 
@@ -187,7 +171,7 @@ def handle_optimize_all_arg_parsing(args: Namespace) -> Namespace:
         except git.exc.InvalidGitRepositoryError:
             logger.exception(
                 "I couldn't find a git repository in the current directory. "
-                "I need a git repository to run --all and open PRs for optimizations. Exiting...",
+                "I need a git repository to run --all and open PRs for optimizations. Exiting..."
             )
             apologize_and_exit()
         if not args.no_pr and not check_and_push_branch(git_repo):

{codeflash-0.7.2 → codeflash-0.7.4}/codeflash/cli_cmds/cli_common.py

@@ -10,17 +10,13 @@ import inquirer
 
 def apologize_and_exit() -> None:
     click.echo(
-        "💡 If you're having trouble, see https://docs.codeflash.ai/getting-started/local-installation for further help getting started with Codeflash!",
+        "💡 If you're having trouble, see https://docs.codeflash.ai/getting-started/local-installation for further help getting started with Codeflash!"
     )
     click.echo("👋 Exiting...")
     sys.exit(1)
 
 
-def inquirer_wrapper(
-    func: Callable[..., str | bool],
-    *args: str | bool,
-    **kwargs: str | bool,
-) -> str | bool:
+def inquirer_wrapper(func: Callable[..., str | bool], *args: str | bool, **kwargs: str | bool) -> str | bool:
     new_args = []
     new_kwargs = {}
 
@@ -29,10 +25,7 @@ def inquirer_wrapper(
     else:
         message = kwargs["message"]
         new_kwargs = kwargs.copy()
-    split_messages = split_string_to_cli_width(
-        message,
-        is_confirm=func == inquirer.confirm,
-    )
+    split_messages = split_string_to_cli_width(message, is_confirm=func == inquirer.confirm)
     for split_message in split_messages[:-1]:
         click.echo(split_message)
 
@@ -77,14 +70,7 @@ def inquirer_wrapper_path(*args: str, **kwargs: str) -> dict[str, str]:
     new_kwargs["message"] = last_message
     new_args.append(args[0])
 
-    return cast(
-        dict[str, str],
-        inquirer.prompt(
-            [
-                inquirer.Path(*new_args, **new_kwargs),
-            ],
-        ),
-    )
+    return cast(dict[str, str], inquirer.prompt([inquirer.Path(*new_args, **new_kwargs)]))
 
 
 def split_string_to_fit_width(string: str, width: int) -> list[str]:

{codeflash-0.7.2 → codeflash-0.7.4}/codeflash/cli_cmds/cmd_init.py

@@ -21,18 +21,10 @@ from codeflash.api.cfapi import is_github_app_installed_on_repo
 from codeflash.cli_cmds.cli_common import apologize_and_exit, inquirer_wrapper, inquirer_wrapper_path
 from codeflash.code_utils.compat import LF
 from codeflash.code_utils.config_parser import parse_config_file
-from codeflash.code_utils.env_utils import (
-    get_codeflash_api_key,
-)
+from codeflash.code_utils.env_utils import get_codeflash_api_key
 from codeflash.code_utils.git_utils import get_repo_owner_and_name
-from codeflash.code_utils.github_utils import (
-    get_github_secrets_page_url,
-    require_github_app_or_exit,
-)
-from codeflash.code_utils.shell_utils import (
-    get_shell_rc_path,
-    save_api_key_to_rc,
-)
+from codeflash.code_utils.github_utils import get_github_secrets_page_url, require_github_app_or_exit
+from codeflash.code_utils.shell_utils import get_shell_rc_path, save_api_key_to_rc
 from codeflash.telemetry.posthog_cf import ph
 from codeflash.version import __version__ as version
 
@@ -78,12 +70,10 @@ def init_codeflash() -> None:
             f"    codeflash --file <path-to-file> to optimize all functions in a file{LF}"
             f"    codeflash --all to optimize all functions in all files in the module you selected ({setup_info.module_root}){LF}"
             f"-or-{LF}"
-            f"    codeflash --help to see all options{LF}",
+            f"    codeflash --help to see all options{LF}"
         )
         if did_add_new_key:
-            click.echo(
-                "🐚 Don't forget to restart your shell to load the CODEFLASH_API_KEY environment variable!",
-            )
+            click.echo("🐚 Don't forget to restart your shell to load the CODEFLASH_API_KEY environment variable!")
             click.echo("Or run the following command to reload:")
             if os.name == "nt":
                 click.echo(f"  call {get_shell_rc_path()}")
@@ -111,30 +101,16 @@ def collect_setup_info() -> SetupInfo:
     curdir = Path.cwd()
     # Check if the cwd is writable
     if not os.access(curdir, os.W_OK):
-        click.echo(
-            f"❌ The current directory isn't writable, please check your folder permissions and try again.{LF}",
-        )
+        click.echo(f"❌ The current directory isn't writable, please check your folder permissions and try again.{LF}")
         click.echo("It's likely you don't have write permissions for this folder.")
         sys.exit(1)
 
     # Check for the existence of pyproject.toml or setup.py
     project_name = check_for_toml_or_setup_file()
 
-    ignore_subdirs = [
-        "venv",
-        "node_modules",
-        "dist",
-        "build",
-        "build_temp",
-        "build_scripts",
-        "env",
-        "logs",
-        "tmp",
-    ]
+    ignore_subdirs = ["venv", "node_modules", "dist", "build", "build_temp", "build_scripts", "env", "logs", "tmp"]
     valid_subdirs = [
-        d
-        for d in next(os.walk("."))[1]
-        if not d.startswith(".") and not d.startswith("__") and d not in ignore_subdirs
+        d for d in next(os.walk("."))[1] if not d.startswith(".") and not d.startswith("__") and d not in ignore_subdirs
     ]
 
     valid_module_subdirs = [d for d in valid_subdirs if d != "tests"]
@@ -165,9 +141,7 @@ def collect_setup_info() -> SetupInfo:
         message="Where are your tests located? "
         f"(If you don't have any tests yet, I can create an empty tests{os.pathsep} directory for you)",
         choices=test_subdir_options,
-        default=(
-            default_tests_subdir if default_tests_subdir in test_subdir_options else test_subdir_options[0]
-        ),
+        default=(default_tests_subdir if default_tests_subdir in test_subdir_options else test_subdir_options[0]),
     )
 
     if tests_root_answer == create_for_me_option:
@@ -285,11 +259,7 @@ def check_for_toml_or_setup_file() -> str | None:
     else:
         if setup_py_path.exists():
             setup_py_content = setup_py_path.read_text(encoding="utf8")
-            project_name_match = re.search(
-                r"setup\s*\([^)]*?name\s*=\s*['\"](.*?)['\"]",
-                setup_py_content,
-                re.DOTALL,
-            )
+            project_name_match = re.search(r"setup\s*\([^)]*?name\s*=\s*['\"](.*?)['\"]", setup_py_content, re.DOTALL)
             if project_name_match:
                 project_name = project_name_match.group(1)
                 click.echo(f"✅ Found setup.py for your project {project_name}")
@@ -300,7 +270,7 @@ def check_for_toml_or_setup_file() -> str | None:
         click.echo(
             f"💡 I couldn't find a pyproject.toml in the current directory ({curdir}).{LF}"
             f"(make sure you're running `codeflash init` from your project's root directory!){LF}"
-            f"I need this file to store my configuration settings.",
+            f"I need this file to store my configuration settings."
         )
         ph("cli-no-pyproject-toml-or-setup-py")
 
@@ -321,14 +291,12 @@ def check_for_toml_or_setup_file() -> str | None:
 
                 # Check if the pyproject.toml file was created
                 if pyproject_toml_path.exists():
-                    click.echo(
-                        f"✅ Created a pyproject.toml file at {pyproject_toml_path}",
-                    )
+                    click.echo(f"✅ Created a pyproject.toml file at {pyproject_toml_path}")
                     click.pause()
                 ph("cli-created-pyproject-toml")
             except OSError:
                 click.echo(
-                    "❌ Failed to create pyproject.toml. Please check your disk permissions and available space.",
+                    "❌ Failed to create pyproject.toml. Please check your disk permissions and available space."
                 )
                 apologize_and_exit()
         else:
@@ -341,7 +309,7 @@ def check_for_toml_or_setup_file() -> str | None:
 def install_github_actions() -> None:
     try:
         click.echo(
-            "⚡️ Codeflash can automatically optimize new Github PRs for you when they're opened. Let's get that set up!",
+            "⚡️ Codeflash can automatically optimize new Github PRs for you when they're opened. Let's get that set up!"
         )
         config, config_file_path = parse_config_file()
 
@@ -360,10 +328,7 @@ def install_github_actions() -> None:
             message=f"I'm going to create a new GitHub actions workflow file at {optimize_yaml_path} ... is this OK?",
             default=True,
         )
-        ph(
-            "cli-github-optimization-confirm-workflow-creation",
-            {"confirm_creation": confirm_creation_yes},
-        )
+        ph("cli-github-optimization-confirm-workflow-creation", {"confirm_creation": confirm_creation_yes})
         if not confirm_creation_yes:
             click.echo("⏩️ Exiting workflow creation.")
             ph("cli-github-workflow-skipped")
@@ -374,14 +339,9 @@ def install_github_actions() -> None:
         py_version = sys.version_info
         python_version_string = f"'{py_version.major}.{py_version.minor}'"
         optimize_yml_content = (
-            files("codeflash")
-            .joinpath("cli_cmds", "workflows", "codeflash-optimize.yaml")
-            .read_text(encoding="utf-8")
-        )
-        optimize_yml_content = optimize_yml_content.replace(
-            "{{ python_version }}",
-            python_version_string,
+            files("codeflash").joinpath("cli_cmds", "workflows", "codeflash-optimize.yaml").read_text(encoding="utf-8")
         )
+        optimize_yml_content = optimize_yml_content.replace("{{ python_version }}", python_version_string)
         with optimize_yaml_path.open("w", encoding="utf8") as optimize_yml_file:
             optimize_yml_file.write(optimize_yml_content)
         click.echo(f"✅ Created {optimize_yaml_path}{LF}")
@@ -398,7 +358,7 @@ def install_github_actions() -> None:
         click.echo(
             "🐙 I opened your Github secrets page! Note: if you see a 404, you probably don't have access to this "
             "repo's secrets; ask a repo admin to add it for you, or (not super recommended) you can temporarily "
-            f"hard-code your api key into the workflow file.{LF}",
+            f"hard-code your api key into the workflow file.{LF}"
         )
         click.pause()
         click.echo()
@@ -414,13 +374,13 @@ def install_github_actions() -> None:
         click.launch(optimize_yaml_path.as_posix())
         click.echo(
             "📝 I opened the workflow file in your editor! You'll need to edit the steps that install the right Python "
-            f"version and any project dependencies. See the comments in the file for more details.{LF}",
+            f"version and any project dependencies. See the comments in the file for more details.{LF}"
         )
         click.pause()
         click.echo()
         click.echo(
             f"Please commit and push this GitHub actions file to your repo, and you're all set!{LF}"
-            f"🚀 Codeflash is now configured to automatically optimize new Github PRs!{LF}",
+            f"🚀 Codeflash is now configured to automatically optimize new Github PRs!{LF}"
         )
         ph("cli-github-workflow-created")
     except KeyboardInterrupt:
@@ -436,14 +396,12 @@ def configure_pyproject_toml(setup_info: SetupInfo) -> None:
     except FileNotFoundError:
         click.echo(
             f"I couldn't find a pyproject.toml in the current directory.{LF}"
-            f"Please create a new empty pyproject.toml file here, OR if you use poetry then run `poetry init`, OR run `codeflash init` again from a directory with an existing pyproject.toml file.",
+            f"Please create a new empty pyproject.toml file here, OR if you use poetry then run `poetry init`, OR run `codeflash init` again from a directory with an existing pyproject.toml file."
         )
         apologize_and_exit()
 
     codeflash_section = tomlkit.table()
-    codeflash_section.add(
-        tomlkit.comment("All paths are relative to this pyproject.toml's directory."),
-    )
+    codeflash_section.add(tomlkit.comment("All paths are relative to this pyproject.toml's directory."))
     codeflash_section["module-root"] = setup_info.module_root
     codeflash_section["tests-root"] = setup_info.tests_root
     codeflash_section["test-framework"] = setup_info.test_framework
@@ -457,7 +415,7 @@ def configure_pyproject_toml(setup_info: SetupInfo) -> None:
     elif formatter == "other":
         formatter_cmds.append("your-formatter $file")
         click.echo(
-            "🔧 In pyproject.toml, please replace 'your-formatter' with the command you use to format your code.",
+            "🔧 In pyproject.toml, please replace 'your-formatter' with the command you use to format your code."
         )
     elif formatter == "don't use a formatter":
         formatter_cmds.append("disabled")
@@ -483,9 +441,7 @@ def install_github_app() -> None:
     owner, repo = get_repo_owner_and_name(git_repo)
 
     if is_github_app_installed_on_repo(owner, repo):
-        click.echo(
-            "🐙 Looks like you've already installed the Codeflash GitHub app on this repository! Continuing…",
-        )
+        click.echo("🐙 Looks like you've already installed the Codeflash GitHub app on this repository! Continuing…")
 
     else:
         click.prompt(
@@ -512,7 +468,7 @@ def install_github_app() -> None:
                 click.echo(
                     f"❌ It looks like the Codeflash GitHub App is not installed on the repository {owner}/{repo}.{LF}"
                     f"You won't be able to create PRs with Codeflash until you install the app.{LF}"
-                    f"In the meantime you can make local only optimizations by using the '--no-pr' flag with codeflash.{LF}",
+                    f"In the meantime you can make local only optimizations by using the '--no-pr' flag with codeflash.{LF}"
                 )
                 break
             click.prompt(
@@ -549,15 +505,10 @@ def prompt_api_key() -> bool:
         existing_api_key = None
     if existing_api_key:
         display_key = f"{existing_api_key[:3]}****{existing_api_key[-4:]}"
-        click.echo(
-            f"🔑 I found a CODEFLASH_API_KEY in your environment [{display_key}]!",
-        )
+        click.echo(f"🔑 I found a CODEFLASH_API_KEY in your environment [{display_key}]!")
 
         use_existing_key = inquirer_wrapper(
-            inquirer.confirm,
-            message="Do you want to use this key?",
-            default=True,
-            show_default=False,
+            inquirer.confirm, message="Do you want to use this key?", default=True, show_default=False
         )
         if use_existing_key:
             ph("cli-existing-api-key-used")
@@ -584,7 +535,7 @@ def enter_api_key_and_save_to_rc() -> None:
         if not browser_launched:
             click.echo(
                 f"Opening your Codeflash API key page. Grab a key from there!{LF}"
-                "You can also open this link manually: https://app.codeflash.ai/app/apikeys",
+                "You can also open this link manually: https://app.codeflash.ai/app/apikeys"
             )
             click.launch("https://app.codeflash.ai/app/apikeys")
             browser_launched = True  # This does not work on remote consoles
@@ -664,22 +615,11 @@ def test_sort():
 
 
 def run_end_to_end_test(args: Namespace, bubble_sort_path: str, bubble_sort_test_path: str) -> None:
-    command = [
-        "codeflash",
-        "--file",
-        "bubble_sort.py",
-        "--function",
-        "sorter",
-    ]
+    command = ["codeflash", "--file", "bubble_sort.py", "--function", "sorter"]
     sys.stdout.write("Running sample optimization... ")
     sys.stdout.flush()
     try:
-        process = subprocess.run(
-            command,
-            text=True,
-            cwd=args.module_root,
-            check=False,
-        )
+        process = subprocess.run(command, text=True, cwd=args.module_root, check=False)
     finally:
         # Delete the bubble_sort.py file after the test
         Path(bubble_sort_path).unlink(missing_ok=True)
@@ -688,10 +628,8 @@ def run_end_to_end_test(args: Namespace, bubble_sort_path: str, bubble_sort_test
         click.echo(f"{LF}🗑️ Deleted {bubble_sort_test_path}")
 
     if process.returncode == 0:
-        click.echo(
-            f"{LF}✅ End-to-end test passed. Codeflash has been correctly set up!",
-        )
+        click.echo(f"{LF}✅ End-to-end test passed. Codeflash has been correctly set up!")
     else:
         click.echo(
-            f"{LF}❌ End-to-end test failed. Please check the logs above, and take a look at https://docs.codeflash.ai/getting-started/local-installation for help and troubleshooting.",
+            f"{LF}❌ End-to-end test failed. Please check the logs above, and take a look at https://docs.codeflash.ai/getting-started/local-installation for help and troubleshooting."
         )