Create dedicated server entry point to prevent crashes

codeflash/lsp/beta.py

@@ -292,10 +292,3 @@ def perform_function_optimization(  # noqa: PLR0911
         "extra": f"Speedup: {speedup:.2f}x faster",
         "optimization": optimized_source,
     }
-
-
-if __name__ == "__main__":
-    from codeflash.cli_cmds.console import console
-
-    console.quiet = True
-    server.start_io()

codeflash/lsp/server.py

@@ -3,7 +3,7 @@
 from pathlib import Path
 from typing import TYPE_CHECKING, Any
 
-from lsprotocol.types import INITIALIZE
+from lsprotocol.types import INITIALIZE, MessageType, LogMessageParams
 from pygls import uris
 from pygls.protocol import LanguageServerProtocol, lsp_method
 from pygls.server import LanguageServer
@@ -55,3 +55,24 @@ def initialize_optimizer(self, config_file: Path) -> None:
         args.no_pr = True  # LSP server should not create PRs
         args = process_pyproject_config(args)
         self.optimizer = Optimizer(args)
+
+    def show_message_log(self, message: str, message_type: str) -> None:
+        """Send a log message to the client's output channel.
+
+        Args:
+            message: The message to log
+            message_type: String type - "Info", "Warning", "Error", or "Log"
+        """
+        # Convert string message type to LSP MessageType enum
+        type_mapping = {
+            "Info": MessageType.Info,
+            "Warning": MessageType.Warning, 
+            "Error": MessageType.Error,
+            "Log": MessageType.Log
+        }
+
+        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)
+        self.lsp.notify("window/logMessage", log_params)

codeflash/lsp/server_entry.py

@@ -0,0 +1,48 @@
+"""This script is the dedicated entry point for the Codeflash Language Server.
+It initializes the server and redirects its logs to stderr so that the
+VS Code client can display them in the output channel.
+
+This script is run by the VS Code extension and is not intended to be
+executed directly by users.
+"""
+
+import logging
+import sys
+
+from codeflash.lsp.beta import server
+
+
+# Configure logging to stderr for VS Code output channel
+def setup_logging():
+    # Clear any existing handlers to prevent conflicts
+    root_logger = logging.getLogger()
+    root_logger.handlers.clear()
+
+    # Set up stderr handler for VS Code output channel with [LSP-Server] prefix
+    handler = logging.StreamHandler(sys.stderr)
+    handler.setFormatter(logging.Formatter("[LSP-Server] %(asctime)s [%(levelname)s]: %(message)s"))
+
+    # Configure root logger
+    root_logger.addHandler(handler)
+    root_logger.setLevel(logging.INFO)
+
+    # Also configure the pygls logger specifically
+    pygls_logger = logging.getLogger("pygls")
+    pygls_logger.setLevel(logging.INFO)
+
+    return root_logger
+
+
+if __name__ == "__main__":
+    # Set up logging
+    log = setup_logging()
+    log.info("Starting Codeflash Language Server...")
+
+    # Silence the console module to prevent stdout pollution
+    from codeflash.cli_cmds.console import console
+
+    console.quiet = True
+    # console.enable()
+
+    # Start the language server
+    server.start_io()