fix: import bug (#493)

# Motivation

<!-- Why is this change necessary? -->

# Content

<!-- Please include a summary of the change -->

# Testing

<!-- How was the change tested? -->

# Please check the following before marking your PR as ready for review

- [ ] I have added tests for my changes
- [ ] I have updated the documentation or added new documentation as
needed

Author: bagel897

src/codegen/cli/commands/lsp/lsp.py

@@ -9,8 +9,8 @@
 def lsp_command():
     try:
         from codegen.extensions.lsp.lsp import server
-    except ImportError:
-        logger.exception("LSP is not installed. Please install it with `uv tool install codegen[lsp]`")
+    except (ImportError, ModuleNotFoundError):
+        logger.exception("LSP is not installed. Please install it with `uv tool install codegen[lsp] --prerelease=allow`")
         return
     logging.basicConfig(level=logging.INFO)
     server.start_io()