growthbook 1.4.0__tar.gz → 1.4.1__tar.gz

{growthbook-1.4.0/growthbook.egg-info → growthbook-1.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: growthbook
-Version: 1.4.0
+Version: 1.4.1
 Summary: Powerful Feature flagging and A/B testing for Python apps
 Home-page: https://github.com/growthbook/growthbook-python
 Author: GrowthBook

{growthbook-1.4.0 → growthbook-1.4.1}/growthbook/__init__.py

@@ -18,5 +18,5 @@ from .plugins import (
 )
 
 # x-release-please-start-version
-__version__ = "1.4.0"
+__version__ = "1.4.1"
 # x-release-please-end

{growthbook-1.4.0 → growthbook-1.4.1}/growthbook/growthbook.py

@@ -120,12 +120,13 @@ class InMemoryStickyBucketService(AbstractStickyBucketService):
 
 
 class SSEClient:
-    def __init__(self, api_host, client_key, on_event, reconnect_delay=5, headers=None):
+    def __init__(self, api_host, client_key, on_event, reconnect_delay=5, headers=None, timeout = 30):
         self.api_host = api_host
         self.client_key = client_key
 
         self.on_event = on_event
         self.reconnect_delay = reconnect_delay
+        self.timeout = timeout
 
         self._sse_session = None
         self._sse_thread = None
@@ -173,7 +174,8 @@ class SSEClient:
         
         while self.is_running:
             try:
-                async with aiohttp.ClientSession(headers=self.headers) as session:
+                async with aiohttp.ClientSession(headers=self.headers, 
+                    timeout=aiohttp.ClientTimeout(total=self.timeout)) as session:
                     self._sse_session = session
 
                     async with session.get(url) as response:
@@ -407,10 +409,10 @@ class FeatureRepository(object):
         return data
 
 
-    def startAutoRefresh(self, api_host, client_key, cb):
+    def startAutoRefresh(self, api_host, client_key, cb, streaming_timeout=30):
         if not client_key:
             raise ValueError("Must specify `client_key` to start features streaming")
-        self.sse_client = self.sse_client or SSEClient(api_host=api_host, client_key=client_key, on_event=cb)
+        self.sse_client = self.sse_client or SSEClient(api_host=api_host, client_key=client_key, on_event=cb, timeout=streaming_timeout)
         self.sse_client.connect()
 
     def stopAutoRefresh(self):
@@ -443,6 +445,7 @@ class GrowthBook(object):
         sticky_bucket_identifier_attributes: List[str] = None,
         savedGroups: dict = {},
         streaming: bool = False,
+        streaming_timeout: int = 30,
         plugins: List = None,
         # Deprecated args
         trackingCallback=None,
@@ -471,6 +474,7 @@ class GrowthBook(object):
         self._trackingCallback = on_experiment_viewed or trackingCallback
 
         self._streaming = streaming
+        self._streaming_timeout = streaming_timeout
 
         # Deprecated args
         self._user = user
@@ -587,7 +591,8 @@ class GrowthBook(object):
         feature_repo.startAutoRefresh(
             api_host=self._api_host, 
             client_key=self._client_key,
-            cb=self._dispatch_sse_event
+            cb=self._dispatch_sse_event,
+            streaming_timeout=self._streaming_timeout
         )
 
     def stopAutoRefresh(self):

{growthbook-1.4.0 → growthbook-1.4.1/growthbook.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: growthbook
-Version: 1.4.0
+Version: 1.4.1
 Summary: Powerful Feature flagging and A/B testing for Python apps
 Home-page: https://github.com/growthbook/growthbook-python
 Author: GrowthBook

{growthbook-1.4.0 → growthbook-1.4.1}/setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 1.4.0
+current_version = 1.4.1
 commit = True
 tag = True