fix: add type annotations to context manager methods (#12)

Author: IISweetHeartII

agentgram/client.py

@@ -81,11 +81,11 @@ def close(self) -> None:
         """Close the HTTP client and cleanup resources."""
         self._http.close()
 
-    def __enter__(self):
+    def __enter__(self) -> "AgentGram":
         """Context manager entry."""
         return self
 
-    def __exit__(self, *args):
+    def __exit__(self, *args: object) -> None:
         """Context manager exit."""
         self.close()
 
@@ -164,10 +164,10 @@ async def close(self) -> None:
         """Close the async HTTP client and cleanup resources."""
         await self._http.close()
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "AsyncAgentGram":
         """Async context manager entry."""
         return self
 
-    async def __aexit__(self, *args):
+    async def __aexit__(self, *args: object) -> None:
         """Async context manager exit."""
         await self.close()

agentgram/http.py

@@ -113,11 +113,11 @@ def close(self) -> None:
         """Close the HTTP client."""
         self._client.close()
 
-    def __enter__(self):
+    def __enter__(self) -> "HTTPClient":
         """Context manager entry."""
         return self
 
-    def __exit__(self, *args):
+    def __exit__(self, *args: object) -> None:
         """Context manager exit."""
         self.close()
 
@@ -221,10 +221,10 @@ async def close(self) -> None:
         """Close the async HTTP client."""
         await self._client.aclose()
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "AsyncHTTPClient":
         """Async context manager entry."""
         return self
 
-    async def __aexit__(self, *args):
+    async def __aexit__(self, *args: object) -> None:
         """Async context manager exit."""
         await self.close()