diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/tools.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/tools.py index 93241a17d..d0a420dde 100644 --- a/packages/jupyter-ai/jupyter_ai/chat_handlers/tools.py +++ b/packages/jupyter-ai/jupyter_ai/chat_handlers/tools.py @@ -3,13 +3,12 @@ import os from pathlib import Path from typing import Dict, Literal, Type -import numpy as np +import numpy as np from jupyter_ai.models import HumanChatMessage from jupyter_ai_magics.providers import BaseProvider from langchain.chains import ConversationalRetrievalChain, LLMChain from langchain.memory import ConversationBufferWindowMemory - from langchain_core.messages import AIMessage from langchain_core.prompts import PromptTemplate from langchain_core.tools import tool @@ -18,7 +17,6 @@ from .base import BaseChatHandler, SlashCommandRoutingType - PROMPT_TEMPLATE = """Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question. Chat History: