refactor: improve package exports and API consistency

Author: observerw

src/lsp_client/__init__.py

@@ -13,11 +13,10 @@
 Example:
     ```python
     import anyio
-    from lsp_client.clients.pyrefly import PyreflyClient, PyreflyLocalServer
-    from lsp_client import Position
+    from lsp_client import PyreflyClient, Position
 
     async def main():
-        async with PyreflyClient(server=PyreflyLocalServer()) as client:
+        async with PyreflyClient() as client:
             refs = await client.request_references(
                 file_path="src/main.py",
                 position=Position(21, 19),
@@ -37,7 +36,11 @@ async def main():
 from loguru import logger
 
 from .client.abc import LSPClient
+from .clients import PythonClient, RustClient, TypeScriptClient
+from .clients.pyrefly import PyreflyClient
 from .server.abc import LSPServer
+from .server.container import ContainerServer
+from .server.local import LocalServer
 from .utils.types import *  # noqa: F403
 
 logger.disable(__name__)
@@ -58,8 +61,14 @@ def disable_logging() -> None:
 }
 
 __all__ = [
+    "ContainerServer",
     "LSPClient",
     "LSPServer",
+    "LocalServer",
+    "PyreflyClient",
+    "PythonClient",
+    "RustClient",
+    "TypeScriptClient",
     "disable_logging",
     "enable_logging",
 ]

src/lsp_client/capability/notification/__init__.py

@@ -13,4 +13,5 @@
 __all__ = [
     "WithNotifyDidChangeConfiguration",
     "WithNotifyTextDocumentSynchronize",
+    "capabilities",
 ]

src/lsp_client/capability/request/__init__.py

@@ -40,4 +40,5 @@
     "WithRequestTypeDefinition",
     "WithRequestTypeHierarchy",
     "WithRequestWorkspaceSymbol",
+    "capabilities",
 ]

src/lsp_client/capability/server_notification/__init__.py

@@ -19,4 +19,5 @@
     "WithReceiveLogTrace",
     "WithReceivePublishDiagnostics",
     "WithReceiveShowMessage",
+    "capabilities",
 ]

src/lsp_client/capability/server_request/__init__.py

@@ -19,4 +19,5 @@
     "WithRespondShowDocumentRequest",
     "WithRespondShowMessageRequest",
     "WithRespondWorkspaceFoldersRequest",
+    "capabilities",
 ]

src/lsp_client/clients/__init__.py

@@ -19,6 +19,7 @@
 TypeScriptClient = TypescriptClient
 
 __all__ = [
+    "PyreflyClient",
     "PythonClient",
     "RustClient",
     "TypeScriptClient",