fix: Make lsp an optional dependency (#492)

Author: bagel897

pyproject.toml

@@ -70,9 +70,6 @@ dependencies = [
   "numpy>=2.2.2",
   "mcp[cli]",
   "neo4j",
-  "pygls>=2.0.0a2",
-  "lsprotocol==2024.0.0b1",
-  "attrs>=25.1.0",
 ]
 
 license = { text = "Apache-2.0" }
@@ -120,6 +117,7 @@ types = [
   "types-requests>=2.32.0.20241016",
   "types-toml>=0.10.8.20240310",
 ]
+lsp = ["pygls>=2.0.0a2", "lsprotocol==2024.0.0b1", "attrs>=25.1.0"]
 [tool.uv]
 cache-keys = [{ git = { commit = true, tags = true } }]
 dev-dependencies = [
@@ -162,7 +160,6 @@ dev-dependencies = [
   "pytest-lsp>=1.0.0b1",
 ]
 
-
 [tool.uv.workspace]
 exclude = ["codegen-examples"]
 

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

@@ -2,10 +2,15 @@
 
 import click
 
-from codegen.extensions.lsp.lsp import server
+logger = logging.getLogger(__name__)
 
 
 @click.command(name="lsp")
 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]`")
+        return
     logging.basicConfig(level=logging.INFO)
     server.start_io()