@@ -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