From 536e5a94a73964302941f74ec427d904c71ed48d Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 18 Sep 2024 16:31:57 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- packages/jupyter-ai/jupyter_ai/chat_handlers/tools.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) 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: