From 5ac3cf307dad46c409c5f4ea3673ceefe9af4a05 Mon Sep 17 00:00:00 2001
From: krassowski <5832902+krassowski@users.noreply.github.com>
Date: Thu, 2 May 2024 18:09:24 +0100
Subject: [PATCH] Fix use of `jai_config_manager`

---
 .../jupyter-ai/jupyter_ai/completions/handlers/model_mixin.py | 4 ++--
 1 file changed, 2 insertions(+), 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
index 4bd3f9ce6..fd19498bd 100644
--- 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