Merge remote-tracking branch 'origin/main' into blocking-wall-FOR

Author: KRRT7

codeflash/LICENSE

@@ -3,7 +3,7 @@ Business Source License 1.1
 Parameters
 
 Licensor:             CodeFlash Inc.
-Licensed Work:        Codeflash Client version 0.9.x
+Licensed Work:        Codeflash Client version 0.10.x
                       The Licensed Work is (c) 2024 CodeFlash Inc.
 
 Additional Use Grant: None. Production use of the Licensed Work is only permitted
@@ -13,7 +13,7 @@ Additional Use Grant: None. Production use of the Licensed Work is only permitte
                       Platform. Please visit codeflash.ai for further
                       information.
 
-Change Date:          2029-01-06
+Change Date:          2029-02-25
 
 Change License:       MIT
 

codeflash/cli_cmds/cmd_init.py

@@ -220,19 +220,26 @@ def collect_setup_info() -> SetupInfo:
         carousel=True,
     )
 
+    git_remote = ""
     try:
         repo = Repo(str(module_root), search_parent_directories=True)
         git_remotes = get_git_remotes(repo)
-        if len(git_remotes) > 1:
-            git_remote = inquirer_wrapper(
-                inquirer.list_input,
-                message="What git remote do you want Codeflash to use for new Pull Requests? ",
-                choices=git_remotes,
-                default="origin",
-                carousel=True,
-            )
+        if git_remotes:  # Only proceed if there are remotes
+            if len(git_remotes) > 1:
+                git_remote = inquirer_wrapper(
+                    inquirer.list_input,
+                    message="What git remote do you want Codeflash to use for new Pull Requests? ",
+                    choices=git_remotes,
+                    default="origin",
+                    carousel=True,
+                )
+            else:
+                git_remote = git_remotes[0]
         else:
-            git_remote = git_remotes[0]
+            click.echo(
+                "No git remotes found. You can still use Codeflash locally, but you'll need to set up a remote "
+                "repository to use GitHub features."
+            )
     except InvalidGitRepositoryError:
         git_remote = ""
 
@@ -372,6 +379,19 @@ def install_github_actions() -> None:
         workflows_path = git_root / ".github" / "workflows"
         optimize_yaml_path = workflows_path / "codeflash.yaml"
 
+        # Check if the workflow file already exists
+        if optimize_yaml_path.exists():
+            confirm_overwrite = inquirer_wrapper(
+                inquirer.confirm,
+                message=f"⚡️ GitHub Actions workflow already exists at {optimize_yaml_path}. Overwrite?",
+                default=False,  # Don't overwrite by default
+            )
+            ph("cli-github-optimization-confirm-workflow-overwrite", {"confirm_overwrite": confirm_overwrite})
+            if not confirm_overwrite:
+                click.echo("⏩️ Skipping workflow creation.")
+                ph("cli-github-workflow-skipped")
+                return
+
         confirm_creation_yes = inquirer_wrapper(
             inquirer.confirm,
             message="⚡️Shall I set up a GitHub action that will continuously optimize all new code in GitHub PRs"
@@ -574,6 +594,11 @@ def configure_pyproject_toml(setup_info: SetupInfo) -> None:
         )
     elif formatter == "don't use a formatter":
         formatter_cmds.append("disabled")
+    if formatter in ["black", "ruff"]:
+        try:
+            result = subprocess.run([formatter], capture_output=True, check=False)
+        except FileNotFoundError as e:
+            click.echo(f"⚠️ Formatter not found: {formatter}, please ensure it is installed")
     codeflash_section["formatter-cmds"] = formatter_cmds
     # Add the 'codeflash' section, ensuring 'tool' section exists
     tool_section = pyproject_data.get("tool", tomlkit.table())

codeflash/github/PrComment.py

@@ -20,6 +20,13 @@ class PrComment:
     winning_benchmarking_test_results: TestResults
 
     def to_json(self) -> dict[str, Union[dict[str, dict[str, int]], int, str]]:
+
+        report_table = {
+            test_type.to_name(): result
+            for test_type, result in self.winning_behavioral_test_results.get_test_pass_fail_report_by_type().items()
+            if test_type.to_name()
+        }
+
         return {
             "optimization_explanation": self.optimization_explanation,
             "best_runtime": humanize_runtime(self.best_runtime),
@@ -29,10 +36,7 @@ def to_json(self) -> dict[str, Union[dict[str, dict[str, int]], int, str]]:
             "speedup_x": self.speedup_x,
             "speedup_pct": self.speedup_pct,
             "loop_count": self.winning_benchmarking_test_results.number_of_loops(),
-            "report_table": {
-                test_type.to_name(): result
-                for test_type, result in self.winning_behavioral_test_results.get_test_pass_fail_report_by_type().items()
-            },
+            "report_table": report_table
         }
 
 

codeflash/version.py

@@ -1,3 +1,3 @@
 # These version placeholders will be replaced by poetry-dynamic-versioning during `poetry build`.
-__version__ = "0.9.2"
-__version_tuple__ = (0, 9, 2)
+__version__ = "0.10.0"
+__version_tuple__ = (0, 10, 0)

pyproject.toml

@@ -67,7 +67,7 @@ exclude = [
 
 # Versions here the minimum required versions for the project. These should be as loose as possible.
 [tool.poetry.dependencies]
-python = "^3.9"
+python = ">=3.9"
 unidiff = ">=0.7.4"
 pytest = ">=7.0.0"
 gitpython = ">=3.1.31"