|
5 | 5 | conversation sessions with the Copilot CLI. |
6 | 6 | """ |
7 | 7 |
|
| 8 | +from __future__ import annotations |
| 9 | + |
8 | 10 | import asyncio |
9 | 11 | import inspect |
10 | 12 | import threading |
@@ -106,7 +108,7 @@ def __init__(self, session_id: str, client: Any, workspace_path: str | None = No |
106 | 108 | self._shell_exit_handlers_lock = threading.Lock() |
107 | 109 | self._tracked_process_ids: set[str] = set() |
108 | 110 | self._tracked_process_ids_lock = threading.Lock() |
109 | | - self._register_shell_process: Callable[[str, "CopilotSession"], None] | None = None |
| 111 | + self._register_shell_process: Callable[[str, CopilotSession], None] | None = None |
110 | 112 | self._unregister_shell_process_fn: Callable[[str], None] | None = None |
111 | 113 | self._rpc: SessionRpc | None = None |
112 | 114 |
|
@@ -347,7 +349,7 @@ def _untrack_shell_process(self, process_id: str) -> None: |
347 | 349 |
|
348 | 350 | def _set_shell_process_callbacks( |
349 | 351 | self, |
350 | | - register: Callable[[str, "CopilotSession"], None], |
| 352 | + register: Callable[[str, CopilotSession], None], |
351 | 353 | unregister: Callable[[str], None], |
352 | 354 | ) -> None: |
353 | 355 | """Set the registration callbacks for shell process tracking. |
@@ -807,7 +809,7 @@ async def destroy(self) -> None: |
807 | 809 | ) |
808 | 810 | await self.disconnect() |
809 | 811 |
|
810 | | - async def __aenter__(self) -> "CopilotSession": |
| 812 | + async def __aenter__(self) -> CopilotSession: |
811 | 813 | """Enable use as an async context manager.""" |
812 | 814 | return self |
813 | 815 |
|
|
0 commit comments