Adjust name of variable to reflect new implementation

jupyterlab · Apr 8, 2024 · a50f81a · a50f81a
1 parent 30a5792
commit a50f81a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/jupyter-ai/jupyter_ai/completions/handlers/llm_mixin.py b/packages/jupyter-ai/jupyter_ai/completions/handlers/llm_mixin.py
@@ -41,19 +41,19 @@ def get_llm(self) -> Optional[BaseProvider]:
             else None
         )
 
-        should_recreate_chain = False
+        should_recreate_llm = False
         if curr_lm_id != next_lm_id:
             self.log.info(
                 f"Switching {self.handler_kind} language model from {curr_lm_id} to {next_lm_id}."
             )
-            should_recreate_chain = True
+            should_recreate_llm = True
         elif self._llm_params != lm_provider_params:
             self.log.info(
                 f"{self.handler_kind} model params changed, updating the llm chain."
             )
-            should_recreate_chain = True
+            should_recreate_llm = True
 
-        if should_recreate_chain:
+        if should_recreate_llm:
             self._llm = self.create_llm(lm_provider, lm_provider_params)
             self._llm_params = lm_provider_params