diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/fix.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/fix.py index 95c3efcaa..390b93cf6 100644 --- 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