getstream 3.0.2__tar.gz → 3.0.3__tar.gz

{getstream-3.0.2 → getstream-3.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: getstream
-Version: 3.0.2
+Version: 3.0.3
 Summary: GetStream Python SDK - Build scalable activity feeds, chat, and video calling applications
 Author-email: sachaarbonel <sacha.arbonel@hotmail.fr>, tbarbugli <tbarbugli@gmail.com>
 License-File: LICENSE.md

{getstream-3.0.2 → getstream-3.0.3}/getstream/stream.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from contextlib import AsyncExitStack
 from functools import cached_property
 import time
 from typing import List, Optional
@@ -207,6 +208,20 @@ class AsyncStream(BaseStream, AsyncCommonClient):
             user_agent=self.user_agent,
         )
 
+    async def aclose(self):
+        """Close all child clients and the main HTTPX client."""
+        # AsyncExitStack ensures all clients are closed even if one fails.
+        # video/chat/moderation are @cached_property - only close if accessed.
+        async with AsyncExitStack() as stack:
+            cached = self.__dict__
+            if "video" in cached:
+                stack.push_async_callback(self.video.aclose)
+            if "chat" in cached:
+                stack.push_async_callback(self.chat.aclose)
+            if "moderation" in cached:
+                stack.push_async_callback(self.moderation.aclose)
+            stack.push_async_callback(super().aclose)
+
     @cached_property
     def feeds(self):
         raise NotImplementedError("Feeds not supported for async client")