feat: Add codegen lsp (#491)

Author: bagel897

pyproject.toml

@@ -70,6 +70,9 @@ 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" }
@@ -117,7 +120,6 @@ 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 = [

src/codegen/cli/cli.py

@@ -9,6 +9,7 @@
 from codegen.cli.commands.list.main import list_command
 from codegen.cli.commands.login.main import login_command
 from codegen.cli.commands.logout.main import logout_command
+from codegen.cli.commands.lsp.lsp import lsp_command
 from codegen.cli.commands.notebook.main import notebook_command
 from codegen.cli.commands.profile.main import profile_command
 from codegen.cli.commands.reset.main import reset_command
@@ -43,7 +44,7 @@ def main():
 main.add_command(reset_command)
 main.add_command(update_command)
 main.add_command(config_command)
-
+main.add_command(lsp_command)
 
 if __name__ == "__main__":
     main()

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

@@ -0,0 +1,11 @@
+import logging
+
+import click
+
+from codegen.extensions.lsp.lsp import server
+
+
+@click.command(name="lsp")
+def lsp_command():
+    logging.basicConfig(level=logging.INFO)
+    server.start_io()