fix

Author: observerw

src/lsp_client/client/abc.py

@@ -122,6 +122,7 @@ def get_config_map(self) -> ConfigurationMap:
     def get_server(self) -> Server:
         return self._server
 
+    @classmethod
     @classmethod
     @abstractmethod
     def create_default_servers(cls) -> DefaultServers:

src/lsp_client/clients/deno/client.py

@@ -148,8 +148,9 @@ def get_language_config(cls) -> LanguageConfig:
             project_files=["deno.json", "deno.jsonc"],
         )
 
+    @classmethod
     @override
-    def create_default_servers(self) -> DefaultServers:
+    def create_default_servers(cls) -> DefaultServers:
         return DefaultServers(
             local=DenoLocalServer(),
             container=DenoContainerServer(),

src/lsp_client/clients/gopls.py

@@ -107,8 +107,9 @@ class GoplsClient(
     - VSCode Extension: https://marketplace.visualstudio.com/items?itemName=golang.go
     """
 
+    @classmethod
     @override
-    def create_default_servers(self) -> DefaultServers:
+    def create_default_servers(cls) -> DefaultServers:
         return DefaultServers(
             local=GoplsLocalServer(),
             container=GoplsContainerServer(),

src/lsp_client/clients/pyrefly.py

@@ -114,8 +114,9 @@ class PyreflyClient(
     - VSCode Extension: https://github.com/facebook/pyrefly/tree/main/lsp
     """
 
+    @classmethod
     @override
-    def create_default_servers(self) -> DefaultServers:
+    def create_default_servers(cls) -> DefaultServers:
         return DefaultServers(
             local=PyreflyLocalServer(),
             container=PyreflyContainerServer(),

src/lsp_client/clients/pyright.py

@@ -107,8 +107,9 @@ class PyrightClient(
     - VSCode Extension: https://github.com/microsoft/pyright/tree/main/packages/vscode-pyright
     """
 
+    @classmethod
     @override
-    def create_default_servers(self) -> DefaultServers:
+    def create_default_servers(cls) -> DefaultServers:
         return DefaultServers(
             local=PyrightLocalServer(),
             container=PyrightContainerServer(),

src/lsp_client/clients/rust_analyzer.py

@@ -112,8 +112,9 @@ class RustAnalyzerClient(
     - VSCode Extension: https://marketplace.visualstudio.com/items?itemName=rust-lang.rust-analyzer
     """
 
+    @classmethod
     @override
-    def create_default_servers(self) -> DefaultServers:
+    def create_default_servers(cls) -> DefaultServers:
         return DefaultServers(
             local=RustAnalyzerLocalServer(),
             container=RustAnalyzerContainerServer(),

src/lsp_client/clients/ty.py

@@ -110,8 +110,9 @@ class TyClient(
     - VSCode Extension: https://docs.astral.sh/ty/editors/vscode/
     """
 
+    @classmethod
     @override
-    def create_default_servers(self) -> DefaultServers:
+    def create_default_servers(cls) -> DefaultServers:
         return DefaultServers(
             local=TyLocalServer(),
             container=TyContainerServer(),

src/lsp_client/clients/typescript.py

@@ -113,8 +113,9 @@ class TypescriptClient(
     - VSCode Extension: Built-in TypeScript support in VS Code
     """
 
+    @classmethod
     @override
-    def create_default_servers(self) -> DefaultServers:
+    def create_default_servers(cls) -> DefaultServers:
         return DefaultServers(
             local=TypescriptLocalServer(),
             container=TypescriptContainerServer(),

src/lsp_client/server/abc.py

@@ -93,8 +93,9 @@ async def _dispatch(self, sender: Sender[ServerRequest]) -> None:
                 tg.soonify(self._handle_package)(sender, package)
 
     async def request(self, request: RawRequest) -> RawResponsePackage:
+        rx = self._resp_table.reserve(request["id"])
         await self.send(request)
-        return await self._resp_table.receive(request["id"])
+        return await rx.receive()
 
     async def notify(self, notification: RawNotification) -> None:
         await self.send(notification)

src/lsp_client/server/container.py

@@ -211,6 +211,7 @@ def format_args(self, workspace: Workspace) -> list[str]:
 
         return args
 
+    @override
     @asynccontextmanager
     async def run(
         self, workspace: Workspace, sender: Sender[ServerRequest]
@@ -220,5 +221,8 @@ async def run(
 
         self._local = LocalServer(program=self.backend, args=args)
 
-        async with self._local.run(workspace, sender=sender):
+        async with (
+            self._local.run_process(workspace),
+            super().run(workspace, sender=sender),
+        ):
             yield self