Skip to content

Commit 3a3e913

Browse files
committed
fix!: Rename LDAIClient.model_config to LDAIClient.config
1 parent 9a86f0a commit 3a3e913

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

ldai/client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,15 +87,15 @@ class LDAIClient:
8787
def __init__(self, client: LDClient):
8888
self.client = client
8989

90-
def model_config(
90+
def config(
9191
self,
9292
key: str,
9393
context: Context,
9494
default_value: AIConfig,
9595
variables: Optional[Dict[str, Any]] = None,
9696
) -> AIConfig:
9797
"""
98-
Get the value of a model configuration asynchronously.
98+
Get the value of a model configuration.
9999
100100
:param key: The key of the model configuration.
101101
:param context: The context to evaluate the model configuration in.

ldai/testing/test_model_config.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ def test_model_config_interpolation(ldai_client: LDAIClient, tracker):
131131
)
132132
variables = {'name': 'World'}
133133

134-
config = ldai_client.model_config('model-config', context, default_value, variables)
134+
config = ldai_client.config('model-config', context, default_value, variables)
135135

136136
assert config.messages is not None
137137
assert len(config.messages) > 0
@@ -148,7 +148,7 @@ def test_model_config_no_variables(ldai_client: LDAIClient, tracker):
148148
context = Context.create('user-key')
149149
default_value = AIConfig(tracker=tracker, enabled=True, model=ModelConfig('fake-model'), messages=[])
150150

151-
config = ldai_client.model_config('model-config', context, default_value, {})
151+
config = ldai_client.config('model-config', context, default_value, {})
152152

153153
assert config.messages is not None
154154
assert len(config.messages) > 0
@@ -166,7 +166,7 @@ def test_context_interpolation(ldai_client: LDAIClient, tracker):
166166
default_value = AIConfig(tracker=tracker, enabled=True, model=ModelConfig('fake-model'), messages=[])
167167
variables = {'name': 'World'}
168168

169-
config = ldai_client.model_config(
169+
config = ldai_client.config(
170170
'ctx-interpolation', context, default_value, variables
171171
)
172172

@@ -187,7 +187,7 @@ def test_model_config_multiple(ldai_client: LDAIClient, tracker):
187187
default_value = AIConfig(tracker=tracker, enabled=True, model=ModelConfig('fake-model'), messages=[])
188188
variables = {'name': 'World', 'day': 'Monday'}
189189

190-
config = ldai_client.model_config(
190+
config = ldai_client.config(
191191
'multiple-messages', context, default_value, variables
192192
)
193193

@@ -207,7 +207,7 @@ def test_model_config_disabled(ldai_client: LDAIClient, tracker):
207207
context = Context.create('user-key')
208208
default_value = AIConfig(tracker=tracker, enabled=False, model=ModelConfig('fake-model'), messages=[])
209209

210-
config = ldai_client.model_config('off-config', context, default_value, {})
210+
config = ldai_client.config('off-config', context, default_value, {})
211211

212212
assert config.model is not None
213213
assert config.enabled is False
@@ -220,7 +220,7 @@ def test_model_initial_config_disabled(ldai_client: LDAIClient, tracker):
220220
context = Context.create('user-key')
221221
default_value = AIConfig(tracker=tracker, enabled=False, model=ModelConfig('fake-model'), messages=[])
222222

223-
config = ldai_client.model_config('initial-config-disabled', context, default_value, {})
223+
config = ldai_client.config('initial-config-disabled', context, default_value, {})
224224

225225
assert config.enabled is False
226226
assert config.model is None
@@ -231,7 +231,7 @@ def test_model_initial_config_enabled(ldai_client: LDAIClient, tracker):
231231
context = Context.create('user-key')
232232
default_value = AIConfig(tracker=tracker, enabled=False, model=ModelConfig('fake-model'), messages=[])
233233

234-
config = ldai_client.model_config('initial-config-enabled', context, default_value, {})
234+
config = ldai_client.config('initial-config-enabled', context, default_value, {})
235235

236236
assert config.enabled is True
237237
assert config.model is None

0 commit comments

Comments
 (0)