small fixes

Author: mohammedahmed18

codeflash/code_utils/git_utils.py

@@ -258,7 +258,7 @@ def remove_worktree(worktree_dir: Path) -> None:
         logger.exception(f"Failed to remove worktree: {worktree_dir}")
 
 
-def create_diff_from_worktree(worktree_dir: Path, files: list[str], fto_name: str) -> Path:
+def create_diff_patch_from_worktree(worktree_dir: Path, files: list[str], fto_name: str) -> Path:
     repository = git.Repo(worktree_dir, search_parent_directories=True)
     uni_diff_text = repository.git.diff(None, "HEAD", *files, ignore_blank_lines=True, ignore_space_at_eol=True)
 

codeflash/lsp/beta.py

@@ -9,7 +9,7 @@
 from pygls import uris
 
 from codeflash.api.cfapi import get_codeflash_api_key, get_user_id
-from codeflash.code_utils.git_utils import create_diff_from_worktree
+from codeflash.code_utils.git_utils import create_diff_patch_from_worktree
 from codeflash.code_utils.shell_utils import save_api_key_to_rc
 from codeflash.either import is_successful
 from codeflash.lsp.server import CodeflashLanguageServer, CodeflashLanguageServerProtocol
@@ -322,7 +322,7 @@ def perform_function_optimization(  # noqa: PLR0911
 
         # generate a patch for the optimization
         relative_file_paths = [code_string.file_path for code_string in code_context.read_writable_code.code_strings]
-        patch_file = create_diff_from_worktree(
+        patch_file = create_diff_patch_from_worktree(
             server.optimizer.current_worktree,
             relative_file_paths,
             server.optimizer.current_function_optimizer.function_to_optimize.qualified_name,

codeflash/lsp/server.py

@@ -56,7 +56,6 @@ def prepare_optimizer_arguments(self, config_file: Path) -> None:
         args.config_file = config_file
         args.no_pr = True  # LSP server should not create PRs
         args.worktree = True
-        args.verbose = True
         args = process_pyproject_config(args)
         self.args = args
         # avoid initializing the optimizer during initialization, because it can cause an error if the api key is invalid
@@ -78,7 +77,7 @@ def show_message_log(self, message: str, message_type: str) -> None:
             "Debug": MessageType.Debug,
         }
 
-        lsp_message_type = type_mapping.get(message_type, MessageType.Debug)
+        lsp_message_type = type_mapping.get(message_type, MessageType.Info)
 
         # Send log message to client (appears in output channel)
         log_params = LogMessageParams(type=lsp_message_type, message=message)

codeflash/main.py

@@ -8,7 +8,7 @@
 
 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.console import console, paneled_text
+from codeflash.cli_cmds.console import paneled_text
 from codeflash.code_utils.checkpoint import ask_should_use_checkpoint_get_functions
 from codeflash.code_utils.config_parser import parse_config_file
 from codeflash.telemetry import posthog_cf
@@ -17,7 +17,6 @@
 
 def main() -> None:
     """Entry point for the codeflash command-line interface."""
-    console.quiet = False
     paneled_text(
         CODEFLASH_LOGO, panel_args={"title": "https://codeflash.ai", "expand": False}, text_args={"style": "bold gold3"}
     )

codeflash/optimization/function_optimizer.py

@@ -1259,7 +1259,7 @@ def process_review(
                 trace_id=self.function_trace_id, is_optimization_found=best_optimization is not None
             )
 
-        # If worktree mode, do not revert code and helpers, we create snapshot commit after each function optimization
+        # If worktree mode, do not revert code and helpers,, otherwise we would have an empty diff when writing the patch in the lsp
         if self.args.worktree:
             return
 

codeflash/optimization/optimizer.py

@@ -18,7 +18,7 @@
 from codeflash.code_utils.git_utils import (
     check_running_in_git_repo,
     create_detached_worktree,
-    create_diff_from_worktree,
+    create_diff_patch_from_worktree,
     create_worktree_snapshot_commit,
     remove_worktree,
 )
@@ -346,7 +346,7 @@ def run(self) -> None:
                                 relative_file_paths = [
                                     code_string.file_path for code_string in read_writable_code.code_strings
                                 ]
-                                patch_path = create_diff_from_worktree(
+                                patch_path = create_diff_patch_from_worktree(
                                     self.current_worktree,
                                     relative_file_paths,
                                     self.current_function_optimizer.function_to_optimize.qualified_name,