refactor: add type annotations for __exit__ and __aexit__

src/bentoml/client.py

@@ -19,6 +19,10 @@
 from .exceptions import RemoteException
 from .exceptions import BentoMLException
 from ._internal.service.inference_api import InferenceAPI
+from .exceptions import BentoMLException, RemoteException
+
+if t.TYPE_CHECKING:
+    from types import TracebackType
 
 
 class Client(ABC):
@@ -85,13 +89,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