refactor: add type annotations for __exit__ and __aexit__

src/bentoml/client.py

@@ -20,6 +20,9 @@
 from .exceptions import BentoMLException
 from ._internal.service.inference_api import InferenceAPI
 
+if t.TYPE_CHECKING:
+    from types import TracebackType
+
 
 class Client(ABC):
     server_url: str
@@ -88,13 +91,23 @@ def wait_until_server_is_ready(host: str, port: int, timeout: int) -> None:
     def __enter__(self):
         return self
 
-    def __exit__(self, type, value, traceback):
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> bool | None:
         pass
 
     async def __aenter__(self):
         return self
 
-    async def __aexit__(self, type, value, traceback):
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> bool | None:
         pass
 
     @staticmethod