diff --git a/codeflash/lsp/__init__.py b/codeflash/lsp/__init__.py index e69de29bb..9d75096ef 100644 --- a/codeflash/lsp/__init__.py +++ b/codeflash/lsp/__init__.py @@ -0,0 +1,4 @@ +# Silence the console module to prevent stdout pollution +from codeflash.cli_cmds.console import console + +console.quiet = True diff --git a/codeflash/lsp/server_entry.py b/codeflash/lsp/server_entry.py index 692f7562e..dfb8dd5a3 100644 --- a/codeflash/lsp/server_entry.py +++ b/codeflash/lsp/server_entry.py @@ -39,11 +39,5 @@ def setup_logging() -> logging.Logger: 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()