handle edge cases where config file not found or it's empty

Author: Codeflash Bot

codeflash/cli_cmds/cmd_init.py

@@ -32,6 +32,7 @@
 from codeflash.code_utils.github_utils import get_github_secrets_page_url
 from codeflash.code_utils.shell_utils import get_shell_rc_path, save_api_key_to_rc
 from codeflash.either import is_successful
+from codeflash.lsp.helpers import is_LSP_enabled
 from codeflash.telemetry.posthog_cf import ph
 from codeflash.version import __version__ as version
 
@@ -648,41 +649,41 @@ def check_for_toml_or_setup_file() -> str | None:
             apologize_and_exit()
         create_toml = toml_answers["create_toml"]
         if create_toml:
-            ph("cli-create-pyproject-toml")
-            # Define a minimal pyproject.toml content
-            new_pyproject_toml = tomlkit.document()
-            new_pyproject_toml["tool"] = {"codeflash": {}}
-            try:
-                pyproject_toml_path.write_text(tomlkit.dumps(new_pyproject_toml), encoding="utf8")
-
-                # Check if the pyproject.toml file was created
-                if pyproject_toml_path.exists():
-                    success_panel = Panel(
-                        Text(
-                            f"✅ Created a pyproject.toml file at {pyproject_toml_path}\n\n"
-                            "Your project is now ready for Codeflash configuration!",
-                            style="green",
-                            justify="center",
-                        ),
-                        title="🎉 Success!",
-                        border_style="bright_green",
-                    )
-                    console.print(success_panel)
-                    console.print("\n📍 Press any key to continue...")
-                    console.input()
-                ph("cli-created-pyproject-toml")
-            except OSError:
-                click.echo(
-                    "❌ Failed to create pyproject.toml. Please check your disk permissions and available space."
-                )
-                apologize_and_exit()
-        else:
-            click.echo("⏩️ Skipping pyproject.toml creation.")
-            apologize_and_exit()
+            create_empty_pyproject_toml(pyproject_toml_path)
     click.echo()
     return cast("str", project_name)
 
 
+def create_empty_pyproject_toml(pyproject_toml_path: Path) -> None:
+    ph("cli-create-pyproject-toml")
+    lsp_mode = is_LSP_enabled()
+    # Define a minimal pyproject.toml content
+    new_pyproject_toml = tomlkit.document()
+    new_pyproject_toml["tool"] = {"codeflash": {}}
+    try:
+        pyproject_toml_path.write_text(tomlkit.dumps(new_pyproject_toml), encoding="utf8")
+
+        # Check if the pyproject.toml file was created
+        if pyproject_toml_path.exists() and not lsp_mode:
+            success_panel = Panel(
+                Text(
+                    f"✅ Created a pyproject.toml file at {pyproject_toml_path}\n\n"
+                    "Your project is now ready for Codeflash configuration!",
+                    style="green",
+                    justify="center",
+                ),
+                title="🎉 Success!",
+                border_style="bright_green",
+            )
+            console.print(success_panel)
+            console.print("\n📍 Press any key to continue...")
+            console.input()
+        ph("cli-created-pyproject-toml")
+    except OSError:
+        click.echo("❌ Failed to create pyproject.toml. Please check your disk permissions and available space.")
+        apologize_and_exit()
+
+
 def install_github_actions(override_formatter_check: bool = False) -> None:  # noqa: FBT001, FBT002
     try:
         config, _config_file_path = parse_config_file(override_formatter_check=override_formatter_check)

codeflash/code_utils/config_parser.py

@@ -5,6 +5,8 @@
 
 import tomlkit
 
+from codeflash.lsp.helpers import is_LSP_enabled
+
 PYPROJECT_TOML_CACHE = {}
 ALL_CONFIG_FILES = {}  # map path to closest config file
 
@@ -93,15 +95,23 @@ def parse_config_file(
         msg = f"Error while parsing the config file {config_file_path}. Please recheck the file for syntax errors. Error: {e}"
         raise ValueError(msg) from e
 
+    lsp_mode = is_LSP_enabled()
+
     try:
         tool = data["tool"]
         assert isinstance(tool, dict)
-        config = tool["codeflash"]
+        config = tool.get("codeflash", {})
     except tomlkit.exceptions.NonExistentKey as e:
+        if lsp_mode:
+            # don't fail in lsp mode if codeflash config is not found.
+            return {}, config_file_path
         msg = f"Could not find the 'codeflash' block in the config file {config_file_path}. Please run 'codeflash init' to create the config file."
         raise ValueError(msg) from e
     assert isinstance(config, dict)
 
+    if config == {} and lsp_mode:
+        return {}, config_file_path
+
     # default values:
     path_keys = ["module-root", "tests-root", "benchmarks-root"]
     path_list_keys = ["ignore-paths"]
@@ -139,12 +149,13 @@ def parse_config_file(
         else:
             config[key] = []
 
-    assert config["test-framework"] in {"pytest", "unittest"}, (
-        "In pyproject.toml, Codeflash only supports the 'test-framework' as pytest and unittest."
-    )
+    if config.get("test-framework"):
+        assert config["test-framework"] in {"pytest", "unittest"}, (
+            "In pyproject.toml, Codeflash only supports the 'test-framework' as pytest and unittest."
+        )
     # see if this is happening during GitHub actions setup
-    if len(config["formatter-cmds"]) > 0 and not override_formatter_check:
-        assert config["formatter-cmds"][0] != "your-formatter $file", (
+    if config.get("formatter-cmds") and len(config.get("formatter-cmds")) > 0 and not override_formatter_check:
+        assert config.get("formatter-cmds")[0] != "your-formatter $file", (
             "The formatter command is not set correctly in pyproject.toml. Please set the "
             "formatter command in the 'formatter-cmds' key. More info - https://docs.codeflash.ai/configuration"
         )

codeflash/lsp/beta.py

@@ -15,6 +15,7 @@
     CommonSections,
     SetupInfo,
     configure_pyproject_toml,
+    create_empty_pyproject_toml,
     get_suggestions,
     get_valid_subdirs,
     is_valid_pyproject_toml,
@@ -171,13 +172,19 @@ def _find_pyproject_toml(workspace_path: str) -> tuple[Path | None, bool]:
 @server.feature("writeConfig")
 def write_config(_server: CodeflashLanguageServer, params: WriteConfigParams) -> dict[str, any]:
     cfg = params.config
-    cfg_file = Path(params.config_file)
+    cfg_file = Path(params.config_file) if params.config_file else None
+
+    parsed_config = {}
+
+    if cfg_file and not cfg_file.exists():
+        # the client provided a config path but it doesn't exist
+        create_empty_pyproject_toml(cfg_file)
+    elif cfg_file and cfg_file.exists():
+        try:
+            parsed_config, _ = parse_config_file(cfg_file)
+        except Exception as e:
+            return {"status": "error", "message": f"Failed to parse configuration: {e}"}
 
-    try:
-        parsed_config, _ = parse_config_file(cfg_file)
-    except Exception as e:
-        return {"status": "error", "message": f"Failed to parse configuration: {e}"}
-    _server.show_message_log(f"{parsed_config}", "Info")
     setup_info = SetupInfo(
         module_root=getattr(cfg, "module_root", ""),
         tests_root=getattr(cfg, "tests_root", ""),