Update fix.py

jupyterlab · Oct 28, 2024 · 298db5e · 298db5e
1 parent 9fc4beb
commit 298db5e
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/fix.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/fix.py
@@ -3,7 +3,6 @@
 from jupyter_ai.models import CellWithErrorSelection, HumanChatMessage
 from jupyter_ai_magics.providers import BaseProvider
 from langchain.prompts import PromptTemplate
-from langchain_core.runnables.history import RunnableWithMessageHistory
 
 from .base import BaseChatHandler, SlashCommandRoutingType
 
@@ -76,8 +75,6 @@ def create_llm_chain(
         llm = provider(**unified_parameters)
         self.llm = llm
         prompt_template = FIX_PROMPT_TEMPLATE
-        self.prompt_template = prompt_template
-        print("PROMPT TEMPLATE", prompt_template)
 
         runnable = prompt_template | llm  # type:ignore
         self.llm_chain = runnable