feat: Add support for provider config

Author: keelerm84

ldai/client.py

@@ -73,12 +73,29 @@ def get_attribute(self, key: str) -> Any:
         return self._attributes.get(key)
 
 
+class ProviderConfig:
+    """
+    Configuration related to the provider.
+    """
+
+    def __init__(self, id: str):
+        self._id = id
+
+    @property
+    def id(self) -> str:
+        """
+        The ID of the provider.
+        """
+        return self._id
+
+
 class AIConfig:
-    def __init__(self, tracker: LDAIConfigTracker, enabled: bool, model: Optional[ModelConfig], messages: Optional[List[LDMessage]]):
+    def __init__(self, tracker: LDAIConfigTracker, enabled: bool, model: Optional[ModelConfig], messages: Optional[List[LDMessage]], provider: Optional[ProviderConfig] = None):
         self.tracker = tracker
         self.enabled = enabled
         self.model = model
         self.messages = messages
+        self.provider = provider
 
 
 class LDAIClient:
@@ -124,6 +141,11 @@ def config(
                 for entry in variation['messages']
             ]
 
+        provider_config = None
+        if 'provider' in variation and isinstance(variation['provider'], dict):
+            provider = variation['provider']
+            provider_config = ProviderConfig(provider.get('id', ''))
+
         model = None
         if 'model' in variation:
             model = ModelConfig(
@@ -143,7 +165,8 @@ def config(
             ),
             enabled=bool(enabled),
             model=model,
-            messages=messages
+            messages=messages,
+            provider=provider_config,
         )
 
     def __interpolate_template(self, template: str, variables: Dict[str, Any]) -> str:

ldai/testing/test_model_config.py

@@ -14,6 +14,7 @@ def td() -> TestData:
         .variations(
             {
                 'model': {'modelId': 'fakeModel', 'temperature': 0.5, 'maxTokens': 4096},
+                'provider': {'id': 'fakeProvider'},
                 'messages': [{'role': 'system', 'content': 'Hello, {{name}}!'}],
                 '_ldMeta': {'enabled': True, 'versionKey': 'abcd'},
             },
@@ -161,6 +162,17 @@ def test_model_config_no_variables(ldai_client: LDAIClient, tracker):
     assert config.model.max_tokens == 4096
 
 
+def test_provider_config_handling(ldai_client: LDAIClient, tracker):
+    context = Context.builder('user-key').name("Sandy").build()
+    default_value = AIConfig(tracker=tracker, enabled=True, model=ModelConfig('fake-model'), messages=[])
+    variables = {'name': 'World'}
+
+    config = ldai_client.config('model-config', context, default_value, variables)
+
+    assert config.provider is not None
+    assert config.provider.id == 'fakeProvider'
+
+
 def test_context_interpolation(ldai_client: LDAIClient, tracker):
     context = Context.builder('user-key').name("Sandy").build()
     default_value = AIConfig(tracker=tracker, enabled=True, model=ModelConfig('fake-model'), messages=[])
@@ -225,6 +237,7 @@ def test_model_initial_config_disabled(ldai_client: LDAIClient, tracker):
     assert config.enabled is False
     assert config.model is None
     assert config.messages is None
+    assert config.provider is None
 
 
 def test_model_initial_config_enabled(ldai_client: LDAIClient, tracker):
@@ -236,3 +249,4 @@ def test_model_initial_config_enabled(ldai_client: LDAIClient, tracker):
     assert config.enabled is True
     assert config.model is None
     assert config.messages is None
+    assert config.provider is None