growthbook 1.4.5__tar.gz → 1.4.6__tar.gz

{growthbook-1.4.5/growthbook.egg-info → growthbook-1.4.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: growthbook
-Version: 1.4.5
+Version: 1.4.6
 Summary: Powerful Feature flagging and A/B testing for Python apps
 Home-page: https://github.com/growthbook/growthbook-python
 Author: GrowthBook
@@ -499,7 +499,6 @@ class MyStickyBucketService(AbstractStickyBucketService):
         })
 
 # Pass in an instance of this service to your GrowthBook constructor
-
 gb = GrowthBook(
   sticky_bucket_service = MyStickyBucketService()
 )

{growthbook-1.4.5 → growthbook-1.4.6}/README.md

@@ -464,7 +464,6 @@ class MyStickyBucketService(AbstractStickyBucketService):
         })
 
 # Pass in an instance of this service to your GrowthBook constructor
-
 gb = GrowthBook(
   sticky_bucket_service = MyStickyBucketService()
 )

{growthbook-1.4.5 → growthbook-1.4.6}/growthbook/__init__.py

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

{growthbook-1.4.5 → growthbook-1.4.6}/growthbook/common_types.py

@@ -426,7 +426,7 @@ class Options:
     sticky_bucket_service: Optional[AbstractStickyBucketService] = None
     sticky_bucket_identifier_attributes: Optional[List[str]] = None
     on_experiment_viewed: Optional[Callable[[Experiment, Result, Optional[UserContext]], None]] = None
-    on_feature_usage: Optional[Callable[[str, 'FeatureResult'], None]] = None
+    on_feature_usage: Optional[Callable[[str, 'FeatureResult', UserContext], None]] = None
     tracking_plugins: Optional[List[Any]] = None
 
 

{growthbook-1.4.5 → growthbook-1.4.6}/growthbook/growthbook.py

@@ -503,6 +503,10 @@ class FeatureRepository(object):
     
     def start_background_refresh(self, api_host: str, client_key: str, decryption_key: str, ttl: int = 600, refresh_interval: int = 300) -> None:
         """Start periodic background refresh task"""
+
+        if not client_key:
+            raise ValueError("Must specify `client_key` to refresh features")
+
         with self._refresh_lock:
             if self._refresh_thread is not None:
                 return  # Already running
@@ -620,6 +624,7 @@ class GrowthBook(object):
         self._tracked: Dict[str, Any] = {}
         self._assigned: Dict[str, Any] = {}
         self._subscriptions: Set[Any] = set()
+        self._is_updating_features = False
 
         # support plugins
         self._plugins: List = plugins or []
@@ -662,7 +667,7 @@ class GrowthBook(object):
         if self._streaming:
             self.load_features()
             self.startAutoRefresh()
-        elif self._stale_while_revalidate and self._client_key:
+        elif self._stale_while_revalidate:
             # Start background refresh task for stale-while-revalidate
             self.load_features()  # Initial load
             feature_repo.start_background_refresh(
@@ -752,19 +757,24 @@ class GrowthBook(object):
         return self.set_features(features)
 
     def set_features(self, features: dict) -> None:
-        self._features = {}
-        for key, feature in features.items():
-            if isinstance(feature, Feature):
-                self._features[key] = feature
-            else:
-                self._features[key] = Feature(
-                    rules=feature.get("rules", []),
-                    defaultValue=feature.get("defaultValue", None),
-                )
-        # Update the global context with the new features and saved groups
-        self._global_ctx.features = self._features
-        self._global_ctx.saved_groups = self._saved_groups
-        self.refresh_sticky_buckets()
+        # Prevent infinite recursion during feature updates
+        self._is_updating_features = True
+        try:
+            self._features = {}
+            for key, feature in features.items():
+                if isinstance(feature, Feature):
+                    self._features[key] = feature
+                else:
+                    self._features[key] = Feature(
+                        rules=feature.get("rules", []),
+                        defaultValue=feature.get("defaultValue", None),
+                    )
+            # Update the global context with the new features and saved groups
+            self._global_ctx.features = self._features
+            self._global_ctx.saved_groups = self._saved_groups
+            self.refresh_sticky_buckets()
+        finally:
+            self._is_updating_features = False
 
     # @deprecated, use get_features
     def getFeatures(self) -> Dict[str, Feature]:
@@ -864,6 +874,10 @@ class GrowthBook(object):
     def _ensure_fresh_features(self) -> None:
         """Lazy refresh: Check cache expiry and refresh if needed, but only if client_key is provided"""
         
+        # Prevent infinite recursion when updating features (e.g., during sticky bucket refresh)
+        if self._is_updating_features:
+            return
+        
         if self._streaming or self._stale_while_revalidate or not self._client_key:
             return  # Skip cache checks - SSE or background refresh handles freshness
 
@@ -897,7 +911,7 @@ class GrowthBook(object):
         # Call feature usage callback if provided
         if self._featureUsageCallback:
             try:
-                self._featureUsageCallback(key, result)
+                self._featureUsageCallback(key, result, self._user_ctx)
             except Exception:
                 pass
         return result

{growthbook-1.4.5 → growthbook-1.4.6}/growthbook/growthbook_client.py

@@ -503,7 +503,7 @@ class GrowthBookClient:
             # Call feature usage callback if provided
             if self.options.on_feature_usage:
                 try:
-                    self.options.on_feature_usage(key, result)
+                    self.options.on_feature_usage(key, result, user_context)
                 except Exception:
                     logger.exception("Error in feature usage callback")
             return result
@@ -516,7 +516,7 @@ class GrowthBookClient:
             # Call feature usage callback if provided
             if self.options.on_feature_usage:
                 try:
-                    self.options.on_feature_usage(key, result)
+                    self.options.on_feature_usage(key, result, user_context)
                 except Exception:
                     logger.exception("Error in feature usage callback")
             return result.on
@@ -529,7 +529,7 @@ class GrowthBookClient:
             # Call feature usage callback if provided
             if self.options.on_feature_usage:
                 try:
-                    self.options.on_feature_usage(key, result)
+                    self.options.on_feature_usage(key, result, user_context)
                 except Exception:
                     logger.exception("Error in feature usage callback")
             return result.off
@@ -541,7 +541,7 @@ class GrowthBookClient:
             # Call feature usage callback if provided
             if self.options.on_feature_usage:
                 try:
-                    self.options.on_feature_usage(key, result)
+                    self.options.on_feature_usage(key, result, user_context)
                 except Exception:
                     logger.exception("Error in feature usage callback")
             return result.value if result.value is not None else fallback

{growthbook-1.4.5 → growthbook-1.4.6/growthbook.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: growthbook
-Version: 1.4.5
+Version: 1.4.6
 Summary: Powerful Feature flagging and A/B testing for Python apps
 Home-page: https://github.com/growthbook/growthbook-python
 Author: GrowthBook
@@ -499,7 +499,6 @@ class MyStickyBucketService(AbstractStickyBucketService):
         })
 
 # Pass in an instance of this service to your GrowthBook constructor
-
 gb = GrowthBook(
   sticky_bucket_service = MyStickyBucketService()
 )

{growthbook-1.4.5 → growthbook-1.4.6}/setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 1.4.5
+current_version = 1.4.6
 commit = True
 tag = True
 

{growthbook-1.4.5 → growthbook-1.4.6}/tests/test_growthbook.py

@@ -219,8 +219,8 @@ def test_feature_usage_callback():
     """Test that feature usage callback is called correctly"""
     calls = []
     
-    def feature_usage_cb(key, result):
-        calls.append([key, result])
+    def feature_usage_cb(key, result, user_context):
+        calls.append([key, result, user_context])
     
     gb = GrowthBook(
         attributes={"id": "1"},
@@ -243,12 +243,14 @@ def test_feature_usage_callback():
     assert calls[0][0] == "feature-1"
     assert calls[0][1].value is True
     assert calls[0][1].source == "defaultValue"
+    assert calls[0][2].attributes == {"id": "1"}
     
     # Test is_on
     gb.is_on("feature-2")
     assert len(calls) == 2
     assert calls[1][0] == "feature-2"
     assert calls[1][1].value is False
+    assert calls[1][2].attributes == {"id": "1"}
     
     # Test get_feature_value
     value = gb.get_feature_value("feature-3", "blue")
@@ -256,11 +258,13 @@ def test_feature_usage_callback():
     assert calls[2][0] == "feature-3"
     assert calls[2][1].value == "red"
     assert value == "red"
+    assert calls[2][2].attributes == {"id": "1"}
     
     # Test is_off
     gb.is_off("feature-1")
     assert len(calls) == 4
     assert calls[3][0] == "feature-1"
+    assert calls[3][2].attributes == {"id": "1"}
     
     # Calling same feature multiple times should trigger callback each time
     gb.eval_feature("feature-1")
@@ -273,7 +277,7 @@ def test_feature_usage_callback():
 def test_feature_usage_callback_error_handling():
     """Test that feature usage callback errors are handled gracefully"""
     
-    def failing_callback(key, result):
+    def failing_callback(key, result, user_context):
         raise Exception("Callback error")
     
     gb = GrowthBook(

{growthbook-1.4.5 → growthbook-1.4.6}/tests/test_growthbook_client.py

@@ -776,8 +776,8 @@ async def test_feature_usage_callback():
     """Test that feature usage callback is called correctly"""
     calls = []
     
-    def feature_usage_cb(key, result):
-        calls.append([key, result])
+    def feature_usage_cb(key, result, user_context):
+        calls.append([key, result, user_context])
     
     client = GrowthBookClient(Options(
         api_host="https://localhost.growthbook.io",
@@ -820,12 +820,14 @@ async def test_feature_usage_callback():
             assert calls[0][0] == "feature-1"
             assert calls[0][1].value is True
             assert calls[0][1].source == "defaultValue"
+            assert calls[0][2].attributes == {"id": "1"}
             
             # Test is_on
             await client.is_on("feature-2", user_context)
             assert len(calls) == 2
             assert calls[1][0] == "feature-2"
             assert calls[1][1].value is False
+            assert calls[1][2].attributes == {"id": "1"}
             
             # Test get_feature_value
             value = await client.get_feature_value("feature-3", "blue", user_context)
@@ -833,11 +835,13 @@ async def test_feature_usage_callback():
             assert calls[2][0] == "feature-3"
             assert calls[2][1].value == "red"
             assert value == "red"
+            assert calls[2][2].attributes == {"id": "1"}
             
             # Test is_off
             await client.is_off("feature-1", user_context)
             assert len(calls) == 4
             assert calls[3][0] == "feature-1"
+            assert calls[3][2].attributes == {"id": "1"}
             
             # Calling same feature multiple times should trigger callback each time
             await client.eval_feature("feature-1", user_context)
@@ -852,7 +856,7 @@ async def test_feature_usage_callback():
 async def test_feature_usage_callback_error_handling():
     """Test that feature usage callback errors are handled gracefully"""
     
-    def failing_callback(key, result):
+    def failing_callback(key, result, user_context):
         raise Exception("Callback error")
     
     client = GrowthBookClient(Options(