feat: allow setting bandits initial configuration

Author: rasendubi

eppo_client/__init__.py

@@ -40,6 +40,10 @@ def init(config: Config) -> EppoClient:
         flag_config_store.set_configurations(
             config.initial_configuration._flags_configuration.flags
         )
+        if config.initial_configuration._bandits_configuration:
+            bandit_config_store.set_configurations(
+                config.initial_configuration._bandits_configuration.bandits
+            )
 
     config_requestor = ExperimentConfigurationRequestor(
         http_client=http_client,

eppo_client/configuration.py

@@ -1,4 +1,5 @@
-from eppo_client.models import UfcResponse
+from typing import Union
+from eppo_client.models import UfcResponse, BanditResponse
 
 
 class Configuration:
@@ -7,5 +8,15 @@ class Configuration:
     interpret feature flags.
     """
 
-    def __init__(self, flags_configuration: str):
+    def __init__(
+        self,
+        flags_configuration: Union[bytes, str],
+        bandits_configuration: Union[bytes, str, None] = None,
+    ) -> None:
         self._flags_configuration = UfcResponse.model_validate_json(flags_configuration)
+
+        self._bandits_configuration = None
+        if bandits_configuration is not None:
+            self._bandits_configuration = BanditResponse.model_validate_json(
+                bandits_configuration
+            )

eppo_client/configuration_requestor.py

@@ -80,3 +80,7 @@ def _set_configuration(self, configuration: Configuration):
         self.__flag_config_store.set_configurations(
             configuration._flags_configuration.flags
         )
+        if configuration._bandits_configuration is not None:
+            self.__bandit_config_store.set_configurations(
+                configuration._bandits_configuration.bandits
+            )

eppo_client/configuration_store.py

@@ -6,7 +6,7 @@
 
 
 class ConfigurationStore(Generic[T]):
-    def __init__(self):
+    def __init__(self) -> None:
         self.__is_initialized = False
         self.__cache: Dict[str, T] = {}
         self.__lock = ReadWriteLock()

eppo_client/models.py

@@ -105,3 +105,7 @@ class BanditData(SdkBaseModel):
     bandit_model_version: str = Field(alias="modelVersion")
     bandit_model_data: BanditModelData = Field(alias="modelData")
     updated_at: datetime
+
+
+class BanditResponse(SdkBaseModel):
+    bandits: Dict[str, BanditData]

test/configuration_test.py

@@ -5,7 +5,7 @@
 
 
 def test_init_valid():
-    Configuration(flags_configuration='{"flags": {}}')
+    Configuration(flags_configuration=b'{"flags": {}}')
 
 
 def test_init_invalid_json():
@@ -15,4 +15,4 @@ def test_init_invalid_json():
 
 def test_init_invalid_format():
     with pytest.raises(pydantic.ValidationError):
-        Configuration(flags_configuration='{"flags": []}')
+        Configuration(flags_configuration=b'{"flags": []}')