formatting

Author: Mazyod

lsp_types/process.py

@@ -4,8 +4,8 @@
 import json
 import logging
 import os
-from pathlib import Path
 import typing as t
+from pathlib import Path
 
 from . import requests, types
 

lsp_types/pyrefly/backend.py

@@ -31,7 +31,9 @@ def write_config(self, base_path: Path, options: PyreflyConfig) -> None:
 
         config_path.write_text(toml_content)
 
-    def create_process_launch_info(self, base_path: Path, options: PyreflyConfig) -> ProcessLaunchInfo:
+    def create_process_launch_info(
+        self, base_path: Path, options: PyreflyConfig
+    ) -> ProcessLaunchInfo:
         """Create process launch info for Pyrefly LSP server"""
         # Build command args for Pyrefly LSP server
         cmd_args = ["pyrefly", "lsp"]

lsp_types/pyrefly/config_schema.py

@@ -13,24 +13,24 @@
 class Model(TypedDict):
     """
     Pyrefly Configuration Schema
-    
+
     Based on available CLI options and environment variables.
     Note: Pyrefly's configuration format is still evolving.
     """
-    
+
     # Core options
     verbose: NotRequired[bool]
     threads: NotRequired[int]  # 0 = auto, 1 = sequential, higher = parallel
     color: NotRequired[Literal["auto", "always", "never"]]
-    
+
     # LSP server options
     indexing_mode: NotRequired[IndexingMode]  # Indexing strategy for LSP server
-    
+
     # File inclusion/exclusion (basic patterns)
     include: NotRequired[list[str]]
     exclude: NotRequired[list[str]]
-    
+
     # Environment variables that can be configured
     pyrefly_threads: NotRequired[int]
-    pyrefly_color: NotRequired[str] 
-    pyrefly_verbose: NotRequired[bool]
+    pyrefly_color: NotRequired[str]
+    pyrefly_verbose: NotRequired[bool]

lsp_types/pyright/backend.py

@@ -19,7 +19,9 @@ def write_config(self, base_path: Path, options: PyrightConfig) -> None:
         config_path = base_path / "pyrightconfig.json"
         config_path.write_text(json.dumps(options, indent=2))
 
-    def create_process_launch_info(self, base_path: Path, options: PyrightConfig) -> ProcessLaunchInfo:
+    def create_process_launch_info(
+        self, base_path: Path, options: PyrightConfig
+    ) -> ProcessLaunchInfo:
         """Create process launch info for Pyright LSP server"""
         # NOTE: requires node and basedpyright to be installed and accessible
         return ProcessLaunchInfo(cmd=["pyright-langserver", "--stdio"], cwd=base_path)

lsp_types/session.py

@@ -16,15 +16,19 @@ def write_config(self, base_path: Path, options: t.Mapping) -> None:
         """Write backend-specific configuration file"""
         ...
 
-    def create_process_launch_info(self, base_path: Path, options: t.Mapping) -> ProcessLaunchInfo:
+    def create_process_launch_info(
+        self, base_path: Path, options: t.Mapping
+    ) -> ProcessLaunchInfo:
         """Create process launch info for the LSP server"""
         ...
 
     def get_lsp_capabilities(self) -> types.ClientCapabilities:
         """Get LSP client capabilities"""
         ...
 
-    def get_workspace_settings(self, options: t.Mapping) -> types.DidChangeConfigurationParams:
+    def get_workspace_settings(
+        self, options: t.Mapping
+    ) -> types.DidChangeConfigurationParams:
         """Get workspace settings for didChangeConfiguration"""
         ...
 
@@ -162,9 +166,7 @@ async def get_diagnostics(self) -> list[types.Diagnostic]:
         # For 'unchanged' nothing is appended ⇒ return cached view if desired
         return diagnostics
 
-    async def get_hover_info(
-        self, position: types.Position
-    ) -> types.Hover | None:
+    async def get_hover_info(self, position: types.Position) -> types.Hover | None:
         """Get hover information at the given position"""
         return await self._process.send.hover(
             {"textDocument": {"uri": self._document_uri}, "position": position}

tests/test_pool.py

@@ -11,8 +11,8 @@
 
 import lsp_types
 from lsp_types.pool import LSPProcessPool
-from lsp_types.pyright.backend import PyrightBackend
 from lsp_types.pyrefly.backend import PyreflyBackend
+from lsp_types.pyright.backend import PyrightBackend
 
 
 @pytest.fixture(params=[PyrightBackend, PyreflyBackend])
@@ -63,7 +63,9 @@ async def test_session_pool_acquire_and_recycle(self, session_pool, lsp_backend)
         assert session_pool.available_count == 1
         assert session_pool.current_size == 1
 
-    async def test_session_recycling_with_different_code(self, session_pool, lsp_backend):
+    async def test_session_recycling_with_different_code(
+        self, session_pool, lsp_backend
+    ):
         """Test that recycled sessions work correctly with different code"""
         # First session with initial code
         session1 = await lsp_types.Session.create(
@@ -96,17 +98,21 @@ async def test_session_recycling_with_different_code(self, session_pool, lsp_bac
 
         await session2.shutdown()
 
-    async def test_session_recycling_with_different_options(self, session_pool, lsp_backend, backend_name):
+    async def test_session_recycling_with_different_options(
+        self, session_pool, lsp_backend, backend_name
+    ):
         """Test recycling sessions with different options"""
 
         if backend_name == "pyright":
             from lsp_types.pyright.config_schema import Model as ConfigType
+
             options1: ConfigType = {"strict": ["reportUndefinedVariable"]}
             options2: ConfigType = {"strict": ["reportGeneralTypeIssues"]}
             code1 = "undefined_var = 1"
             code2 = "x: int = 'string'"  # Type error
         else:  # pyrefly
             from lsp_types.pyrefly.config_schema import Model as ConfigType
+
             options1: ConfigType = {"verbose": True, "threads": 1}
             options2: ConfigType = {"verbose": False, "threads": 2}
             code1 = "test_var = 1"
@@ -447,7 +453,9 @@ async def test_idle_cleanup_timing_precision(self, lsp_backend):
 class TestLSPProcessPoolBenchmarks:
     """Benchmark tests comparing pooled vs non-pooled session performance"""
 
-    async def test_benchmark_session_creation_comparison(self, lsp_backend, backend_name):
+    async def test_benchmark_session_creation_comparison(
+        self, lsp_backend, backend_name
+    ):
         """Compare session creation times with and without pooling"""
         pool = LSPProcessPool(max_size=3)
 
@@ -551,7 +559,9 @@ async def test_benchmark_session_reuse_performance(self, lsp_backend, backend_na
         finally:
             await pool.cleanup()
 
-    async def test_benchmark_concurrent_session_creation(self, lsp_backend, backend_name):
+    async def test_benchmark_concurrent_session_creation(
+        self, lsp_backend, backend_name
+    ):
         """Compare concurrent session creation with and without pooling"""
         pool = LSPProcessPool(max_size=5)
 
@@ -580,7 +590,8 @@ async def create_pooled_session(session_id: int):
 
             async def create_fresh_session(session_id: int):
                 session = await lsp_types.Session.create(
-                    lsp_backend, initial_code=f"fresh_concurrent_{session_id} = {session_id}"
+                    lsp_backend,
+                    initial_code=f"fresh_concurrent_{session_id} = {session_id}",
                 )
                 hover_info = await session.get_hover_info(
                     lsp_types.Position(line=0, character=0)
@@ -616,7 +627,7 @@ async def test_pyrefly_config_options_benchmark():
     # Test different threading configurations
     configs: list[PyreflyConfig] = [
         {"threads": 0, "verbose": False},  # Auto
-        {"threads": 1, "verbose": False},  # Sequential 
+        {"threads": 1, "verbose": False},  # Sequential
         {"threads": 2, "verbose": False},  # Parallel
         {"threads": 4, "verbose": False},  # More parallel
     ]
@@ -626,31 +637,31 @@ async def test_pyrefly_config_options_benchmark():
     try:
         for config in configs:
             config_times = []
-            
+
             for i in range(3):
                 start_time = time.perf_counter()
-                
+
                 session = await lsp_types.Session.create(
                     backend,
                     initial_code=f"def test_{i}(x: int) -> int: return x * 2\nresult = test_{i}(5)",
                     options=config,  # type: ignore
-                    pool=pool
+                    pool=pool,
                 )
-                
+
                 # Do some work to test performance
                 hover_info = await session.get_hover_info(
                     lsp_types.Position(line=0, character=4)
                 )
                 assert hover_info is not None
-                
+
                 diagnostics = await session.get_diagnostics()
                 await session.shutdown()
-                
+
                 end_time = time.perf_counter()
                 config_times.append(end_time - start_time)
-            
+
             avg_time = sum(config_times) / len(config_times)
             print(f"\nPyrefly Config {config}: Average time {avg_time:.3f}s")
 
     finally:
-        await pool.cleanup()
+        await pool.cleanup()