cancel tasks with status initializing

Author: mohammedahmed18

codeflash/lsp/beta.py

@@ -344,6 +344,13 @@ def execution_context(**kwargs: str) -> None:
         server.execution_context_vars.reset(token)
 
 
+@server.feature("cleanupCurrentOptimizerSession")
+def cleanup_optimizer(_params: any) -> dict[str, str]:
+    if not server.cleanup_the_optimizer():
+        return {"status": "error", "message": "Failed to cleanup optimizer"}
+    return {"status": "success"}
+
+
 @server.feature("initializeFunctionOptimization")
 def initialize_function_optimization(params: FunctionOptimizationInitParams) -> dict[str, str]:
     with execution_context(task_id=params.task_id):

codeflash/lsp/server.py

@@ -63,10 +63,10 @@ def show_message_log(self, message: str, message_type: str) -> None:
         log_params = LogMessageParams(type=lsp_message_type, message=message)
         self.protocol.notify("window/logMessage", log_params)
 
-    def cleanup_the_optimizer(self) -> None:
+    def cleanup_the_optimizer(self) -> bool:
         self.current_optimization_init_result = None
         if not self.optimizer:
-            return
+            return False
         try:
             self.optimizer.cleanup_temporary_paths()
             # restore args and test cfg
@@ -77,6 +77,8 @@ def cleanup_the_optimizer(self) -> None:
             self.optimizer.current_function_optimizer = None
         except Exception:
             self.show_message_log("Failed to cleanup optimizer", "Error")
+            return False
+        return True
 
     def shutdown(self) -> None:
         """Gracefully shutdown the server."""

codeflash/optimization/function_optimizer.py

@@ -1586,7 +1586,7 @@ def establish_original_code_baseline(
             )
 
         if test_framework == "pytest":
-            with progress_bar("Running line profiling to identify performance bottlenecks..."):
+            with progress_bar("Running line profiler to identify performance bottlenecks..."):
                 line_profile_results = self.line_profiler_step(
                     code_context=code_context, original_helper_code=original_helper_code, candidate_index=0
                 )