Fix use of jai_config_manager

jupyterlab · May 2, 2024 · 5ac3cf3 · 5ac3cf3
1 parent 18e57f4
commit 5ac3cf3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/jupyter-ai/jupyter_ai/completions/handlers/model_mixin.py b/packages/jupyter-ai/jupyter_ai/completions/handlers/model_mixin.py
@@ -26,8 +26,8 @@ def __init__(self, *args, **kwargs) -> None:
         self._llm_params = None
 
     def get_llm(self) -> Optional[BaseProvider]:
-        lm_provider = self.config_manager.completions_lm_provider
-        lm_provider_params = self.config_manager.completions_lm_provider_params
+        lm_provider = self.jai_config_manager.completions_lm_provider
+        lm_provider_params = self.jai_config_manager.completions_lm_provider_params
 
         if not lm_provider or not lm_provider_params:
             return None