@@ -131,7 +131,7 @@ def test_model_config_interpolation(ldai_client: LDAIClient, tracker):
131
131
)
132
132
variables = {'name' : 'World' }
133
133
134
- config = ldai_client .model_config ('model-config' , context , default_value , variables )
134
+ config = ldai_client .config ('model-config' , context , default_value , variables )
135
135
136
136
assert config .messages is not None
137
137
assert len (config .messages ) > 0
@@ -148,7 +148,7 @@ def test_model_config_no_variables(ldai_client: LDAIClient, tracker):
148
148
context = Context .create ('user-key' )
149
149
default_value = AIConfig (tracker = tracker , enabled = True , model = ModelConfig ('fake-model' ), messages = [])
150
150
151
- config = ldai_client .model_config ('model-config' , context , default_value , {})
151
+ config = ldai_client .config ('model-config' , context , default_value , {})
152
152
153
153
assert config .messages is not None
154
154
assert len (config .messages ) > 0
@@ -166,7 +166,7 @@ def test_context_interpolation(ldai_client: LDAIClient, tracker):
166
166
default_value = AIConfig (tracker = tracker , enabled = True , model = ModelConfig ('fake-model' ), messages = [])
167
167
variables = {'name' : 'World' }
168
168
169
- config = ldai_client .model_config (
169
+ config = ldai_client .config (
170
170
'ctx-interpolation' , context , default_value , variables
171
171
)
172
172
@@ -187,7 +187,7 @@ def test_model_config_multiple(ldai_client: LDAIClient, tracker):
187
187
default_value = AIConfig (tracker = tracker , enabled = True , model = ModelConfig ('fake-model' ), messages = [])
188
188
variables = {'name' : 'World' , 'day' : 'Monday' }
189
189
190
- config = ldai_client .model_config (
190
+ config = ldai_client .config (
191
191
'multiple-messages' , context , default_value , variables
192
192
)
193
193
@@ -207,7 +207,7 @@ def test_model_config_disabled(ldai_client: LDAIClient, tracker):
207
207
context = Context .create ('user-key' )
208
208
default_value = AIConfig (tracker = tracker , enabled = False , model = ModelConfig ('fake-model' ), messages = [])
209
209
210
- config = ldai_client .model_config ('off-config' , context , default_value , {})
210
+ config = ldai_client .config ('off-config' , context , default_value , {})
211
211
212
212
assert config .model is not None
213
213
assert config .enabled is False
@@ -220,7 +220,7 @@ def test_model_initial_config_disabled(ldai_client: LDAIClient, tracker):
220
220
context = Context .create ('user-key' )
221
221
default_value = AIConfig (tracker = tracker , enabled = False , model = ModelConfig ('fake-model' ), messages = [])
222
222
223
- config = ldai_client .model_config ('initial-config-disabled' , context , default_value , {})
223
+ config = ldai_client .config ('initial-config-disabled' , context , default_value , {})
224
224
225
225
assert config .enabled is False
226
226
assert config .model is None
@@ -231,7 +231,7 @@ def test_model_initial_config_enabled(ldai_client: LDAIClient, tracker):
231
231
context = Context .create ('user-key' )
232
232
default_value = AIConfig (tracker = tracker , enabled = False , model = ModelConfig ('fake-model' ), messages = [])
233
233
234
- config = ldai_client .model_config ('initial-config-enabled' , context , default_value , {})
234
+ config = ldai_client .config ('initial-config-enabled' , context , default_value , {})
235
235
236
236
assert config .enabled is True
237
237
assert config .model is None
0 commit comments