Add explicit context manager type hints to clients

Co-authored-by: Shri Sukhani <shrisukhani@users.noreply.github.com>

Author: cursoragent

hyperbrowser/client/async_client.py

@@ -1,4 +1,5 @@
-from typing import Mapping, Optional
+from types import TracebackType
+from typing import Mapping, Optional, Type
 
 from ..config import ClientConfig
 from ..transport.async_transport import AsyncTransport
@@ -44,8 +45,13 @@ def __init__(
     async def close(self) -> None:
         await self.transport.close()
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "AsyncHyperbrowser":
         return self
 
-    async def __aexit__(self, exc_type, exc_val, exc_tb):
+    async def __aexit__(
+        self,
+        exc_type: Optional[Type[BaseException]],
+        exc_val: Optional[BaseException],
+        exc_tb: Optional[TracebackType],
+    ) -> None:
         await self.close()

hyperbrowser/client/sync.py

@@ -1,4 +1,5 @@
-from typing import Mapping, Optional
+from types import TracebackType
+from typing import Mapping, Optional, Type
 
 from ..config import ClientConfig
 from ..transport.sync import SyncTransport
@@ -44,8 +45,13 @@ def __init__(
     def close(self) -> None:
         self.transport.close()
 
-    def __enter__(self):
+    def __enter__(self) -> "Hyperbrowser":
         return self
 
-    def __exit__(self, exc_type, exc_val, exc_tb):
+    def __exit__(
+        self,
+        exc_type: Optional[Type[BaseException]],
+        exc_val: Optional[BaseException],
+        exc_tb: Optional[TracebackType],
+    ) -> None:
         self.close()