async with

livekit · Nov 13, 2024 · d861c2f · d861c2f
1 parent ea4875b
commit d861c2f
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/livekit-api/livekit/api/livekit_api.py b/livekit-api/livekit/api/livekit_api.py
@@ -87,3 +87,15 @@ async def aclose(self):
         
         Call this before your application exits or when the API client is no longer needed."""
         await self._session.close()
+
+    async def __aenter__(self):
+        """@private
+        
+        Support for `async with`"""
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        """@private
+        
+        Support for `async with`"""
+        await self.aclose()