feat: add support for ty (Astral's Python type checker)

Author: observerw

container/registry.toml

@@ -14,6 +14,10 @@ repo = "rust-lang/rust-analyzer"
 type = "pypi"
 package = "pyrefly"
 
+[ty]
+type = "pypi"
+package = "ty"
+
 [deno]
 type = "github"
 repo = "denoland/deno"

container/ty/ContainerFile

@@ -0,0 +1,14 @@
+# Multi-stage build for ty
+ARG VERSION=latest
+FROM python:3.12-slim AS builder
+ARG VERSION
+RUN pip install --no-cache-dir ty==${VERSION}
+
+FROM python:3.12-slim
+ARG VERSION
+LABEL org.opencontainers.image.version="${VERSION}"
+COPY --from=builder /usr/local/lib/python3.12/site-packages /usr/local/lib/python3.12/site-packages
+COPY --from=builder /usr/local/bin/ty /usr/local/bin/ty
+
+WORKDIR /workspace
+ENTRYPOINT ["ty", "server"]

src/lsp_client/clients/__init__.py

@@ -4,17 +4,20 @@
 from .pyrefly import PyreflyClient
 from .pyright import PyrightClient
 from .rust_analyzer import RustAnalyzerClient
+from .ty import TyClient
 from .typescript import TypescriptClient
 
 PythonClient = PyrightClient
 RustClient = RustAnalyzerClient
 TypeScriptClient = TypescriptClient
 DenoLSPClient = DenoClient
+TyLSPClient = TyClient
 
 __all__ = [
     "DenoLSPClient",
     "PyreflyClient",
     "PythonClient",
     "RustClient",
+    "TyLSPClient",
     "TypeScriptClient",
 ]

src/lsp_client/clients/ty.py

@@ -0,0 +1,101 @@
+from __future__ import annotations
+
+import shutil
+import sys
+from functools import partial
+from subprocess import CalledProcessError
+from typing import Any, Literal, override
+
+import anyio
+from attrs import define
+from loguru import logger
+
+from lsp_client.capability.notification import (
+    WithNotifyDidChangeConfiguration,
+)
+from lsp_client.capability.request import (
+    WithRequestDefinition,
+    WithRequestHover,
+    WithRequestReferences,
+)
+from lsp_client.capability.server_notification import (
+    WithReceiveLogMessage,
+    WithReceivePublishDiagnostics,
+)
+from lsp_client.capability.server_request import (
+    WithRespondConfigurationRequest,
+    WithRespondWorkspaceFoldersRequest,
+)
+from lsp_client.client.abc import LSPClient
+from lsp_client.server.abc import LSPServer
+from lsp_client.server.container import ContainerServer
+from lsp_client.server.local import LocalServer
+from lsp_client.utils.types import lsp_type
+
+TyContainerServer = partial(
+    ContainerServer, image="ghcr.io/observerw/lsp-client/ty:latest"
+)
+
+
+@define
+class TyClient(
+    LSPClient,
+    WithNotifyDidChangeConfiguration,
+    WithRequestDefinition,
+    WithRequestHover,
+    WithRequestReferences,
+    WithReceiveLogMessage,
+    WithReceivePublishDiagnostics,
+    WithRespondConfigurationRequest,
+    WithRespondWorkspaceFoldersRequest,
+):
+    """
+    - Language: Python
+    - Homepage: https://docs.astral.sh/ty/
+    - Github: https://github.com/astral-sh/ty
+    """
+
+    log_level: Literal["trace", "debug", "info", "warn", "error"] = "info"
+    diagnostic_mode: Literal["openFilesOnly", "workspace"] = "openFilesOnly"
+    inlay_hints_variable_types: bool = True
+    inlay_hints_call_argument_names: bool = True
+
+    @override
+    def get_language_id(self) -> lsp_type.LanguageKind:
+        return lsp_type.LanguageKind.Python
+
+    @override
+    def create_default_server(self) -> LSPServer:
+        return LocalServer(command=["ty", "server"])
+
+    @override
+    def create_initialization_options(self) -> dict[str, Any]:
+        return {
+            "logLevel": self.log_level,
+            "diagnosticMode": self.diagnostic_mode,
+            "inlayHints": {
+                "variableTypes": self.inlay_hints_variable_types,
+                "callArgumentNames": self.inlay_hints_call_argument_names,
+            },
+        }
+
+    @override
+    def check_server_compatibility(self, info: lsp_type.ServerInfo | None) -> None:
+        return
+
+    @override
+    async def ensure_installed(self) -> None:
+        if shutil.which("ty"):
+            return
+
+        logger.warning("ty not found, attempting to install via pip...")
+
+        try:
+            await anyio.run_process([sys.executable, "-m", "pip", "install", "ty"])
+            logger.info("Successfully installed ty via pip")
+            return
+        except CalledProcessError as e:
+            raise RuntimeError(
+                "Could not install ty. Please install it manually with 'pip install ty' or 'uv tool install ty'. "
+                "See https://docs.astral.sh/ty/installation/ for more information."
+            ) from e