fix: add timeout to SSE client

add timeout to SSE client

Author: madhuchavva

growthbook/growthbook.py

@@ -120,12 +120,13 @@ def destroy(self) -> None:
 
 
 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 @@ async def _init_session(self):
 
         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 @@ async def _fetch_features_async(
         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 @@ def __init__(
         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 @@ def __init__(
         self._trackingCallback = on_experiment_viewed or trackingCallback
 
         self._streaming = streaming
+        self._streaming_timeout = streaming_timeout
 
         # Deprecated args
         self._user = user
@@ -587,7 +591,8 @@ def startAutoRefresh(self):
         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):