Add option to enable/disable telemetry in codeflash init (CF-623)

codeflash/cli_cmds/cmd_init.py

@@ -676,13 +676,16 @@ def configure_pyproject_toml(setup_info: SetupInfo) -> None:
         )
         apologize_and_exit()
 
+    enable_telemetry = ask_for_telemetry()
+
     codeflash_section = tomlkit.table()
     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
     codeflash_section["benchmarks-root"] = setup_info.benchmarks_root if setup_info.benchmarks_root else ""
     codeflash_section["ignore-paths"] = setup_info.ignore_paths
+    codeflash_section["disable-telemetry"] = not enable_telemetry
     if setup_info.git_remote not in ["", "origin"]:
         codeflash_section["git-remote"] = setup_info.git_remote
     formatter = setup_info.formatter
@@ -934,3 +937,17 @@ def run_end_to_end_test(args: Namespace, bubble_sort_path: str, bubble_sort_test
             console.rule()
             Path(path).unlink(missing_ok=True)
             logger.info(f"🗑️  Deleted {path}")
+
+
+def ask_for_telemetry() -> bool:
+    """Prompt the user to enable or disable telemetry."""
+    from rich.prompt import Confirm
+
+    enable_telemetry = Confirm.ask(
+        "⚡️ Would you like to enable telemetry to help us improve the Codeflash experience?",
+        default=True,
+        show_default=True,
+    )
+
+    ph("cli-telemetry-prompt", {"telemetry_enabled": enable_telemetry})
+    return enable_telemetry

codeflash/main.py

@@ -5,7 +5,7 @@
 from pathlib import Path
 
 from codeflash.cli_cmds.cli import parse_args, process_pyproject_config
-from codeflash.cli_cmds.cmd_init import CODEFLASH_LOGO, ask_run_end_to_end_test
+from codeflash.cli_cmds.cmd_init import CODEFLASH_LOGO, ask_run_end_to_end_test, ask_for_telemetry
 from codeflash.cli_cmds.console import paneled_text
 from codeflash.code_utils.config_parser import parse_config_file
 from codeflash.optimization import optimizer

codeflash/version.py

@@ -1,3 +1,3 @@
 # These version placeholders will be replaced by poetry-dynamic-versioning during `poetry build`.
-__version__ = "0.12.2"
-__version_tuple__ = (0, 12, 2)
+__version__ = "0.12.2.post2.dev0+b6bbeecc"
+__version_tuple__ = (0, 12, 2, "post2", "dev0", "b6bbeecc")