diff --git a/packages/jupyter-ai/jupyter_ai/tests/test_config_manager.py b/packages/jupyter-ai/jupyter_ai/tests/test_config_manager.py index 9a9ce09be..4a739f6e5 100644 --- a/packages/jupyter-ai/jupyter_ai/tests/test_config_manager.py +++ b/packages/jupyter-ai/jupyter_ai/tests/test_config_manager.py @@ -468,6 +468,7 @@ def test_returns_chat_model_fields(cm): expected_model_args = configure_with_fields(cm) assert cm.lm_provider_params == expected_model_args + def test_returns_completion_model_fields(cm): expected_model_args = configure_with_fields(cm, completions=True) assert cm.completions_lm_provider_params == expected_model_args