revert change to LLMChain import module

jupyterlab · Oct 18, 2024 · 2b1083b · 2b1083b
1 parent 388b0bb
commit 2b1083b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/fix.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/fix.py
@@ -2,7 +2,7 @@
 
 from jupyter_ai.models import CellWithErrorSelection, HumanChatMessage
 from jupyter_ai_magics.providers import BaseProvider
-from langchain.chains.llm import LLMChain
+from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 
 from .base import BaseChatHandler, SlashCommandRoutingType