Skip to content
9 changes: 5 additions & 4 deletions plugin/core/sessions.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,8 @@
from .settings import client_configs
from .settings import globalprefs
from .settings import userprefs
from .transports import Transport
from .transports import Json
from .transports import TransportWrapper
from .transports import TransportCallbacks
from .types import Capabilities
from .types import ClientConfig
Expand Down Expand Up @@ -1265,11 +1266,11 @@ def check_applicable(self, sb: SessionBufferProtocol) -> None:
_PARTIAL_RESULT_PROGRESS_PREFIX = "$ublime-partial-result-progress-"


class Session(TransportCallbacks):
class Session(TransportCallbacks[Json]):

def __init__(self, manager: Manager, logger: Logger, workspace_folders: list[WorkspaceFolder],
config: ClientConfig, plugin_class: type[AbstractPlugin] | None) -> None:
self.transport: Transport | None = None
self.transport: TransportWrapper[Json] | None = None
self.working_directory: str | None = None
self.request_id = 0 # Our request IDs are always integers.
self._logger = logger
Expand Down Expand Up @@ -1533,7 +1534,7 @@ def initialize_async(
self,
variables: dict[str, str],
working_directory: str | None,
transport: Transport,
transport: TransportWrapper[Json],
init_callback: InitCallback
) -> None:
self.transport = transport
Expand Down
Loading
Loading