better naming

better names

Author: KRRT7

codeflash/api/aiservice.py

@@ -4,6 +4,7 @@
 import os
 import platform
 import time
+from importlib.metadata import packages_distributions
 from typing import TYPE_CHECKING, Any
 
 import requests
@@ -182,6 +183,8 @@ def optimize_python_code_line_profiler(  # noqa: D417
             "python_version": platform.python_version(),
             "experiment_metadata": experiment_metadata,
             "codeflash_version": codeflash_version,
+            "lsp_mode": console.quiet,  # if quiet mode is enabled, then we're in LSP mode
+            "installed_packages": list(packages_distributions().keys()),
         }
 
         logger.info("Generating optimized candidates…")

codeflash/lsp/beta.py

@@ -19,7 +19,7 @@ class OptimizableFunctionsParams:
 
 
 @dataclass
-class OptimizeFunctionParams:
+class FunctionOptimizationParams:
     textDocument: types.TextDocumentIdentifier  # noqa: N815
     functionName: str  # noqa: N815
 
@@ -41,8 +41,10 @@ def get_optimizable_functions(
     return path_to_qualified_names
 
 
-@server.feature("optimizeFunction")
-def optimize_function(server: CodeflashLanguageServer, params: OptimizeFunctionParams) -> dict[str, str]:
+@server.feature("initializeFunctionOptimization")
+def initialize_function_optimization(
+    server: CodeflashLanguageServer, params: FunctionOptimizationParams
+) -> dict[str, str]:
     file_path = Path(uris.to_fs_path(params.textDocument.uri))
     server.optimizer.args.function = params.functionName
     server.optimizer.args.file = file_path
@@ -54,8 +56,8 @@ def optimize_function(server: CodeflashLanguageServer, params: OptimizeFunctionP
     return {"functionName": params.functionName, "status": "success", "info": fto.server_info}
 
 
-@server.feature("second_step_in_optimize_function")
-def second_step_in_optimize_function(server: CodeflashLanguageServer, params: OptimizeFunctionParams) -> dict[str, str]:
+@server.feature("discoverFunctionTests")
+def discover_function_tests(server: CodeflashLanguageServer, params: FunctionOptimizationParams) -> dict[str, str]:
     current_function = server.optimizer.current_function_being_optimized
 
     optimizable_funcs = {current_function.file_path: [current_function]}
@@ -65,8 +67,10 @@ def second_step_in_optimize_function(server: CodeflashLanguageServer, params: Op
     return {"functionName": params.functionName, "status": "success", "generated_tests": str(num_discovered_tests)}
 
 
-@server.feature("third_step_in_optimize_function")
-def third_step_in_optimize_function(server: CodeflashLanguageServer, params: OptimizeFunctionParams) -> dict[str, str]:
+@server.feature("performFunctionOptimization")
+def perform_function_optimization(
+    server: CodeflashLanguageServer, params: FunctionOptimizationParams
+) -> dict[str, str]:
     current_function = server.optimizer.current_function_being_optimized
 
     module_prep_result = server.optimizer.prepare_module_for_optimization(current_function.file_path)

codeflash/verification/concolic_testing.py

@@ -25,6 +25,10 @@
 def generate_concolic_tests(
     test_cfg: TestConfig, args: Namespace, function_to_optimize: FunctionToOptimize, function_to_optimize_ast: ast.AST
 ) -> tuple[dict[str, set[FunctionCalledInTest]], str]:
+    # Skip concolic test generation when running under LSP
+    if console.quiet:
+        return {}, ""
+
     start_time = time.perf_counter()
     function_to_concolic_tests = {}
     concolic_test_suite_code = ""